LeetCode 54. Spiral Matrix

最新推荐文章于 2024-07-30 15:23:09 发布

原创最新推荐文章于 2024-07-30 15:23:09 发布 · 683 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #C++

LeetCode 同时被 2 个专栏收录

186 篇文章

订阅专栏

LeetCode Problems

183 篇文章

订阅专栏

本文介绍了一种螺旋遍历二维矩阵的方法，并提供了两种实现方案。通过定义边界，算法能够有效地按螺旋顺序输出矩阵中的元素。

注意边界的处理。

代码：

class Solution 
{
public:
    vector<int> spiralOrder(vector<vector<int> > &matrix) 
    {
        if (matrix.empty() == true)
        {
            return vector<int>();
        }
        int x1 = 0, y1 = 0;
        int x2 = matrix.size()-1, y2 = matrix[0].size()-1;
        vector<int> ret;

        for ( ; x1<=x2 && y1<=y2; ++x1, ++y1, --x2, --y2)
        {
            for (int j = y1;   j <= y2; ret.push_back(matrix[x1][j++])) {} 
            for (int i = x1+1; i <= x2; ret.push_back(matrix[i++][y2])) {}
            for (int j = y2-1; x1!=x2 && j>=y1;   ret.push_back(matrix[x2][j--])) {} // x1!=x2
            for (int i = x2-1; y1!=y2 && i>=x1+1; ret.push_back(matrix[i--][y1])) {} // y1!=y2
        }

        return ret;
    }
};

之前的写法：

class Solution 
{
public:
    vector<int> spiralOrder(vector<vector<int> > &matrix) 
    {
        if (matrix.empty() == true)
        {
            return vector<int>();
        }
        int x1 = 0, y1 = 0;
        int x2 = matrix.size()-1, y2 = matrix[0].size()-1;
        vector<int> ret;
        while (x1<=x2 && y1<=y2)
        {
            for (int j = y1; j <= y2; ++ j)
            {
                ret.push_back( matrix[x1][j] );
            }
            for (int i = x1+1; i <= x2; ++ i)
            {
                ret.push_back( matrix[i][y2] );
            }
            for (int j = y2-1; x1!=x2 && j>=y1; -- j) // 加上x1!=x2避免重复输出
            {
                ret.push_back( matrix[x2][j] );
            }
            for (int i = x2-1; y1!=y2 && i>x1; -- i) // 加上y1!=y2避免重复输出
            {
                ret.push_back( matrix[i][y1] );
            }
            ++x1, ++y1, --x2, --y2;
        }
        return ret;
    }
};