数组篇 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
这道题并不涉及到什么算法,就是考察对代码的掌控能力。
这道题还是顺时针顺序填充:
从左到右
从上到下
从右到左
从下到上
但是要注意别重复或者漏掉单元格,还是要找一个不漏不重的规则,即for循环控制条件,这里有两种方法:
方法一:
参考https://github.com/youngyangyang04/leetcode-master/blob/master/problems/0059.螺旋矩阵II.md
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
matrix=[[0]*n for _ in range(n)] #将一行复制为n行,n*n矩阵
left,right,up,down=0,n-1,0,n-1 #左(0列)右(n-1列)上(0行)下(n-1行)四条边
number=1
while left<right and up<down: #结束条件不能设为num<=n*n
for i in range(left,right): #注意左闭右开
matrix[up][i]=number
number+=1
for j in range(up,down):
matrix[j][right]=number
number+=1
for x in range(right,left,-1):
matrix[down][x]=number
number+=1
for y in range(down,up,-1):
matrix[y][left]=number
number+=1
left+=1
right-=1
up+=1
down-=1
if n%2: #奇数情况正中心不能在循环中覆盖到 需要另外考虑
matrix[n//2][n//2]=number
return matrix
方法二:
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
matrix = [[0]*n for _ in range(n)] #将一行复制为n行
left, right, up, down = 0, n-1, 0, n-1
number = 1
while number <= n * n:
#从左到右填充上面
for x in range(left,right+1):
matrix[up][x]=number
number +=1
up+=1
#从上到下填充右边
for y in range(up,down+1):
matrix[y][right] = number
number+=1
right-=1
#从右到左填充下边
for x in range(right,left-1,-1):
matrix[down][x]=number
number+=1
down-=1
#从下到上填充左边
for y in range(down,up-1,-1):
matrix[y][left]=number
number+=1
left+=1
return matrix
这个方法比较好理解,但是写的时候注意for循环的范围。
时间空间复杂度都是0(n2)