例一:对角线遍历
链接:https://leetcode-cn.com/problems/diagonal-traverse/
题目描述:给你一个大小为 m x n 的矩阵 mat ,请以对角线遍历的顺序,用一个数组返回这个矩阵中的所有元素。
示例 1:
输入:mat = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,4,7,5,3,6,8,9]
示例 2:
输入:mat = [[1,2],[3,4]]
输出:[1,2,3,4]
提示:
m == mat.length
n == mat[i].length
1 <= m, n <= 104
1 <= m * n <= 104
-105 <= mat[i][j] <= 105
题目分析:
刚看到这个题目的时候,我心里飘过的想法是就普普通通的遍历嘛,但是在真正准备写代码的时候发现并没有这么简单。
经过分析可以发现数学规律:
1、上图中红色线的对角线上的坐标之和均为偶数且在遍历时是向上移动。
2、黄色线对角线上的坐标之和均为奇数,且均为向下移动。
3、对角线上遍历时(不考虑边界)上移是(x+1,y-1),下移是(x-1,y+1)。
在弄清楚上面几条过后,这个题的思路就已经清晰了,但是我们还需要考虑边界问题,即上图中的1,4,3,8等位置。
而边界主要是第一列,最后一列和第一行,最后一行
具体边界边界判断及处理如下:
if((x+y)%2==0){//对角线坐标之和为偶数,向上移动
if(y