leetcode59题 螺旋矩阵 II（C++）

本文参考卡哥的代码随想录，加上了一点自己的理解，感谢Carl的分享。
代码随想录
视频讲解

一. 题目

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

示例:

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

二. 解题思路

本题主要考查对边界判断，代码思路并不难，就是从左上角起点开始，顺时针，从外到内遍历数组。
需要思考以下几个问题：

1. 题目给定任意整数n，肯定是以一圈一个循环，应该遍历几圈？n为奇数和偶数的区别。
2. 遍历每一圈的时候怎么找到一种简单清晰方式？

如图所示：

• 循环圈数为n / 2圈，当n为奇数时，最中间的元素单独赋值为n*n，即nums[n/2][n/2]== n*n 。
• 每一条边的循环采用相同的遍历方式，从第一个元素开始到倒数第二个元素结束。

三. 代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0));  // 使用vector定义一个二维数组
        int startx = 0, starty = 0;  // 用来定义每圈循环的起始位置
        int loop = n / 2;  // 为了生成目标数组所需几圈循环
        int mid = n / 2;  // 当n为奇数时，最中间元素的下标
        int count = 1;  // 用来给数组中的元素赋值
        int offset = 1;  // 用来计算每条边的终止位置
        int i , j;
        while ( loop--) {
            i = startx;
            for (j = starty; j < n - offset; j++) {  //第一条边 
                res[i][j] = count; // i = startx
                count++;
            }
            for (i; i < n - offset; i++) {  // 第二条边
                res[i][j] = count;  // j = n - offset
                count++;
            }
            for (; j > starty; j--) {  // 第三条边
                res[i][j] = count;  // i = n - offset
                count++;
            }
            for (;i > startx; i--) {  // 第四条边
                res[i][j] = count;  // j = starty
                count++;
            }
            startx++;  // 下一圈循环的起点
            starty++;  // 下一圈循环的起点
            offset++;  // 用来计算下一圈循环中每条边的终点
        }
        if (n % 2) {  // n为基数，赋值最中间的元素
            res[mid][mid] = count;
        }
        return res;
    }
};