螺旋矩阵代码

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

示例 1：
输入：n = 3
输出：[[1,2,3],[8,9,4],[7,6,5]]

 

代码：

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        int loop=n/2;
        vector<vector<int>> result(n, vector<int>(n, 0));
        int count=1;
        int startrow,startcolumn=0;
        int offset=1;
        int i,j;
        
        while(loop--){
            int row=startrow; 
            int column=startcolumn;

            for(i=column;i<n-offset;i++){
                result[row][i]=count++;   //从左到右，i列到最大值
            }
            for(j=row;j<n-offset;j++){
                result[j][i]=count++;//从上到下，j行到最大值
            }
            for(;i>column;i--){
                result[j][i]=count++;//从右到左，i列到最小值
            }
            for(;j>row;j--){
                result[j][i]=count++;//从下到上，j行到最小值
            }
            startrow++;
            startcolumn++;
            offset++;
        }
        if(n%2){
            result[(n/2)][(n/2)]=count;
        }

        return result;
    }
};

可以想成从外圈往里圈写数，总共有n/2圈。每一圈的四边都分别用for循环写入数值。每条边写数时注意最后一个的数值是在下一条边开始时写入的，每一圈结束后需要缩进一圈。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/spiral-matrix-ii