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

解题思路：循环嵌套，大循环停止条件是遍历完矩阵中所有的元素（mat[i][j]）。

大循环内两个小循环，小循环1沿对角线向上（i--, j++），1的执行条件是（i>=0 && j<n），小循环2延对角线向下(i++, j--)，2的执行条件是（i<m && j >= 0）。

两个小循环之间需要确定下一个小循环的起始点：

（1）当矩阵m=n时，循环2的起始元素是循环1终止元素的向右一个（j++），当向右无元素时（j+1>=n），向下一个（i++）。循环1的起始元素是循环2终止元素的向下一个（i++），当向下无元素时，向右一个（j++）。

(2)当矩阵m>n和m<n时，

              

情况与m=n时完全相同。

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