Leetcode 59. 螺旋矩阵 II

最新推荐文章于 2021-02-19 21:12:09 发布

Bendaai

最新推荐文章于 2021-02-19 21:12:09 发布

阅读量275

点赞数

本文链接：https://blog.csdn.net/Bendaai/article/details/80221034

版权

和Leetcode 54. 螺旋矩阵类似

class Solution {
public:
    int dx[4]={0,1,0,-1},dy[4]={1,0,-1,0},f=0;
    vector<vector<int>> generateMatrix(int n) {
        vector<vector<int>> ans(n, vector<int>(n,0));
        int cnt=0,x=0,y=0;
        while(++cnt!=n*n+1){
            ans[x][y]=cnt;
            int nx=x+dx[f],ny=y+dy[f];
            if(!(nx>=0&&nx<n && ny>=0 && ny<n && ans[nx][ny]==0))
                f=(f+1)&3;
            x+=dx[f],y+=dy[f];
        }
        return ans;
    }
};