【LeetCode】54. 螺旋矩阵 (C++)

原题地址：https://leetcode-cn.com/problems/spiral-matrix/description/

题目描述：

给定一个包含 m x n 个元素的矩阵（m 行, n 列），请按照顺时针螺旋顺序，返回矩阵中的所有元素。

示例 1:

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]

示例 2:

输入:
[
  [1, 2, 3, 4],
  [5, 6, 7, 8],
  [9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]

解题方案：

刚开始想的是要把循环次数算出来，然后计算每圈要输出的个数，后来发现还是直接判断坐标比较方便。

代码：

class Solution {
public:
    vector spiralOrder(vector>& matrix) {
        vector ans;
        if(matrix.size() == 0)
            return ans;
        int m = matrix.size();
        int n = matrix[0].size();
        int l_left = 0, l_right = m - 1;
        int r_left = 0, r_right = n - 1;
        while(l_left <= l_right && r_left <= r_right)
        {
            for(int i = r_left; i <= r_right; i ++)
                ans.push_back(matrix[l_left][i]);
            if(l_left + 1 <= l_right - 1)//两行中间还有一行或多行，否则竖着的没有可输出的
                for(int i = l_left + 1; i <= l_right - 1; i ++)
                    ans.push_back(matrix[i][r_right]);
            if(l_left != l_right)//不是只有一行
                for(int i = r_right; i >= r_left; i --)
                    ans.push_back(matrix[l_right][i]);
            if(l_left + 1 <= l_right - 1 && r_left != r_right)
//两行中间还有一行或多行，并且不是只有一列
                for(int i = l_right - 1; i >= l_left + 1; i --)
                    ans.push_back(matrix[i][r_left]);
            l_left ++;
            l_right --;
            r_left ++;
            r_right --;
        }

        return ans;
    }
};