leetcode 54

这题做到一半才想起来原来在程序设计课的作业上做过这道题,原理就是按照螺旋的方式将二维数组的数存进一维数组中,用up down left right分别表示上下左右四条边哪已经存进数组中。若存进则up++ down-- left++ right--若left>right或up>down就break 表示所有的数都存进了数组中。

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        if(matrix.empty()||matrix[0].empty())return {};
        vector<int> res;
        int m=matrix.size();
        int n=matrix[0].size();
        int up=0,left=0,down=m-1,right=n-1;
        while(1)
        {
            for(int i=left;i<=right;i++)res.push_back(matrix[up][i]);
            if(++up>down)break;
            for(int i=up;i<=down;i++)res.push_back(matrix[i][right]);
            if(--right<left)break;
            for(int i=right;i>=left;i--)res.push_back(matrix[down][i]);
            if(--down<up)break;
            for(int i=down;i>=up;i--)res.push_back(matrix[i][left]);
            if(++left>right)break;
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/TempterCyn/article/details/83511357
今日推荐