刷题_day4:螺旋矩阵

题目描述：

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

示例:

输入: 3 输出: [ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]

题目没有过多的算法，注重思维层面。而求解本题要坚持循环不变量原则。

模拟顺时针画矩阵的过程:

填充上行从左到右
填充右列从上到下
填充下行从右到左
填充左列从下到上
由外向内一圈一圈这么画下去。每一圈要画四条边，每一条边都要坚持左闭右开原则。

代码如下：

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0));
        int startx = 0;  // 定义每循环一个圈的起始位置
        int starty = 0;
        int time = n/2;  // 每个圈循环几次，例如n为奇数3，那么time = 1 只是循环一圈，矩阵中间的值需要单独处理
        int mid = n/2;
        int count = 1;
        int offset = 1;  // 需要控制每一条边遍历的长度，每次循环右边界收缩一位
        int i, j;
        while(time--)
        {
            i = startx;
            j = starty;
            for(j = starty;j < n-offset;j++)  //→
            {
                res[startx][j] = count++;
            }
            for(i = startx;i < n-offset;i++)  //↓
            {
                res[i][j] = count++;
            }
            for(;j > starty;j--)  //←
            {
                res[i][j] = count++;
            }
            for(;i > startx;i--)  //↑
            {
                res[i][j] = count++;
            }
            startx++; //第二圈开始，其实位置要加一 从（0,0）变为（1,1）
            starty++;
            offset++;          
        }
        if(n % 2)
        {
            res[mid][mid] = count;
        }
        return res;
    }
};