螺旋矩阵II-力扣59题

题目描述

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

思路

输入为一个整数n，输出则为一个二维数组，行列大小均为n。
然后，顺时针遍历矩阵并填充数据，即左-右，上-下，右-左，下-上，为了填充方便，这里采用左闭右开的原则填充数据，并对n为奇数时，进行特别处理，当mid等于n对2取余的结果时，[mid][mid]即为中心点的坐标。
每一次循环需要控制边界，经过第一次循环后，要进入内循环，这里设置一个初始值为1的offset控制边界条件，每经过一轮循环，offset的值加一，即边界向内缩一圈。

C++代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        int x = 0, y = 0; //循环的起始位置
        int loop = n / 2; //循环的圈数
        int mid = n / 2; //如果n为奇数，[mid][mid]即为最后中心位置的数
        int offset = 1; //控制内循环边界
        int count = 1; //用于矩阵赋值
        int i, j;
        vector<vector<int>> pr(n, vector<int>(n, 0)); //定义返回的数组
        while(loop --){
            j = y, i = x;
            //顺时针，先j后x, 循环方式采用左闭右开
            for(j; j < n - offset; j++) pr[i][j] = count++;
            for(i; i < n - offset; i++) pr[i][j] = count++;
            for(j; j > y; j--) pr[i][j] = count++;
            for(i; i > x; i--) pr[i][j] = count++;
            y++, x++, offset++;
        }
        if(n % 2) pr[mid][mid] = count; //处理n为奇数时，中心位置数据
        return pr;
    }
};