生成循环增加的二维数组

顺时针生成二维增加的数组

题目：给定一个正整数n,生成一个包含1 到n2所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
输入:3输出:[[1,2,3],[8,9,4],[7,6,5]

分析：螺旋生成时候，注意边界控制，全部遵循左闭右开的原则

#include <vector>
#include <iostream>
using namespace std;

class Solution{
public:
    vector<vector<int>> generateMatrix(int n){
        vector<vector<int>> res(n, vector<int>(n,0));
        int loop = n / 2; // 一共循环多少圈
        int offset = 1; // 不同圈结尾偏移量
        int startx = 0, starty = 0; //不同圈开始的位置
        int mid = n / 2;
        int i, j;
        int count = 1; // 输入元素
        while (loop--){
            i = startx; //开始一圈
            j = starty;
            for (j = starty; j < starty + n - offset; j++){
                res[startx][j] = count++;
            }
            for (i = startx; i < startx + n - offset; i++){
                res[i][j] = count++;
            }
            for (; j > starty; j--){
                res[i][j] = count++;
            }
            for (; i > startx; i--){
                res[i][j] = count++;
            }
            // 一圈结束 更新下一圈的开始位置和偏移量
            startx++;
            starty++;
            offset += 2;
        }
        if (n % 2){ // 如果n为奇数则最中间的单独赋值
            res[mid][mid] = count;
        }
        return res;
    }
};

int main(){
    Solution s;
    vector<vector<int>> res = s.generateMatrix(4);
    for (int i = 0; i < res.size(); i++){
        for (int j = 0; j < res[0].size(); j++){
            cout << res[i][j] << endl;
        }
    }
}