中等 · 矩阵 · 原题

题目信息

LeetCode #54 · Spiral Matrix

标签：数组、矩阵、模拟

题目要求

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

示例 1：

1
2
输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,3,6,9,8,7,4,5]

示例 2：

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

提示：

  • m == matrix.length

  • n == matrix[i].length

  • 1 <= m, n <= 10

  • -100 <= matrix[i][j] <= 100

代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class Solution {
    static constexpr int DIRS[4][2] = {{0,1},{1,0},{0,-1},{-1,0}};
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        int m=matrix.size(),n=matrix[0].size();
        int size=m*n;
        vector<int> ans;
        int i=0,j=-1;
        for(int di=0;ans.size()<size;di=(di+1)%4){
            for(int k=0;k<n;k++){
                i+=DIRS[di][0];
                j+=DIRS[di][1];
                ans.push_back(matrix[i][j]);
            }
            m--;
            swap(m,n);
        }
        return ans;
    }
};

返回目录