【LeetCode】59. 螺旋矩阵 II

最新推荐文章于 2024-07-30 15:23:09 发布

原创最新推荐文章于 2024-07-30 15:23:09 发布 · 566 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

LeetCode 专栏收录该内容

32 篇文章

订阅专栏

本文介绍了一种算法，用于生成一个正方形矩阵，该矩阵包含从1到n^2的所有元素，这些元素按照顺时针螺旋顺序排列。示例中，当输入为3时，输出的矩阵为[[1,2,3],[8,9,4],[7,6,5]]。算法通过迭代过程填充矩阵的每一层，直到完成整个矩阵。

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

示例:

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

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        int res[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; 
        
        vector<vector<int>> a;
        for(int i=0; i<n; i++)
        {
            vector<int> a1;
            for(int j=0; j<n; j++)
            {
                a1.push_back(res[i][j]);
            }
            a.push_back(a1);
        }
        return(a);
    }
};