LeetCode 刷题 [C++] 第54题.螺旋矩阵

题目描述

给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。

题目分析

1. 根据题意可知，我们不需要记录已经走过的路径，只需要通过调整矩阵的上下左右边界即可完成任务；
2. 首先创建出矩阵的上下左右边界；
3. 开始遍历矩阵：
   先从左上角，从左向右遍历第一行，然后删除第一行，即重新定义上边界；
   然后判断重新定义后的上下边界是否交错，如果交错，则表示螺旋矩阵遍历结束，跳出循环，返回数组；
   如果重新定义后的边界没有交错，则接着遍历矩阵；
4. 使用3中类似的逻辑接着向下向左向上移动；
5. 不断循环步骤3和4，直至某两个边界交错，跳出循环为止，最后返回结果。

Code

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        if (matrix.empty()) return {};
        vector<int> res;
        res.reserve(matrix.size()*matrix[0].size()); //预分配容器的大小
        int upper = 0, down = matrix.size() - 1;
        int left = 0, right = matrix[0].size() - 1;
        while (true) {
        	// 向右遍历
            for (int i = left; i <= right; ++i) {
                res.emplace_back(matrix[upper][i]);
            }
            if (++upper > down) {
                break;
            }
			// 向下遍历
            for (int i = upper; i <= down; ++i){
                res.emplace_back(matrix[i][right]);
            }
            if (--right < left) {
                break;
            }
			// 向左遍历
            for (int i = right; i >= left; --i) {
                res.emplace_back(matrix[down][i]);
            }
            if (--down < upper) {
                break;
            }
			// 向上遍历
            for (int i = down; i >= upper; --i) {
                res.emplace_back(matrix[i][left]);
            }
            if (++left > right) {
                break;
            }
        }
        return res;
    }
};