1424.对角线遍历 II

最新推荐文章于 2026-05-23 23:35:41 发布

原创最新推荐文章于 2026-05-23 23:35:41 发布 · 325 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

力扣Leetcode

当前文章被收录于：

力扣Leetcode

23 篇文章 0 人学习

刷题

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍了一种矩阵对角线遍历的算法实现，通过将矩阵元素按其横纵坐标之和分类，利用二维数组存储，最终反转输出遍历结果。此方法适用于左上至右下对角线遍历场景。

在这里插入图片描述
思路：可以发现每条对角线上元素，横坐标加纵坐标的和相等，且和越小的先被遍历，所以我们可以按照x+y去分类，开一个二维数组ans[ i ] [ j ]，i表示（x+y），然后正常遍历这个nums，每遍历到一个元素，看看它的x+y，把它丢到ans[x+y]中。由于我们是从左到右从上到下遍历nums的，所以答案是反过来的，最后再reverse一下就行了。

class Solution {
public:

    vector<int> findDiagonalOrder(vector<vector<int>>& nums) {
        vector<int> ans[200005];
        vector<int> solve;
        int maxz=0;
        for(int i=0;i<nums.size();i++)
        {
            for(int j=0;j<nums[i].size();++j)
            {
                ans[i+j].push_back(nums[i][j]);
                if(i+j>maxz) maxz=i+j;
            }
        }
        for(int i=0;i<=maxz;i++)
        {
            for(int j=ans[i].size()-1;j>=0;--j)
            {
                solve.push_back(ans[i][j]);
            }
        }
        return solve;
    }
};