leetcode 59. Spiral Matrix II-螺旋矩阵|清晰思路

最新推荐文章于 2025-01-09 18:09:55 发布

Code_Granker

最新推荐文章于 2025-01-09 18:09:55 发布

阅读量1.8k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode Java Python

本文链接：https://blog.youkuaiyun.com/happyaaaaaaaaaaa/article/details/51542934

LeetCode 专栏收录该内容

167 篇文章

订阅专栏

本文介绍了一种生成螺旋矩阵的方法，并提供了Java与Python两种语言的实现。通过定义x、y坐标及移动步数等变量，实现了从外向内螺旋填充数字的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题链接：59. Spiral Matrix II

【思路】

遍历图解如上图所示，n 分为偶数和奇数两种情况。当 n 为偶数时，最后一次遍历刚好添加完毕；当 n 为奇数时，最后还需要进行一次添加操作⑤。另外，注意我所定义的 x 方向是竖向，y 方向是横向的。

public class Solution {
    public int[][] generateMatrix(int n) {
        int[][] res = new int[n][n];
        int num=1, d=n-1, x=(n-d)/2, y=(n-d)/2;
        while (d > 0) {
            for(int i=0; i<d; i++) res[x][y+i] = num++;  //①
            y += d;
            for(int i=0; i<d; i++) res[x+i][y] = num++;  //②
            x += d;
            for(int i=0; i<d; i++) res[x][y-i] = num++;  //③
            y = (n-d) / 2;
            for(int i=0; i<d; i++) res[x-i][y] = num++;  //④
            d -= 2;
            x = (n-d) / 2;
            y = (n-d) / 2;
        }
        if (d==0) res[x][y] = num;  //⑤
        return res;
    }
}

21 / 21 test cases passed. Runtime: 0 ms Your runtime beats 20.33% of javasubmissions.

Python 在语法上做了小小的优化，将4个循环合并：

class Solution(object):
    def generateMatrix(self, n):
        """
        :type n: int
        :rtype: List[List[int]]
        """
        res, step, x, y, num = [[0]*n for i in range(n)], n-1, 0, 0, 1
        dx, dy = [0,1,0,-1], [1,0,-1,0]
        while step > 0 :
            for i in range(4) :
                for j in range(step) :
                    res[x][y] = num
                    x, y, num = x+dx[i], y+dy[i], num+1
            step -= 2
            x,y = (n-step)/2, (n-step)/2
        if step == 0 : res[x][y] = num
        return res

21 / 21 test cases passed. Runtime: 60 ms Your runtime beats 22.66% of pythonsubmissions.