LeetCode59. 螺旋矩阵 II

代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        
        int startx = 0,starty = 0; //每个loop起始位置
        int loop = n / 2; //循环次数，n为奇数要另外考虑中心位置
        int count = 1; //矩阵元素，从1—n2
        int offset = 1; //控制每层循环的边长变化

        //用vector生成二维数组
        vector<vector<int>> res (n);
        for(int i = 0;i < n;i++)
            res[i].resize(n);
        
        while(loop--){
            int i = startx;
            int j = starty;

            //上边：从左到右
            for(; j < starty + n - offset;j++)
                res[startx][j] = count++;
            //右边：从上到下
            for(; 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 != 0) res[n/2][n/2] = n*n;
        
        return res;
    }
};

将整个矩阵的起点设为坐上角，该点为坐标原点，然后水平向右为x轴，竖直向下为y轴。一定要按照一定的顺序去写（顺时针），并且要遵循循环不变量原则，每一条边的遍历范围为左闭右开。