59. Spiral Matrix II

仅供自己学习

 

思路：

同样设置上下左右边界，因为是n*n的方阵，所以右边界和下边界为n-1。同时定义一个count=1用来给每个方阵赋值。

从左上角开始螺旋遍历，第一个for从左边界移动到右边界，然后上边界+1。第二个for从上边界移动到下边界，然后右边界-1，再从右边界移动到左边界，下边界-1，再从下边界移动到上边界，然后左边界+1.循环结束的条件是 count<=n*n。

 

代码：

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        int t=0;
        int d=n-1;
        int l=0;
        int r=n-1;
        vector<vector<int>> res(n,vector<int>(n));
        int count=1;
        while(count<=n*n){
            for(int i=l;i<=r;++i) res[t][i]=count++;
            if(++t>d) break;
            for(int i=t;i<=d;++i) res[i][r]=count++;
            if(--r<l) break;
            for(int i=r;i>=l;--i) res[d][i]=count++;
            if(--d<t) break;
            for(int i=d;i>=t;--i) res[i][l]=count++;
            if(++l>r) break;
        }
        return res;
    }
};