Leetcode 498. 对角线遍历

JehanRio

已于 2023-10-02 09:21:32 修改

阅读量94

点赞数 1

分类专栏： Leetcode专栏文章标签： leetcode 算法职场和发展

于 2023-07-04 20:14:22 首次发布

本文链接：https://blog.csdn.net/weixin_51322383/article/details/131542201

版权

Leetcode专栏专栏收录该内容

278 篇文章 0 订阅

订阅专栏

文章目录

题目
代码（首刷看解析）
代码（10.2 二刷看解析）

题目

在这里插入图片描述
Leetcode 498. 对角线遍历

代码（首刷看解析）

作图模拟一遍会理解更快。

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& mat) {
        int m = mat.size(), n = mat[0].size();
        vector<int> res;
        // 一共m+n-1条对角线
        for(int i = 0; i < m+n-1; i++) {
            if(i % 2 == 0) {   // 从下往上
                int x = i < m ? i : m - 1;
                int y = i < m ? 0 : i - m + 1;
                while(x >= 0 && y < n) {
                    res.emplace_back(mat[x--][y++]);
                }
            } else {
                int x = i < n ? 0 : i - n + 1;
                int y = i < n ?i : n - 1;
                while(x < m && y >= 0) {
                    res.emplace_back(mat[x++][y--]);
                }
            }
        }
        return res;
    }
};

代码（10.2 二刷看解析）

class Solution {
public:
    vector<int> findDiagonalOrder(vector<vector<int>>& mat) {
        int m = mat.size(), n = mat[0].size();
        vector<int> res;
        for(int i = 0; i < m+n-1; i++) {
            if(i%2==0) {
                int x = i < m ? i : m-1;
                int y = i < m ? 0 : i-m+1;
                while(x >= 0 && y < n)
                    res.emplace_back(mat[x--][y++]);
            } else {
                int x = i < n ? 0 : i-n+1;
                int y = i < n ? i : n-1;
                while(x < m && y >= 0)
                    res.emplace_back(mat[x++][y--]);
            }
        }
        return res;
    }
};