59.螺旋矩阵II

题目描述

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

示例 1：

输入：n = 3
输出：[[1,2,3],[8,9,4],[7,6,5]]

示例 2：

输入：n = 1
输出：[[1]]

提示：

• 1 <= n <= 20

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        // 初始化一个n*n的矩阵，所有元素值为0
        vector<vector<int>> num(n, vector<int>(n, 0));
        int k = 1; // 这是我们将要填充到矩阵中的数字

        // 我们一圈圈地填充数字，每次填充一圈
        for(int huan = 0; huan < n / 2; huan++) {
            // 每一圈的起始坐标都是(huan, huan)
            int i = huan, j = huan;

            // 从左到右填充上边，注意右边界是开区间，所以是n-1-huan
            for(j; j < n - 1 - huan; j++) num[i][j] = k++; 

            // 从上到下填充右边
            for(i; i < n - 1 - huan; i++) num[i][j] = k++;

            // 从右到左填充下边
            for(j; j > huan; j--) num[i][j] = k++;

            // 从下到上填充左边
            for(i; i > huan; i--) num[i][j] = k++;
        }

        // 如果矩阵的大小是奇数，那么中间的元素还没有被填充
        // 这里将它填充上
        if(n % 2 != 0) num[n / 2][n / 2] = k;

        // 返回填充好的矩阵
        return num;
    }
};