題目：

給你一個大小為 m x n 的矩陣 mat ，請以對角線遍歷的順序，用一個數(shù)組返回這個矩陣中的所有元素。

示例 1：

image.png

輸入：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 <= 10^4
1 <= m * n <= 10^4
-10^5 <= mat[i][j] <= 10^5

java代碼：

class Solution {
        public int[] findDiagonalOrder(int[][] mat) {
            int m = mat.length;
            int 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 == 1) {
                    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--;
                    }
                } else {
                    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++;
                    }
                }
            }
            return res;
        }
    }