leetcode刷题笔记-05-螺旋矩阵 II

mac2024-03-23  25

螺旋矩阵 II

题目要求我的解题思路我的思路缺点

题目要求

给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。

示例: 输入: 3 输出: [ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]

我的解题思路

生成一个 n×n 空矩阵 mat,随后模拟整个向内环绕的填入过程:

1.首先定义当前矩阵左右上下边界 l,r,t,b,初始值 num = 1,迭代终止值 tar = n * n; 2.当 num <= tar 时,始终按照 从左到右 从上到下 从右到左 从下到上 填入顺序循环,每次填入后: 2.1执行 num += 1:得到下一个需要填入的数字; 2.2更新边界:例如从左到右填完后,上边界 t += 1,相当于上边界向内缩 1。 ps:1.使用num <= tar而不是l < r || t < b作为迭代条件,是为了解决当n为奇数时,矩阵中心数字无法在迭代过程中被填充的问题。 2.注意第一个range中起始是l左边界,结束是r+1

最终返回 mat 即可。

class Solution: def generateMatrix(self, n: int) -> List[List[int]]: l,t,r,b = 0,0,n-1,n-1 num,target = 1,n * n mat = [[0 for _ in range(n)] for _ in range(n)] while num <= target: for i in range(l,r+1): mat[t][i] = num num += 1 t += 1 for i in range(t,b+1): mat[i][r] = num num += 1 r -= 1 for i in range(r,l-1,-1): mat[b][i] = num num += 1 b -= 1 for i in range(b,t-1,-1): mat[i][l] = num num += 1 l += 1 return mat

我的思路缺点

执行用时 :32 ms, 在所有 python3 提交中击败了99.20% 的用户 内存消耗 :13.8 MB, 在所有 python3 提交中击败了5.14%的用户

最新回复(0)