leetcode_59. 螺旋矩阵 II

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

给你一个正整数 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认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
个人GitHub网址为:https://github.com/nickhuang1996
相关推荐
©️2020 CSDN 皮肤主题: 博客之星2020 设计师:CY__ 返回首页
实付 29.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值