和螺旋矩阵Ⅰ类似
把取出的操作换成写入就可以了
具体的分析可以看上一题
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
result = [[0 for i in range(n)] for j in range(n)]
T = 0
D = n - 1
L = 0
R = n - 1
i = 0
dir = 0
while L <= R and T <= D:
if dir == 0:
for j in range(L,R+1):
i = i + 1
result[T][j] = i
T = T + 1
dir = 1
if dir == 1:
for j in range(T,D+1):
i = i + 1
result[j][R] = i
R = R - 1
dir = 2
if dir == 2:
for j in range(R,L-1,-1):
i = i + 1
result[D][j] = i
D = D - 1
dir = 3
if dir == 3:
for j in range(D,T-1,-1):
i = i + 1
result[j][L] = i
L = L + 1
dir = 0
return result