力扣刷题(python)50天——第二十天:螺旋矩阵 II
题目描述
给定一个包含 m x n 个元素的矩阵(m 行, n 列),请按照顺时针螺旋顺序,返回矩阵中的所有元素。
示例 1:
输入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]
示例 2:
输入:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法
这道题和上一题很类似,如果用上一题的方法,之前是由外向内剖,那么这次也可以用递归由外到内。但我想换换方法,想着能不能模拟加入的过程
参考:
https://leetcode-cn.com/problems/spiral-matrix-ii/solution/spiral-matrix-ii-mo-ni-fa-she-ding-bian-jie-qing-x/
采用模拟过程,实际操作中的错误答案:
class Solution(object): def generateMatrix(self, n): """ :type n: int :rtype: List[List[int]] """ ans=[] for i in range(n): ans.append([]) l=0 r=n-1 t=0 b=n-1 num=1 while num<n*n: for i in range(l,r+1): ans[t].append(num) num+=1 t+=1 for i in range(t,b+1): ans[i].insert(-1,num) num+=1 r-=1 for i in range(r,l-1,-1): ans[b].append(num) num+=1 ans[b].reverse() b-=1 for i in range(b,t-1,-1): ans[i].insert(0,num) num+=1 l-=1 if num==n*n: ans[n/2].insert(n/2,n*n) return ans改进后,我发现其实可以不用只设置一个含n个空列表的矩阵,可以直接设置一个n*n的矩阵出来,会非常方便调用我们的某一项,而我的方法就是此处出了问题。虽然之前想过,但不知道为什么觉得没必要?。。。。
解答
class Solution: def generateansrix(self, n): l, r, t, b = 0, n - 1, 0, n - 1 ans = [[0 for _ in range(n)] for _ in range(n)] num, tar = 1, n * n while num <= tar: for i in range(l, r + 1): ans[t][i] = num num += 1 t += 1 for i in range(t, b + 1): ans[i][r] = num num += 1 r -= 1 for i in range(r, l - 1, -1): ans[b][i] = num num += 1 b -= 1 for i in range(b, t - 1, -1): ans[i][l] = num num += 1 l += 1 return ans设置二维数组后调用位置真的比之后慢慢添加要快得多。。
执行结果
心得:
设置二维数组主要可以用来设定好位置参数方便调用。