【LeetCode】498. 对角线遍历

【LeetCode】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. 走到了第一行，但是列还没到最大列： i == 0 && j < n - 1
2. 没走到第一行，但是走到了最大列：i > 0 && j == n - 1
3. 走到第一行的同时也走到了最大列：i == 0 && j == n - 1

当这一对角线遍历完，下一次对角线遍历的起始位置在哪里呢？

1. 走到了第一行，但是列还没到最大列 => 行不变，列加一：j++
2. 没走到第一行，但是走到了最大列 => 列不变，行加一：i++
3. 走到第一行的同时也走到了最大列 => 列不变，行加一：i++

可以发现2、3两种情况可以合并为：i >= 0 && j == n - 1 => i++

最终：

1. 走到了第一行，但是列还没到最大列： i == 0 && j < n - 1 => j++
2. 没走到第一行或者走到第一行，但是走到了最大列：i >= 0 && j == n - 1 => i++

从上到下：

1. 走到最后一行，但是列没到最小列：i == m - 1 && j > 0
2. 走到最后一行，列到了最小列：i == m - 1 && j == 0
3. 没走到最后一行，列到了最小列：i < m - 1 && j == 0

当这一对角线遍历完，下一次对角线遍历的起始位置在哪里呢？

1. 走到最后一行，但是列没到最小列 => 行不变，列加一 ：j++
2. 走到最后一行，列到了最小列 => 行不变，列加一：j++
3. 没走到最后一行，列到了最小列 => 列不变，行加一：i++

可以发现1、2两种情况可以合并为：i == m - 1 && j >= 0 => j++

最终：

1. 走到最后一行，列到最小列或者没到最小列：i == m - 1 && j >= 0=>j++
2. 没走到最后一行，列到了最小列：i < m - 1 && j == 0=>i++

将规律写成代码：

class Solution {
    public int[] findDiagonalOrder(int[][] mat) {
        int m = mat.length;
        int n = mat[0].length;
        int len = m * n;
        int[] res = new int[len];
        int index = 0;
        int i = 0;
        int j = 0;
        while(index < len) {
            while(true) {
                if (index == len) {
                    break;
                }
                res[index++] = mat[i][j];
                if(i == 0 && j < n - 1) {
                    j++;
                    break;
                }else if(i >= 0 && j == n - 1) {
                    i++;
                    break;
                }else {
                    i--;
                    j++;
                }
            }
            while(true) {
                if(index == len) {
                    break;
                }
                res[index++] = mat[i][j];
                if(i < m - 1 && j == 0) {
                    i++;
                    break;
                }else if(i == m - 1 && j >= 0) {
                    j++;
                    break;
                }else {
                    i++;
                    j--;
                }
            }
        }
        return res;
    }
}