Leetcode 54：旋转矩阵

最新推荐文章于 2022-09-06 21:33:03 发布

原创最新推荐文章于 2022-09-06 21:33:03 发布 · 361 阅读

0 ·

CC 4.0 BY-SA版权

博客围绕矩阵元素输出问题展开，给定一个 m 行 n 列的矩阵，要求按顺时针螺旋顺序返回其中所有元素，并给出示例，还提及了解题思路，但未详述具体解法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定一个包含 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]

解题思路

vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> ans;
        if(matrix.size() == 0 || matrix[0].size() == 0) return ans;
        ans.push_back(matrix[0][0]);
        int rows = matrix.size(),cols = matrix[0].size();
        int dx = 0,dy = 1,x0 = 0,y0 = 0;
        int step = cols-1,n = 0,stepx = rows-1,stepy = cols-1;
        while(ans.size() < rows*cols){
            for(int i=1;i<=step;i++){
                x0 += dx;
                y0 += dy;
                if(x0>=0 && x0<rows && y0>=0 && y0<cols){
                    ans.push_back(matrix[x0][y0]);
                }
            }
            n = (n+1)%4;
            if(n%2 == 0){
                step = stepy;
                if(n == 2) stepy -= 2;
            }else{
                step = stepx;
                stepx--;
            }
            swap(dx,dy);
            dy = -dy;
            if(ans.size() < rows*cols){
                if(dx == 0 && dy == 1){
                    x0 += dx;
                    y0 += dy;
                    ans.push_back(matrix[x0][y0]);
                }
            }
        }
        return ans;
    }