螺旋矩阵 II
给你一个正整数 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*n的数组,然后给据规则生成就行了(有的是按环数具体可看代码随想录,我这个是按方向一个个生成),一定要注意,代码如下
go
var (
Dir = [4][2]int{{0, 1}, {1, 0}, {0, -1}, {-1, 0}}
)
func generateMatrix(n int) [][]int {
numss := make([][]int, n)
for i := 0; i < n; i++ {
numss[i] = make([]int, n)
}
count := 1
// 从 [1][n-1]处开始
x, y := 0, 0
// guaiwan每循环两次就要再减一
time := 1
// 减过的次数
subnum := 0
guaiwan := n - subnum
dir := 0
for count <= n*n {
numss[x][y] = count
count++
guaiwan--
if guaiwan == 0 {
dir++
if dir == 4 {
dir = 0
}
time--
if time == 0 {
time = 2
subnum++
}
guaiwan = n - subnum
}
x = x + Dir[dir][0]
y = y + Dir[dir][1]
}
return numss
}
虽然只用了一个for循环,但是时间复杂度为O(n²)。
而且用了很多控制方向的变量,空间复杂度较大(我希望有更好的解法)。
注意这种循环处理不好就会报错,在设置条件时一定要细心:
方法二:
如果您有更好的解法,别忘了@博主去学习啊,感谢了