题目:
给你一个大小为
m x n
的矩阵mat
,请以对角线遍历的顺序,用一个数组返回这个矩阵中的所有元素。来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/find-the-difference
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例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
解法:
通过点 (x, y) 模拟其在矩阵中的运行。
/**
* @param {number[][]} mat
* @return {number[]}
*/
var findDiagonalOrder = function (mat) {
const m = mat.length;
const n = mat[0].length;
const res = [];
let x = 0; // 记录当前点所在列数
let y = 0; // 记录当前点所在行数
for (let i = 0; i < m + n - 1; i++) {
if (i % 2 == 0) { // 斜向上走
while (x > 0 && y < n - 1) {
res.push(mat[x][y]);
y++;
x--;
}
// 判断拐弯(两种情况,优先判断触碰矩阵底部)
if (y == n - 1) {
res.push(mat[x][y]);
x++;
} else if (x == 0) {
res.push(mat[x][y]);
y++;
}
} else { // 斜向下走
while (y > 0 && x < m - 1) {
res.push(mat[x][y]);
y--;
x++;
}
// 判断拐弯(两种情况,优先判断触碰矩阵右侧)
if (x == m - 1) {
res.push(mat[x][y]);
y++;
} else if (y == 0) {
res.push(mat[x][y]);
x++;
}
}
}
return res;
};