目录

一、题目内容

二、解题思路

三、代码

一、题目内容

给你一个正整数 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

二、解题思路

和leetcode_54. 螺旋矩阵类似，只需要按照顺序逐次赋值即可。

三、代码

  

   

    

     

    
    

     

      class Solution:
     
    
   

    

     

    
    

     
 def generateMatrix(self, n: int):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       up, down, left, right = 0, n - 1, 0, n - 1  # 上下左右
     
    
   

    

     

    
    

     

       martix = [[0 for _ in range(n)] for _ in range(n)]
     
    
   

    

     

    
    

     

       num = 1
     
    
   

    

     

    
    

     
 while True:
     
    
   

    

     

    
    

     
 # →
     
    
   

    

     

    
    

     
 for i in range(left, right + 1):
     
    
   

    

     

    
    

     

       martix[up][i] = num
     
    
   

    

     

    
    

     

       num += 1
     
    
   

    

     

    
    

     

       up += 1
     
    
   

    

     

    
    

     
 if up > down:
     
    
   

    

     

    
    

     
 break
     
    
   

    

     

    
    

     
 # ↓
     
    
   

    

     

    
    

     
 for i in range(up, down + 1):
     
    
   

    

     

    
    

     

       martix[i][right] = num
     
    
   

    

     

    
    

     

       num += 1
     
    
   

    

     

    
    

     

       right -= 1
     
    
   

    

     

    
    

     
 if right < left:
     
    
   

    

     

    
    

     
 break
     
    
   

    

     

    
    

     
 # ←
     
    
   

    

     

    
    

     
 for i in range(right, left - 1, -1):
     
    
   

    

     

    
    

     

       martix[down][i] = num
     
    
   

    

     

    
    

     

       num += 1
     
    
   

    

     

    
    

     

       down -= 1
     
    
   

    

     

    
    

     
 if down < up:
     
    
   

    

     

    
    

     
 break
     
    
   

    

     

    
    

     
 # ↑
     
    
   

    

     

    
    

     
 for i in range(down, up - 1, -1):
     
    
   

    

     

    
    

     

       martix[i][left] = num
     
    
   

    

     

    
    

     

       num += 1
     
    
   

    

     

    
    

     

       left += 1
     
    
   

    

     

    
    

     
 if left > right:
     
    
   

    

     

    
    

     
 break
     
    
   

    

     

    
    

     
 return martix
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      if __name__ == '__main__':
     
    
   

    

     

    
    

     
 # n = 3
     
    
   

    

     

    
    

     

       n = 1
     
    
   

    

     

    
    

     

       s = Solution()
     
    
   

    

     

    
    

     

       ans = s.generateMatrix(n)
     
    
   

    

     

    
    

     

       print(ans)
     
    
  
 

文章来源: nickhuang1996.blog.csdn.net，作者：悲恋花丶无心之人，版权归原作者所有，如需转载，请联系作者。

原文链接：nickhuang1996.blog.csdn.net/article/details/114867993