螺旋矩阵
给你一个 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]
题解:
这道题倒是蛮经典的,主要注意两个细节,一个是循环结束的条件,另外一个是方向数组的控制,其他的基本写法类似于 BFS
func spiralOrder(matrix [][]int) []int {
dirs := [][]int{{0, 1}, {1, 0}, {0, -1}, {-1, 0}}
vis := make([][]bool, len(matrix))
for i := range len(matrix) {
vis[i] = make([]bool, len(matrix[0]))
}
total := len(matrix) * len(matrix[0])
dirindex := 0
i, j := 0, 0
ans := []int{}
for index := 0; index < total; index++ {
ans = append(ans, matrix[i][j])
vis[i][j] = true
next_x, next_y := i+dirs[dirindex][0], j+dirs[dirindex][1]
if next_x < 0 || next_x >= len(matrix) || next_y < 0 || next_y >= len(matrix[0]) || vis[next_x][next_y] {
dirindex = (dirindex + 1) % 4
}
i, j = i+dirs[dirindex][0], j+dirs[dirindex][1]
}
return ans
}