【leetcode】54. 螺旋矩阵（spiral-matrix）（模拟）[中等]

链接

https://leetcode-cn.com/problems/spiral-matrix/

耗时

解题：47 min
题解：12 min

题意

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

提示：

• m == matrix.length
• n == matrix[i].length
• 1 <= m, n <= 10
• -100 <= matrix[i][j] <= 100

思路

从外向内一圈一圈往里走，把每一圈看成一个循环结，一圈包括 上、右、下、左 四个部分，分别遍历即可，需要注意的是当只有一行或一列时，需要特殊判断一下只遍历 上 或 右 就可以了，再多遍历，位置平行的部分就重复了。还有第一部分要遍历整个一行，否则的话只有一个元素时就不会遍历它。

时间复杂度：$O(mn)$

AC代码

class Solution {
    
    
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
    
    
        int m = matrix.size();
        int n = matrix[0].size();
        vector<int> ans;
        int cnt = (min(n, m)+1)/2;
        for(int i = 0; i < cnt; ++i) {
    
    
            for(int j = i; j <= n-i-1; ++j) {
    
    
                ans.push_back(matrix[i][j]);
            }
            for(int j = i+1; j <= m-i-1; ++j) {
    
    
                ans.push_back(matrix[j][n-i-1]);
            }
            if(m-2*i > 1) {
    
    
                for(int j = n-i-2; j >= i; --j) {
    
    
                    ans.push_back(matrix[m-i-1][j]);
                }
            }
            if(n-2*i > 1) {
    
    
                for(int j = m-i-2; j > i; --j) {
    
    
                    ans.push_back(matrix[j][i]);
                }
            }
        }
        return ans;
    }
};