leetcode—python 50天刷题 第20题 螺旋矩阵ll

题目描述

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

示例:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解答过程

class Solution(object):
    def generateMatrix(self, n):
        """
        :type n: int
        :rtype: List[List[int]]
        """
        left=0
        right=n-1
        up=0
        down=n-1
        num=1
        final = [[0 for _ in range(n)] for _ in range(n)]
        while num<=n*n:
            for i in range(left,right+1):
                final[up][i]=num
                num+=1
            if num>n*n:
                break
            up=up+1
            for i in range(up,down+1):
                final[i][right]=num
                num+=1
            if num>n*n:
                break
            right=right-1
            for i in range(left,right+1):
                final[down][right+left-i]=num
                num+=1
            if num>n*n:
                break
            down=down-1
            for i in range(up,down+1):
                final[down+up-i][left]=num
                num+=1
            if num>n*n:
                break
            left+=1
        return final

执行结果

执行用时 :24 ms, 在所有 Python 提交中击败了73.45% 的用户
内存消耗 :11.7 MB, 在所有 Python 提交中击败了24.91%的用户