LeetCode 59. 螺旋矩阵 II（Spiral Matrix II）

题目描述

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

示例:

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

解题思路

和LeetCode54.螺旋矩阵 的思想差不多，定义左上角的行索引，然后依次从左至右、从上至下、从右至左、从下至上遍历，注意起始索引和终止索引。

代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n, vector<int>(n, 0));
        int idx = 0, num = 1;
        while(idx * 2 < n){
            for(int col = idx; col < n - idx; col++)
                res[idx][col] = num++;
            for(int row = idx + 1; row < n - idx; row++)
                res[row][n - idx - 1] = num++;
            for(int col = n - idx - 2; col >= idx; col--)
                res[n - idx - 1][col] = num++;
            for(int row = n - idx - 2; row > idx; row--)
                res[row][idx] = num++;
            idx++;
        }
        return res;
    }
};