29. 顺时针打印矩阵

剑指offer 29 顺时针打印矩阵 LeetCode 54

54. 螺旋矩阵

LCR 146. 螺旋遍历二维数组

给你一个 m 行 n 列的矩阵matrix，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。

示例 1：

输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,3,6,9,8,7,4,5]

示例 2：

输入：matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出：[1,2,3,4,8,12,11,10,9,5,6,7]

提示：

• m == matrix.length
• n == matrix[i].length
• 1 <= m, n <= 10
• -100 <= matrix[i][j] <= 100

解题思路：偏移量技巧

按照顺时针遍历矩阵，当索引越界或者元素已经被访问过则改变方向，方向顺序为右，下，左，上，这是之前多次介绍过的偏移量技巧的典型题目。

Java代码

class Solution {
    public int[] spiralOrder(int[][] matrix) {
         /*按照顺时针遍历矩阵，当索引越界或者元素已经被访问过则改变方向，方向顺序为右，下，左，上*/
        List<Integer> res = new ArrayList<Integer>();
        if(matrix == null || matrix.length==0 || matrix[0] == null) return new int[0];
        boolean[][] isVisited = new boolean[matrix.length][matrix[0].length];
        int[] dx = {-1,0,1,0},dy = {0,1,0,-1};//偏移量技巧
        int x = 0,y = 0,d = 1;//最开始从（0，0）位置开始，向右遍历
        for(int i = 0;i < matrix.length * matrix[0].length;i++){
            res.add(matrix[x][y]);
            isVisited[x][y] = true;//标记该点被访问
            int a = x + dx[d],b = y+dy[d];//遍历下一个点
            if(a < 0 || a >= matrix.length || b < 0 || b >= matrix[0].length || isVisited[a][b]){
                d = (d+1) % 4;//索引越界或者该点被访问过，需要转弯了
                a = x + dx[d];
                b = y + dy[d];
            }
            x = a;
            y = b;//下一次要遍历的点
        }
        int[] result = new int[matrix.length * matrix[0].length];
        for(int i = 0;i < matrix.length * matrix[0].length;i++){
            result[i] = res.get(i);
        }
        return result;
    }
}

Go 代码

func spiralOrder(matrix [][]int) []int {
    /*剑指offer 面试题29，p161,ACWing 40
    按照顺时针遍历矩阵，将二维矩阵看成一维矩阵遍历每个元素，当索引越界或者元素已经被访问过则改变方向，
    改变方向顺序为右，下，左，上*/

    if len(matrix) == 0 || len(matrix[0]) == 0 {
        return []int{}
    }

    res := make([]int,len(matrix) * len(matrix[0]))
    // 定义四个方向  上，右，下，左  (上即行号减一，右即列号加1，以此类推)
    dx := []int{-1,0,1,0}
    dy := []int{0,1,0,-1}
    
    x,y,d := 0,0,1 // 从起点开始往右移动
    // 定义是否访问过的数组，并初始化
    isVisited := make([][]bool,len(matrix))
    for i,_ := range isVisited {
        isVisited[i] = make([]bool,len(matrix[0]))
    }

    for i := 0;i < len(matrix) * len(matrix[0]);i++ {
        res[i] = matrix[x][y] // 当前节点加入结果数组
        isVisited[x][y] = true // 标记当前节点已经访问过
        // 继续遍历下一个节点，但需要用下临时变量，不要用x = x + dx[d],因为等下可能还要用到原x
        a := x + dx[d]
        b := y + dy[d]
        // 如果当前节点越界了，或者已经遍历过了，则可以换方向了
        if a < 0 || a >= len(matrix) || b < 0 || b >= len(matrix[0]) || isVisited[a][b] {
            d = (d + 1) % 4
            // 新方向算出的下一个节点坐标，用到了原x,y
            a = x + dx[d]
            b = y + dy[d]
        }
        // 已经明确找到下一个节点位置了，可以继续遍历了
        x = a
        y = b
    }

    return res
}