LeetCode 螺旋矩阵(Spiral Matrix)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/GreenHandCGL/article/details/82798511

思路:做四个方向的遍历右,下,左,上。重复循环这四个方向即得到螺旋矩阵。

注意问题:

1、边界

2、判断当前位置是否已经遍历(mark标志位)

3、vector为空或者只有一个元素

4、循环退出条件

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> res;
        if(matrix.size() == 0 || matrix[0].size() == 0) return res;
        bool flag = true;
        int m = matrix.size(), n = matrix[0].size();
        vector<int> v(n);
        vector<vector<int>> mark(m,v);
        int i = 0, j = 0;
        res.push_back(matrix[i][j]);
        mark[i][j] = 1;
        while(flag){
            while(j + 1< n && mark[i][j+1] != 1){res.push_back(matrix[i][++j]); mark[i][j] = 1;}
            while(i + 1< m && mark[i+1][j] != 1){res.push_back(matrix[++i][j]); mark[i][j] = 1;}
            while(j > 0 && mark[i][j-1] != 1){res.push_back(matrix[i][--j]); mark[i][j] = 1;}
            while(i > 0 && mark[i-1][j] != 1){res.push_back(matrix[--i][j]); mark[i][j] = 1;}
            if(j + 1 == n) flag = false;
            else if(mark[i][j+1] == 1) flag = false;
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/GreenHandCGL/article/details/82798511