LeetCode-对角线遍历二维数组

LeetCode-对角线遍历二维数组

刷了一道题,是关于二维数组的,我这是第一次做二维数组的题目,因为对二维数组的不熟悉导致思路比较乱,中途甚至想要放弃,直接找题解,最终花了好长时间才做出来.虽然效率比较低,但是我觉得还是有必要记录一下思考的过程.

题目描述

给定一个含有 M x N 个元素的矩阵（M 行，N 列），请以对角线遍历的顺序返回这个矩阵中的所有元素，对角线遍历如下图所示。

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

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

解释: 

说明:

给定矩阵中的元素总数不会超过 100000 。

我的解法思路

有个开发经验比我丰富很多的同事告诉我,做功能的时候先不要想着一步到位,可以先把基本的功能做出来再一点一点加东西.深以为然.

这个问题是对角线遍历,那我们把问题按照这个思路分解一下就是遍历---对角线遍历---符合题目顺序的对角线遍历

第一步:遍历

先把这个二维数组遍历出来.

public int[] findDiagonalOrder(int[][] matrix) {
        int res0[] = new int[0];
        if(matrix.length==0) return res0;
        List<Integer> list = new ArrayList(); 
        for(int i=0;i<matrix.length;i++){
            for(int j=0;j<matrix[i].length;j++){
                list.add(matrix[i][j]);
            }
        }
        int res[] = new int[list.size()];
        for(int i=0;i<list.size();i++){
            res[i] = list.get(i);
        }
        return res;
    }

我的输入:

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

我的输出:

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

第二步:对角线遍历

这个遍历肯定不会是像题目中所示的那样,环形遍历,而是都是从右上角到左下角的遍历.

public int[]