leetcode_59. 螺旋矩阵 II

原创
03/16 17:41
阅读数 18

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

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

示例 1:

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

示例 2:

输入:n = 1
输出:[[1]]

提示:

1 <= n <= 20

二、解题思路

leetcode_54. 螺旋矩阵类似,只需要按照顺序逐次赋值即可。

三、代码

class Solution:
    def generateMatrix(self, n: int):

        up, down, left, right = 0, n - 1, 0, n - 1  # 上下左右
        martix = [[0 for _ in range(n)] for _ in range(n)]
        num = 1
        while True:
            # →
            for i in range(left, right + 1):
                martix[up][i] = num
                num += 1
            up += 1
            if up > down:
                break
            # ↓
            for i in range(up, down + 1):
                martix[i][right] = num
                num += 1
            right -= 1
            if right < left:
                break
            # ←
            for i in range(right, left - 1, -1):
                martix[down][i] = num
                num += 1
            down -= 1
            if down < up:
                break
            # ↑
            for i in range(down, up - 1, -1):
                martix[i][left] = num
                num += 1
            left += 1
            if left > right:
                break
        return martix


if __name__ == '__main__':
    # n = 3
    n = 1
    s = Solution()
    ans = s.generateMatrix(n)
    print(ans)

本文同步分享在 博客“悲恋花丶无心之人”(CSDN)。
如有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一起分享。

展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部