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, ));
         int idx = , num = ;
         while(idx *  < n){
             for(int col = idx; col < n - idx; col++)
                 res[idx][col] = num++;
             for(int row = idx + ; row < n - idx; row++)
                 res[row][n - idx - ] = num++;
             for(int col = n - idx - ; col >= idx; col--)
                 res[n - idx - ][col] = num++;
             for(int row = n - idx - ; row > idx; row--)
                 res[row][idx] = num++;
             idx++;
         }
         return res;
     }
 };