LeetCode——对角线遍历（之字形遍历）

最新推荐文章于 2025-08-13 21:50:19 发布

九千里

最新推荐文章于 2025-08-13 21:50:19 发布

阅读量255

点赞数

CC 4.0 BY-SA版权

分类专栏：刷LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/weixin_37605770/article/details/104960455

刷LeetCode 专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种矩阵对角线遍历的算法实现，通过控制遍历方向和索引变化，实现了从左上到右下，再从右上到左下的Z形遍历路径。适用于MxN矩阵，确保了遍历的有序性和覆盖所有元素。

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

索引和为{偶}数，向上遍历，{横}索引值递减，遍历值依次是(x,0),(x-1,1),(x-2,2),…,(0,x)
索引和为{奇}数，向下遍历，{纵}索引值递减，遍历值依次是(0,y),(1,y-1),(2,y-2),…,(y,0)

题目：https://leetcode-cn.com/explore/featured/card/array-and-string/199/introduction-to-2d-array/774/

给定一个含有 M x N 个元素的矩阵（M 行，N 列），请以对角线遍历的顺序返回这个矩阵中的所有元素，对角线遍历如下图所示。

示例:

输入:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]

输出:  [1,2,4,7,5,3,6,8,9]

解释:

说明:

给定矩阵中的元素总数不会超过 100000 。

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
         vector<int> ans;
        if(matrix.empty()) return ans;
        int i = 0,j = 0;
        int hang = matrix.size();
        int lie = matrix[0].size();
        for(int p = 0;p< hang *lie ; p++){
            ans.push_back(matrix[i][j]);
            if((i+j)%2==0){
                if(i==0 && j==lie-1) i++;
                else if(i==0 && j!=lie-1) j++;
                else if(i!=0 && j == lie-1) i++;
                else {i--;j++;}
            }
            else {
                if(i==hang-1) j++;
                else if(j==0) i++;
                else {i++;j--;}
            }
        }
        return ans;
     }
};