LeetCode 54螺旋矩阵

LeetCode 54螺旋矩阵

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

• 示例

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

• 思路：四指针扫描O(n²)

  1. 从左到右访问第一行，left++
  2. 从上到下访问最后一列，up--
  3. 从右往左访问最后一行，right--
  4. 从下到上访问第一列，down++
  5. up和down以行数为界限，left和right以列数为界限

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> ans;
        if(matrix.empty()) return ans;
        int m = matrix.size(), n = matrix[0].size();//行数列数
        int left = 0, right = n - 1;
        int up = 0, down = m - 1;
        while(up <= down && left <= right)
        {
            for(int i = left; i <= right; i++) ans.push_back(matrix[up][i]);
            if(++up > down) break;//重新设定上边界，若上边界大于下边界，则遍历遍历完成，下同
            for(int i = up; i <= down; i++) ans.push_back(matrix[i][right]);
            if(--right < left) break;//重新设定有边界
            for(int i = right; i >= left; i--) ans.push_back(matrix[down][i]);
            if(--down < up) break;//重新设定下边界
            for(int i = down; i >= up; i--) ans.push_back(matrix[i][left]);     
            if(++left > right) break;//重新设定左边界
        }
        return ans;
    }
};