先画一下n为不同值时的案例:
n=3时:
n=4时:
此类题目整体较为简单,难点在于:保持四个方向的边界条件一致,全部使用左闭右开。
1.首先需要两个变量来作为输出数组的下标
第一圈遍历时:
从左往右:x值不变是starts,y值增加,范围是[starty,n-offset]
从上往下:y值不变是n-offset,x值增加,范围是[startx,n-offset]
从右往左:x值不变是n-offset,y值减小,范围是[n-offset,0,-1]
从下往上:y值不变是starty,x值减小,范围是[n-offset,0,-1]
第二圈遍历时:x值加一,y值加一,offset加一
从左往右:x值不变,y值增加,范围是[new_x,n-offset]
2.可以看出在遍历过程中需要一个偏移量offset,且偏移量等于转圈次数=n//2,比如n=3时,转1圈,n=4时转两圈,n=5时转两圈
3.遍历结束的条件:offset=n//2,loop=n//2
4.n为奇数和偶数时处理方法略有不同,n为奇数时会剩下最后一个中心元素,可以直接赋值 res[n//2][n//2]=n**2
有了以上思路就可以开始写代码了:
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
#生成一个n*n的空数组
matrix=[[0]*n for _ in range(n)]
cnt=1
offset=1
startx,starty=0,0
loop=n//2
for offset in range(1,loop+1):
#从左往右遍历,y值增加,x值不变
for i in range(starty,n-offset):
matrix[startx][i]=cnt
cnt+=1
#从上往下遍历,x值增加,y值不变
for i in range(startx,n-offset):
matrix[i][n-offset]=cnt
cnt+=1
#从右往左遍历
for i in range(n-offset,starty,-1):
matrix[n-offset][i]=cnt
cnt+=1
#从下往上遍历
for i in range(n-offset,startx,-1):
matrix[i][starty]=cnt
cnt+=1
#一圈结束后
startx+=1
starty+=1
#判断
if matrix[loop][loop]==0:
matrix[loop][loop]=n*n
return matrix