LeetCode—— 498， 对角线遍历

/**
     * 给定一个含有 M x N 个元素的矩阵（M 行，N 列），请以对角线遍历的顺序返回这个矩阵中的所有元素，对角线遍历如下图所示。
     * <p>
     * 输入:
     * [
     * [ 1(0,0), 2(0,1), 3(0,2) ],
     * [ 4(1,0), 5(1,1), 6(1,2) ],
     * [ 7(2,0), 8(2,1), 9(2,2) ]
     * ]
     * <p>
     * 输出:  [1,2,4,7,5,3,6,8,9]
     *
     * @param matrix
     * @return
     */
    public int[] findDiagonalOrder(int[][] matrix) {
        if (matrix.length == 0) {
            return new int[0];
        }
        int row = matrix.length;
        int col = matrix[0].length;
        int totalNum = row * col;
        if (totalNum == 1) {
            return new int[]{matrix[0][0]};
        }
        int result[] = new int[totalNum];
        int currentRow = 0;