Leetcode 59. 螺旋矩阵 II

题目叙述

题解代码：

螺旋矩阵虽然不涉及具体的算法，但是模拟的过程却十分考验代码的能力，一不小心就容易将其写乱。

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n,0));
        int startx = 0, starty = 0; //定义每循环一个圈的起始位置
        int loop = n / 2; //定义每个圈要循环几次,如果是奇数，例如3，则中心只有一个数，单独处理即可，只用循环3/2=1次
        int mid = n / 2; //矩阵的中心位置
        int count = 1; //用来依次给矩阵的每一个位置赋值
        int offset = 1; //每循环一圈，借助offset控制每条边的遍历长度
        int i, j;
        while(loop--)
        {
            i = startx;
            j = starty;
            //四个for循环模拟旋转一圈
            
            //模拟填充上行从左到右（左闭右开区间）
            for(j = starty; j < starty + n - offset; j++){
                res[i][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++;
            }

            //循环下一圈时，起始位置各加1，如第一圈为(0,0)，则第二圈起始位置为(1,1)
            startx++;
            starty++;
            
            //借助offset控制每一圈的每条边的遍历长度
            offset += 2;

        }
        if(n % 2) //如果n为奇数，则其最内圈只有一个数字，单独赋值即可
        {
            res[mid][mid] = count;
        }
        return res;

    }

};

 提交结果