【Leetcode:498. 对角线遍历 + 模拟】

🚀 算法题 🚀

🌲 算法刷题专栏 | 面试必备算法 | 面试高频算法 🍀
🌲 越难的东西,越要努力坚持，因为它具有很高的价值，算法就是这样✨
🌲 作者简介：硕风和炜，CSDN-Java领域优质创作者🏆，保研|国家奖学金|高中学习JAVA|大学完善JAVA开发技术栈|面试刷题|面经八股文|经验分享|好用的网站工具分享💎💎💎
🌲 恭喜你发现一枚宝藏博主,赶快收入囊中吧🌻
🌲 人生如棋，我愿为卒，行动虽慢，可谁曾见我后退一步？🎯🎯

🚀 算法题 🚀

🚩 题目链接

• 498. 对角线遍历

⛲ 题目描述

给你一个大小为 m x n 的矩阵 mat ，请以对角线遍历的顺序，用一个数组返回这个矩阵中的所有元素。

示例 1：

输入：mat = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,4,7,5,3,6,8,9]
示例 2：

输入：mat = [[1,2],[3,4]]
输出：[1,2,3,4]

提示：

m == mat.length
n == mat[i].length
1 <= m, n <= 104
1 <= m * n <= 104
-105 <= mat[i][j] <= 105

🌟 求解思路&实现代码&运行结果

---

⚡ 模拟

🥦 求解思路

1. 按照对角线进行遍历。设矩阵的行数为 m, 矩阵的列数为 n,一共有 m+n−1 条对角线，当 i 为偶数时，则第 i 条对角线的走向是从下往上遍历，当 i 为奇数时，则第 i 条对角线的走向是从上往下遍历。
2. 有了基本的思路，接下来我们就来通过代码来实现一下的解法。

🥦 实现代码

class Solution {
    public int[] findDiagonalOrder(int[][] mat) {
        int M = mat.length, N = mat[0].length;
        int[] res = new int[M * N];
        int pos = 0;
        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[pos] = mat[x][y];
                    pos++;
                    x--;
                    y++;
                }
            } else {
                int x = i < N ? 0 : i - N + 1;
                int y = i < N ? i : N - 1;
                while (x < M && y >= 0) {
                    res[pos] = mat[x][y];
                    pos++;
                    x++;
                    y--;
                }
            }
        }
        return res;
    }
}

🥦 运行结果

---

💬 共勉

最后，我想和大家分享一句一直激励我的座右铭，希望可以与大家共勉！