描述
给你一个大小为 m x n 的矩阵 mat ,请以对角线遍历的顺序,用一个数组返回这个矩阵中的所有元素。
分析
找规律,什么情况向右上遍历,什么情况向左下遍历。什么情况需要调整方向掉头。
规律是:i+j的和为奇数时向左下遍历,是偶数是向右上遍历。
对于向左下遍历的情况,在遇到i等于n-1时,向右平移一位;在j等于0时,向下平移一位;且先判断i是否等于n-1,再判断j是否等于0,不能理解的想象一下当前向左下遍历的时矩阵的对角线时应该怎样正确的调整。
对于向右上遍历的情况,在遇到j等于m-1时,向下平移一位;在i等于0时,向右平移一位。
class Solution {
public int[] findDiagonalOrder(int[][] mat) {
int n = mat.length;
if(n == 0){
return null;
}
int m = mat[0].length;
int[] res = new int[n*m];
int k = 0, i = 0, j = 0;
while(k < res.length){
res[k++] = mat[i][j];
if((i+j)%2 == 0){
if(j == m-1){
i++;
}else if(i == 0){
j++;
}else{
i--;
j++;
}
}else{
if(i == n-1){
j++;
}else if(j == 0){
i++;
}else{
i++;
j--;
}
}
}
return res;
}
}