Leetcode 54. 螺旋矩阵（二维数组移动坐标）

ooold_six2024-06-08 20:55

使用vis数组记录该位置是否已经被访问

定义一个int型dir来记录方向，0123分别代表右下左上

当越界或碰壁已访问的位置后，修改dir并计算下一个位置

否则根据原dir计算下一个位置

java 复制代码

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> list = new ArrayList<>();
        int n = matrix.length;
        int m = matrix[0].length;
        int dir = 0; // 0右  1下  2左  3上
        boolean vis[][] = new boolean[n][m];
        int x = 0;
        int y = 0;
        for(int i = 0; i < n*m; i ++){
            list.add(matrix[x][y]);
            vis[x][y] = true;
            if(dir == 0){
                if(y + 1 >= m || vis[x][y+1] == true){
                    dir = (dir + 1) % 4;
                    x = x + 1;
                }
                else{
                    y = y + 1;
                }
            }
            else if(dir == 1){
                if(x + 1 >= n || vis[x+1][y] == true){
                    dir = (dir + 1) % 4;
                    y = y - 1;
                }
                else{
                    x = x + 1;
                }
            }
            else if(dir == 2){
                if(y - 1 < 0 || vis[x][y-1] == true){
                    dir = (dir + 1) % 4;
                    x = x - 1;
                }
                else{
                    y = y - 1;
                }
            }
            else{
                if(x - 1 < 0 || vis[x-1][y] == true){
                    dir = (dir + 1) % 4;
                    y = y + 1;
                }
                else{
                    x = x - 1;
                }
            }
        }
        return list;
    }
}

简化

常用思路，使用二维数组dirs[4][2]进行坐标的上下左右四个方位移动，简化代码

（也常用dirs[8][2]获取坐标的周围8个位置）

使用dirs和dir计算下一个将要使用的新坐标，若越界或碰壁，则将dir修改，获得正确的新坐标

java 复制代码

class Solution {
    public List<Integer> spiralOrder(int[][] matrix) {
        List<Integer> list = new ArrayList<>();
        int n = matrix.length;
        int m = matrix[0].length;
        int dir = 0; // 0右  1下  2左  3上
        int dirs[][] = {{0,1},{1,0},{0,-1},{-1,0}}; // 对应dir四种移动坐标变化
        boolean vis[][] = new boolean[n][m];
        int x = 0;
        int y = 0;
        for(int i = 0; i < n*m; i ++){
            list.add(matrix[x][y]);
            vis[x][y] = true;
            // 判断
            int newX = x + dirs[dir][0];
            int newY = y + dirs[dir][1];
            if(newX < 0 || newX >= n || newY < 0 || newY >= m || vis[newX][newY] == true){
                dir = (dir + 1) % 4;
            }
            // 移动
            x += dirs[dir][0];
            y += dirs[dir][1];
        }
        return list;
    }
}