LeetCode刷题日记之对角线遍历

最新推荐文章于 2022-06-14 23:59:42 发布

ych12345xx

最新推荐文章于 2022-06-14 23:59:42 发布

阅读量186

点赞数

分类专栏：学习笔记

本文链接：https://blog.csdn.net/ych12345xx/article/details/103439577

版权

学习笔记专栏收录该内容

63 篇文章 0 订阅

订阅专栏

这篇博客详细介绍了如何解决LeetCode上的对角线遍历问题。作者从问题描述出发，分析了对角线遍历的规律，即索引和为0的元素沿右上方向遍历，索引和为1的元素沿左下方向遍历，同时强调了边界条件的处理。接着，文章进入问题求解部分，未给出具体解决方案。

摘要由CSDN通过智能技术生成

一：问题描述

对角线遍历

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

示例:

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

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

解释:

说明:

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

二：问题分析

分析问题易知，二维数组中索引和相同的元素为同一次对角遍历的所有元素，索引和%2为0的元素对角遍历顺序为向右上遍历，索引和%2为1的元素对角遍历顺序为向左下遍历。特别要注意边界问题，右上遍历的元素碰见行上边界(行为0时)，下一个元素位置为当前行不变，列加一，右上遍历元素碰见列右边界时，下一个元素位置为当前行加一，列不变，同时碰见行上边界和列右边界的元素，下一个元素位置为当前行加一，列不变；左下遍历的元素同理类推。

三：问题求解

class Solution {
    public int[] findDiagonalOrder(int[][] matrix) {
        if(matrix.length==0||matrix[0].length==0){
            return new int[0];
        }
        int rows=matrix.length;
        int cols=matrix[0].length;
        int[] res=new int[rows*cols];
        int i=0;
        int j=0;
        for(int n=0;n<rows*cols;n++){
            res[n]=matrix[i][j];
            if((i+j)%2==0){
                //下面if代码顺序不可变
                if(j==cols-1){
                    i++;
                }
                else if(i==0){
                    j++;
                }
                else{
                    i--;
                    j++;
                }
            }
            else{
                //下面if代码顺序不可变
                if(i==rows-1){
                    j++;
                }
                else if(j==0){
                    i++;
                }
                else{
                    i++;
                    j--;
                }
            }
        }
        return res;
    }
}