leetcode做题笔记 59螺旋矩阵II python

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

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

 leetcode做题笔记 59螺旋矩阵II python_第1张图片

 这道题想了很久都没有思路,最后还是参考了官方答案看了题解才写出来:

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        left, right, top, bottom = 0, n-1, 0, n-1
        matric = [[0 for _ in range(n)]for _ in range(n)]
        count, target = 1, n*n
        while count <= target:
            for i in range(left, right + 1): #从左到右
                matric[top][i] = count
                count += 1
            top += 1
            for i in range(top, bottom + 1): #从上到下
                matric[i][right] = count
                count += 1
            right -= 1
            for i in range(right, left - 1, -1): #从右到左
                matric[bottom][i] = count
                count += 1
            bottom -= 1
            for i in range(bottom, top - 1, -1): #从下到上
                matric[i][left] = count
                count += 1
            left += 1
        return matric

你可能感兴趣的:(学习笔记,python)