题目:
给你一个正整数 n ,生成一个包含 1 到 n的平方的 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例:
思路:
类似于上一个螺旋矩阵的题目,给定一个矩阵,然后按照顺时针旋转输出。这里由给定的正整数n生成n的平方个元素,然后顺时针旋转排列构成矩阵,首先,初始化全0矩阵,设置左右上下边界,第一个放进矩阵的值1,然后顺时针遍历位置,放进元素,比如,第一行的左边界到右边界,依次放置元素,每放一个,k+1,这一循环结束后,上边界下移一行,然后从上边界往下边界遍历,循环结束,右边界向左移动一位,后边再向左和向上遍历,和前面类似,直到全部遍历完,返回矩阵。
class Solution(object):
def generateMatrix(self, n):
"""
:type n: int
:rtype: List[List[int]]
"""
ans = [[0]* n for _ in range(n)]
left ,right ,up , down = 0,n-1,0,n-1
k = 1
while left <= right and up <= down:
for i in range(left,right+1):
ans[up][i] = k
k += 1
up += 1
for i in range(up,down+1):
ans[i][right] = k#这里注意要交换i和right的位置,right相当于y保持不变
k += 1
right -= 1
for i in range(right,left-1,-1):
ans[down][i] = k
k += 1
down -= 1
for i in range(down,up-1,-1):
ans[i][left] = k
k += 1
left += 1
return ans