数组_螺旋矩阵II

一、leetcode-59

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

样例输入：n = 3

样例输出：[[1,2,3],[8,9,4],[7,6,5]]

解释:子数组 [4,3] 是该条件下的长度最小的子数组。

二、题解

1.代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> ans(n,vector<int>(n));
        int startx = 0, starty = 0; // 定义每循环一个圈的起始位置
        int loop = n / 2; // 每个圈循环几次，例如n为奇数3，那么loop = 1 只是循环一圈，矩阵中间的值需要单独处理
        int mid = n / 2; // 矩阵中间的位置，例如：n为3， 中间的位置就是(1，1)，n为5，中间位置为(2, 2)
        int count = 1; // 用来给矩阵中每一个空格赋值
        int offset = 1; // 需要控制每一条边遍历的长度，每次循环右边界收缩一位
        int i,j;
        while(loop--){
            i=startx,j=starty;
            // 下面开始的四个for就是模拟转了一圈
            // 模拟填充上行从左到右(左闭右开)
            for(;j<n-offset;j++){
                ans[i][j]=count++;
            }
            // 模拟填充右列从上到下(左闭右开)
            for(;i<n-offset;i++){
                ans[i][j]=count++;
            }
            // 模拟填充下行从右到左(左闭右开)
            for(;j>starty;j--){
                ans[i][j]=count++;
            }
            // 模拟填充左列从下到上(左闭右开)
            for(;i>startx;i--){
                ans[i][j]=count++;
            }
            // 第二圈开始的时候，起始位置要各自加1， 例如：第一圈起始位置是(0, 0)，第二圈起始位置是(1, 1)
            startx++;
            starty++;
            // offset 控制每一圈里每一条边遍历需要减去的长度
            offset += 1;
        }
        // 如果n为奇数的话，需要单独给矩阵最中间的位置赋值
        if (n%2) {
            ans[mid][mid]=count;
        }
        return ans;
    }
};

2.思考

这道题目可以说在面试中出现频率较高的题目，本题并不涉及到什么算法，就是模拟过程，但却十分考察对代码的掌控能力。

要如何画出这个螺旋排列的正方形矩阵呢？

还记得这篇文章数组：二分法，提到如果要写出正确的二分法一定要坚持循环不变量原则。

而求解本题依然是要坚持循环不变量原则。

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

填充上行从左到右
填充右列从上到下
填充下行从右到左
填充左列从下到上
由外向内一圈一圈这么画下去。

每次填充使用左闭右开的，这样可以分成四个一样的形状，便于循环。