54_spiral_matrix

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

代码参考：

package main

import "fmt"

func main() {
    fmt.Println(spiralOrder([][]int{
        // {1, 2, 3, 4},
        // {5, 6, 7, 8},
        // {9, 10, 11, 12},

        // {1, 2},
        // {3, 4},

        {2, 3, 4},
        {5, 6, 7},
        {8, 9, 10},
        {11, 12, 13},
        {14, 15, 16},
    }))
}

func spiralOrder(matrix [][]int) []int {
    if len(matrix) <= 0 || len(matrix[0]) <= 0 {
        return nil
    }
    if len(matrix) == 1 {
        return matrix[0]
    }
    return order(matrix, 0, len(matrix[0])-1, 0, len(matrix)-1, []int{})
}

func order(matrix [][]int, start, end int, up, down int, nums []int) []int {
    if start > end || up > down {
        return nums
    }
    // 向右走
    for i := start; i <= end; i++ {
        nums = append(nums, matrix[up][i])
    }

    // 向下走
    stop := true
    for i := up + 1; i <= down; i++ {
        nums = append(nums, matrix[i][end])
        stop = false
    }
    if stop {
        return nums // 无路可走
    }

    // 向左走
    stop = true
    for i := end - 1; i >= start; i-- {
        nums = append(nums, matrix[down][i])
        stop = false
    }
    if stop {
        return nums // 无路可走
    }

    // 向上走
    for i := down - 1; i > up; i-- {
        if end > 0 { // 单列的情况，下一趟不能往上走
            nums = append(nums, matrix[i][start])
        }
    }
    return order(matrix, start+1, end-1, up+1, down-1, nums)
}