题目描述
给定一个含有 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;2,4;3,5,7…
- 由题意分析得,当第奇数个对角线遍历时,其为反向输出,我们可以先将其正向遍历出来,再反转
- 最后将所有得到的对角线存入结果数组中即可
3. 具体步骤
- 得到二维数组的长度,创建结果数组
int N = matrix.length;//记录纵坐标
int M = matrix[0].length;//记录横坐标
//声明结果数组
res = new int[M * N];
- 确定所有对角线的起点坐标,进行循环,对角线起点应该为第一行所有以及最后一列所有
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