通过万岁!!!
- 题目:给你一个m*n的矩形,然后你按照对角线,呈s型进行遍历到一个一位数组中。
- 思路:首先我们按照层进行遍历,我们可以发现,对角线一共有n+m-1条。并且我们定义一个变量,flag,来记录对角线是右上还是左下。接下来就是确定每条对角线的开始位置,进行遍历即可。其中比较难处理的就是对角线的起始位置。
- 技巧:数组的遍历
java代码
class Solution {
public int[] findDiagonalOrder(int[][] mat) {
int m = mat.length;
int n = mat[0].length;
if (m == 1) {
return mat[0];
}
int[] ans = new int[n * m];
boolean flag = true;// true表示往右上走
// 一共有n*m-1层
int idx = 0;
for (int level = 0; level < n + m - 1; level++) {
if (flag) {// 右上
if (level < m) {// 第level行,第0列开始
for (int i = level, j = 0; i >= 0 && j < n; --i, ++j) {
ans[idx++] = mat[i][j];
}
} else {// m-1行,第level-(m-1)
for (int i = m - 1, j = level - m + 1; i >= 0 && j < n; --i, ++j) {
ans[idx++] = mat[i][j];
}
}
} else {// 左下方向
if (level < n) {// 第0行,第level列
for (int i = 0, j = level; i < m && j >= 0; ++i, --j) {
ans[idx++] = mat[i][j];
}
} else {// 第level-n行 ,第n-1列
for (int i = level - n + 1, j = n - 1; i < m && j >= 0; ++i, --j) {
ans[idx++] = mat[i][j];
}
}
}
flag = !flag;
}
return ans;
}
}
- 总结:题目不是特别难,就是需要找到规律即可。