59.螺旋矩阵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
这道题其实就是模拟顺时针画矩阵的过程:
填充上行从左到右
填充右列从上到下
填充下行从右到左
填充左列从下到上
值得注意的是区间需要统一,要么全部左开右闭,要么全部左闭右开,不然会发生错误
这是运行代码
package main
import "fmt"
func main() {
n := 3
res := generateMatrix(n)
for _, re := range res {
for _, i2 := range re {
fmt.Printf("%d", i2)
}
fmt.Println()
}
}
func generateMatrix(n int) [][]int {
res := make([][]int, n)
for i := 0; i < n; i++ {
res[i] = make([]int, n)
}
left, right := 0, n-1 //left 为左边界,right为右边界
top, bottom := 0, n-1 //top 为上边界,bottom为下边界
count := 1 //count为计数
for count <= n*n {
//最上边 从左到右
for i := left; i <= right; i++ {
res[top][i] = count
count++
}
top++
//最右边 从上到下
for i := top; i <= bottom; i++ {
res[i][right] = count
count++
}
right--
//最下边 从右到左
for i := right; i >= left; i-- {
res[bottom][i] = count
count++
}
bottom--
//最左边 从下到上
for i := bottom; i >= top; i-- {
res[i][left] = count
count++
}
left++
}
return res
}