Description

Given a positive integer n, generate a square matrix filled with elements from 1 to n2 in spiral order.

Example:

Input:

3

Output:

[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

分析

题目的意思是：实现一个n*n大小的螺旋矩阵。

  • 这道题的思路很直接，按照规则顺时针填数就行了，这里注意边界的问题，从左上到右上遍历完以后，行的起始行要+1，从右上到右下的遍历完以后，列的结束行要-1，从右下角到左下角遍历完以后，结束行要-1，从左下角到左上角遍历的时候，起始列要+1.

代码

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> res(n,vector<int>(n,0));
        int row=0;
        int col=0;
        int rows=n-1;
        int cols=n-1;
        int cnt=1;
        while(row<=rows&&col<=cols){
            for(int i=col;i<=cols;i++){
                res[row][i]=cnt;
                cnt++;
            }
            row++;
            for(int i=row;i<=rows;i++){
                res[i][cols]=cnt;
                cnt++;
            }
            cols--;
            for(int i=cols;i>=col;i--){
                res[rows][i]=cnt;
                cnt++;
            }
            rows--;
            for(int i=rows;i>=row;i--){
                res[i][col]=cnt;
                cnt++;
            }
            col++;
        }
        return res;
    }
};