[LeetCode] 498. Diagonal Traverse

Given a matrix of M x N elements (M rows, N columns), return all elements of the matrix in diagonal order as shown in the below image.

Example:

Input:
[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]

Output:  [1,2,4,7,5,3,6,8,9]

Explanation:

Note:

The total number of elements of the given matrix will not exceed 10,000.

对角线遍历。

题意是给一个二维数组，按照规则遍历，输出一个一维数组。

这个题没有什么算法，难点是如何判断到底是往上扫描还是往下扫描以及如何判断边界条件。

首先发现如果是从左下往右上遍历，每个遍历到的点的横纵坐标的加和（x+y）% 2 == 0，比如1的坐标是（0,0），7的坐标是（0,2），5的坐标是（1,1）；如果是从右上往左下遍历，遍历到的点的横纵坐标的加和（x+y）% 2 == 1，比如2的坐标（1，0），4的坐标（0,1）。这是一个仅限于这一题的结论。碰到边界条件只有两种可能，一是左下往右上走的时候确保纵坐标不要超出右边的边界和横坐标不要小于0；二是右上往左下走的时候确保横坐标不要大于矩阵的高和纵坐标不要超出左边的边界。

时间O(mn)

空间O(mn) - 存储output

JavaScript实现

/**
 * @param {number[][]} matrix
 * @return {number[]}
 */
var findDiagonalOrder = function (matrix) {
    // corner case
    if (matrix.length === 0) {
        return [];
    }

    // normal case
    let row = 0;
    let col = 0;
    let m = matrix.length;
    let n = matrix[0].length;
    let res = [];
    for (let k = 0; k < m * n; k++) {
        res.push(matrix[row][col]);
        // moving up
        if ((row + col) % 2 === 0) {
            if (col === n - 1) {
                row++;
            } else if (row === 0) {
                col++;
            } else {
                row--;
                col++;
            }
        }
        // moving down
        else {
            if (row === m - 1) {
                col++;
            } else if (col === 0) {
                row++;
            } else {
                row++;
                col--;
            }
        }
    }
    return res;
};

Java实现

注意一定要先写18行和27行否则会越界，因为往右上走的时候，先越界的一定是col；往左下走的时候，先越界的一定是row。

class Solution {
    public int[] findDiagonalOrder(int[][] matrix) {
        // corner case
        if (matrix == null || matrix.length == 0) {
            return new int[0];
        }

        // normal case
        int row = 0;
        int col = 0;
        int m = matrix.length;
        int n = matrix[0].length;
        int[] res = new int[m * n];
        for (int i = 0; i < res.length; i++) {
            res[i] = matrix[row][col];
            // moving up
            if ((row + col) % 2 == 0) {
                if (col == n - 1) {
                    row++;
                } else if (row == 0) {
                    col++;
                } else {
                    row--;
                    col++;
                }
            } else {
                if (row == m - 1) {
                    col++;
                } else if (col == 0) {
                    row++;
                } else {
                    row++;
                    col--;
                }
            }
        }
        return res;
    }
}

LeetCode 题目总结