遍历对角线——力扣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];
  1. 确定所有对角线的起点坐标,进行循环,对角线起点应该为第一行所有以及最后一列所有
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;
    }
  1. 将起点坐标找到后,正向便利对角线元素,即横坐标向左移动一个单位,纵坐标向下移动一个单位
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;
    }
  1. 当运行到第奇数个起点时,反转链表,对应到代码中,i从0开始,所以当i对2取模为0时,将结果反转。
  2. 存储对角线元素可以使用动态数组存储,最后将其加在数组中即可

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;
    }

}
发布了26 篇原创文章 · 获赞 0 · 访问量 2134
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览