Leetcode 59. 螺旋矩阵 II

Leetcode 59. 螺旋矩阵 II

题目

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

思路

  这道题只要控制好边界问题即可，遵循左闭右开区间，从左往右，从上往下，从右往左，从下往上开始遍历，循环的圈数是n / 2,如果n是奇数，那么，中间还有一个元素在循环结束之后进行填充。

代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n,vector<int>(n,0));// 定义一个二维数组 全部初始化为0
        int startx = 0,starty = 0;// 定义每循环一个圈的起始位置
        int loop = n / 2;  // 每个圈循环几次，如果n时奇数3，那么loop = 1 只是循环一圈
        int mid = n / 2;// 矩阵中间的位置  如果n是3, 中间的位置就是(1,1)
        int count = 1;// 用来各一个矩阵中每一个空格赋值
        int offset = 1;// 需要控制每一条边遍历的长度，每次循环右边界收缩一位
        int i,j;

        // loop 是转几圈的问题 n = 4 转两圈   n = 3 转一圈  中间元素单独赋值
        while(loop --)
        {
            i = startx;
            j = starty;
            // 下面开始的四个for就是模拟转了一圈
            // 模拟填充上面的行 从左到右 （左闭右开）

            // 不能去掉starty  因为每一次循环至少会遍历starty这个元素
            for(j = starty; j < starty + n - offset; j++)
            {
                // j 表示的是列   j < n - offset 让j不能遍历最后一列的元素
                // count表示赋值元素
                res[startx][j] = count;
                count++;
            }
            // j这时候已经遍历到 倒数第二列元素
            // 模拟从上到下填充  (左闭右开)
            for(i = startx; i < startx + n - offset; i++)
            {
                // i表示遍历行  从上到下
                res[i][j] = count;
                count++;
            }
            // 然后开始从右往左进行遍历 进行填充元素  （左闭右开）
            // j > starty 表示第一列元素不进行填充
            for(; j > starty; j--)
            {
                res[i][j] = count++;
            }
            // 然后从下往上进行遍历 进行填充元素(左闭右开)
            for(;i > startx; i--)
            {
                res[i][j] = count++;
            }
            // 第二圈开始的时候  那么起始位置都改变了
            startx++;//  从（0，0）-> (1,1)
            starty++;

            // offset用于控制每一圈中每一条边遍历的长度
            offset += 2;// 自己举例 测试一下
        }
         // 如果n是奇数  则需要单独给矩阵最中间的位置赋值
        if(n % 2)
        {
            res[mid][mid] = count;// 中间单独的元素
        }
        return res;
    }
};