力扣刷题(python)50天——第二十天：螺旋矩阵 II

力扣刷题(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

设置二维数组后调用位置真的比之后慢慢添加要快得多。。

执行结果

心得：

设置二维数组主要可以用来设定好位置参数方便调用。