欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

59. 螺旋矩阵 II

程序员文章站 2022-07-12 08:53:53
...

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

示例:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

此题就是模拟
生成一个 n×n 空矩阵 res,随后模拟整个向内环绕的填入过程:
定义当前左右上下边界 left,right,top,bottom,初始值count = 1,迭代终止值n * n;
当 num <= nn 时,始终按照 从左到右 从上到下 从右到左 从下到上 填入顺序循环,每次填入后:
执行 count += 1:得到下一个需要填入的数字;
更新边界:例如从左到右填完后,上边界 top+= 1,相当于上边界向内缩 1。
使用num <= n
n而不是left < right || top< bottom作为迭代条件,是为了解决当n为奇数时,矩阵中心数字无法在迭代过程中被填充的问题。
最终返回 res 即可。

代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int> > res(n,vector<int>(n,0));
        int count=1;
        int left=0;
        int right=n-1;
        int top=0;
        int bottom=n-1;
        while(count<=n*n)
        {
            for(int i=left;i<=right;i++)
                res[top][i]=count++;
            top++;
            for(int i=top;i<=bottom;i++)
                res[i][right]=count++;
            right--;
            for(int i=right;i>=left;i--)
                res[bottom][i]=count++;
            bottom--;
            for(int i=bottom;i>=top;i--)
                res[i][left]=count++;
            left++;
        }
        return res;
    }
};