59. 螺旋矩阵Ⅱ
描述
给你一个正整数
n
,生成一个包含1
到n2
所有元素,且元素按顺时针顺序螺旋排列的n x n
正方形矩阵matrix
。
示例
示例1:
输入:n = 3 输出:[[1,2,3],[8,9,4],[7,6,5]]
示例2:
输入:n = 1 输出:[[1]]
提示
- 1 <= n <= 20
分析
- 首先,看提示:n的范围给定了1到20,所以不必在进行有效性的判定
- 此题和第54题是类似题,都是蛇形矩阵相关题
- 区别在于54题是取值,本题是放值,直接将54的代码进行修改就能满足需求
代码
func generateMatrix(n int) [][]int {
// 初始化
ret := make([][]int, n)
for i := 0; i < n; i++ {
ret[i] = make([]int, n)
}
// flag用来判断转方向后是否发生了移动,如果连续转了两次方向,那么说明遍历结束了
flag := true
// 定义方向
// right = {0,1},down = {1,0}, left = {0,-1}, up = {-1, 0}
// directs 是上述的方向集
directs := [][]int{
{
0, 1}, {
1, 0}, {
0, -1}, {
-1, 0}}
// aim表示要放的目标个数
aim := n * n
// 初始向右,定义方向索引p 为0
p := 0
direct := directs[p]
// x, y为遍历matrix的行列指针
x := 0
y := 0
ret[0][0] = 1
for i := 1; i < aim; {
// m, n移动
x += direct[0]
y += direct[1]
// 如果当前值为101,说明遍历过了,==row,==col 说明到达边界了,转方向
if x == n || y == n || x < 0 || y < 0 || ret[x][y] != 0 {
// 当前位置不对,退一步
x -= direct[0]
y -= direct[1]
// 退一步之后判断flag,为true那么转方向继续
if flag {
p++
p = p % 4
direct = directs[p]
flag = false
continue
}
// 如果flag 为false,那么就是连续发生了两次转向,未移动,那么说明遍历结束了
return ret
}
ret[x][y] = i+1
i++
// flag = true,表示有移动
flag = true
}
return ret
}
提交结果
蛇形矩阵可以使用不断缩小循环范围的方式去遍历,不设置方向,这样可以在空间上也得到节省。