算法（C++ - 技术栈

算法（C++

蟹老板爱钱2024-05-07 9:55

题目：螺旋矩阵（59. 螺旋矩阵 II - 力扣（LeetCode））

给你一个正整数 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

题解:

cpp 复制代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0));
        int starti=0,startj=0;
        int count=1;
        int circle=n/2;
        int endd=n/2;
        int flag=1;
        int i,j;
        while(circle--){
            i=starti;
            j=startj;
            for(;j<n-flag;j++){
                res[i][j]=count++;
            }
            for(;i<n-flag;i++){
                res[i][j]=count++;
            }
            for(;j>startj;j--){
                res[i][j]=count++;
            }
            for(;i>starti;i--){
                res[i][j]=count++;
            }
            flag++;
            starti++;
            startj++;
        }
        if(n%2!=0){
            res[endd][endd]=count;
        }
        return res;
    }
};

理解：

要分好四个步骤