LeetCode 54. 螺旋矩阵

54. 螺旋矩阵

  

给你一个 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

思路：

注：此题与 剑指 Offer 29. 顺时针打印矩阵_dreamer'~的博客-CSDN博客 相同

 参考：LeetCode大佬题解
 

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

    res := make([]int, 0)
    // 左右上下的四个边界，四个边界范围都是不断缩小的
    left, right, up, down := 0, len(matrix[0])-1, 0, len(matrix)-1
    for {
        // 从左到右
        for i := left; i <= right; i++ { // right、down都已经减一了，所以这里 i <= right
            res = append(res, matrix[up][i])
        } 
        up++ // 缩小上边界
        if up > down {
            break
        }

        // 从上到下
        for i := up; i <= down; i++ {
            res = append(res, matrix[i][right])
        }
        right-- // 缩小右边界
        if right < left {
            break
        }

        // 从右到左
        for i := right; i >= left; i-- {
            res = append(res, matrix[down][i])
        }
        down-- // 缩小下边界
        if down < up {
            break
        }

        // 从下到上
        for i := down; i >= up; i-- {
            res = append(res, matrix[i][left])
        }
        left++ // 缩小左边界
        if left > right {
            break
        }
    }
    return res
}