LeetCode498. 对角线遍历

磨叽墨迹墨迹

于 2020-12-09 16:41:07 发布

阅读量164

点赞数

分类专栏： Leetcode 文章标签： leetcode 数据结构

本文链接：https://blog.youkuaiyun.com/qq_37614657/article/details/110929540

版权

Leetcode 专栏收录该内容

67 篇文章

订阅专栏

这是一篇关于LeetCode498题目的博客，讲述了如何进行对角线遍历一个MxN矩阵，并提供了C++的解决方案。虽然原题思路可能有些复杂，但作者提供了简洁的实现代码。

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

原题链接：对角线遍历。

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

思路：https://leetcode-cn.com/problems/diagonal-traverse/solution/xiao-bai-kan-guo-lai-zui-zhi-bai-yi-li-jie-ban-ben/

有点没大看懂。后面有空再细看下。

C++实现如下：

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& matrix) {
        int Rowlength=matrix.size();
        if(Rowlength==0)
            return {};
        int Cowlength=matrix[0].size();
        vector<int> answer;
        //横坐标和纵坐标之和
        int count=Rowlength+Cowlength-1;
        int m=0;
        int n=0;
        for(int i=0;i<count;i++)
        {
            if(i%2==0)//右上方向
            {
                while(m>=0&&n<Cowlength)
                {
                    answer.push_back(matrix[m][n]);
                    m--;
                    n++;
                }
                if(n<Cowlength)
                     m++;
                else
                {
                    m=m+2;
                    n--;
                }
            }
            else{//左下方向
                while(n>=0&&m<Rowlength)
                {
                    answer.push_back(matrix[m][n]);
                    m++;
                    n--;
                }
                if(m<Rowlength)
                    n++;
                else
                {
                    m--;
                    n=n+2;
                }
            }
        }
        return answer;
    }
};