遍历对角线——力扣498

题目描述

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

示例:

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

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

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

---

2. 思路

1. 对角线遍历，那么我们可以先遍历所有的正向对角线，即：1；2，4；3，5，7…
2. 由题意分析得，当第奇数个对角线遍历时，其为反向输出，我们可以先将其正向遍历出来，再反转
3. 最后将所有得到的对角线存入结果数组中即可

3. 具体步骤

1. 得到二维数组的长度，创建结果数组

 int N = matrix.length;//记录纵坐标
        int M = matrix[0].length;//记录横坐标
        //声明结果数组
        res = new int[M * N];

2. 确定所有对角线的起点坐标，进行循环，对角线起点应该为第一行所有以及最后一列所有

for (int i = 0; i < M + N -1; i++){
            //获得对角线起点横纵坐标
            int x = i < M ? 0 : i - M + 1;
            int y = i < M ? i : M - 1;
            //确定横纵坐标之后，向左下开始遍历数组
            list.addAll(get(matrix, x , y, i));
        }
        for (int i = 0; i < list.size(); i++){
            res[i] = list.get(i);
        }
        return res;
    }

3. 将起点坐标找到后，正向便利对角线元素，即横坐标向左移动一个单位，纵坐标向下移动一个单位

public List<Integer> get(int[][] matrix, int x, int y, int i){
        List<Integer> res = new ArrayList<Integer>();
        while(true){
            //当横坐标小于0或者纵坐标大于N后，结束循环
            if (y < 0 || x >= matrix.length){
                break;
            }
            //否则在res中记录当前横纵坐标的数
            res.add(matrix[x][y]);
            x = x + 1;
            y = y - 1;
        }
        if (i % 2 == 0){
            Collections.reverse(res);
        }
        return res;
    }

4. 当运行到第奇数个起点时，反转链表，对应到代码中，i从0开始，所以当i对2取模为0时，将结果反转。
5. 存储对角线元素可以使用动态数组存储，最后将其加在数组中即可

4. 完整代码

class Solution {
     public int[] findDiagonalOrder(int[][] matrix) {
        int[] res;
        if(matrix == null || matrix.length == 0){
            res = new int[0];
            return res;
        }
        int N = matrix.length;//记录纵坐标
        int M = matrix[0].length;//记录横坐标
        //声明结果数组
        res = new int[M * N];
        List<Integer> list = new ArrayList<Integer>();
        //遍历得到对角线起点
        for (int i = 0; i < M + N -1; i++){
            //获得对角线起点横纵坐标
            int x = i < M ? 0 : i - M + 1;
            int y = i < M ? i : M - 1;
            //确定横纵坐标之后，向左下开始遍历数组
            list.addAll(get(matrix, x , y, i));
        }
        res = list.toArray();
        for (int i = 0; i < list.size(); i++){
            res[i] = list.get(i);
        }
        return res;
    }

    public List<Integer> get(int[][] matrix, int x, int y, int i){
        List<Integer> res = new ArrayList<Integer>();
        while(true){
            //当横坐标小于0或者纵坐标大于N后，结束循环
            if (y < 0 || x >= matrix.length){
                break;
            }
            //否则在res中记录当前横纵坐标的数
            res.add(matrix[x][y]);
            x = x + 1;
            y = y - 1;
        }
        if (i % 2 == 0){
            Collections.reverse(res);
        }
        return res;
    }

}