leetcode-498. 对角线遍历（JavaScript）

题目：

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/find-the-difference
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

点此跳转LeetCode-498. 对角线遍历

示例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

解法：

        通过点 (x, y) 模拟其在矩阵中的运行。

/**
 * @param {number[][]} mat
 * @return {number[]}
 */
var findDiagonalOrder = function (mat) {
    const m = mat.length;
    const n = mat[0].length;
    const res = [];
    let x = 0;  // 记录当前点所在列数
    let y = 0;  // 记录当前点所在行数
    for (let i = 0; i < m + n - 1; i++) {
        if (i % 2 == 0) {  // 斜向上走
            while (x > 0 && y < n - 1) {
                res.push(mat[x][y]);
                y++;
                x--;
            }
            // 判断拐弯（两种情况，优先判断触碰矩阵底部）
            if (y == n - 1) {
                res.push(mat[x][y]);
                x++;
            } else if (x == 0) {
                res.push(mat[x][y]);
                y++;
            }
        } else {  // 斜向下走
            while (y > 0 && x < m - 1) {
                res.push(mat[x][y]);
                y--;
                x++;
            }
            // 判断拐弯(两种情况，优先判断触碰矩阵右侧)
            if (x == m - 1) {
                res.push(mat[x][y]);
                y++;
            } else if (y == 0) {
                res.push(mat[x][y]);
                x++;
            }
        }
    }
    return res;
};