题目描述:
给定一个含有 M x N 个元素的矩阵(M 行,N 列),请以对角线遍历的顺序返回这个矩阵中的所有元素,对角线遍历如下图所示。
示例:
输入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
输出: [1,2,4,7,5,3,6,8,9]
解释:
说明:
给定矩阵中的元素总数不会超过 100000 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/diagonal-traverse
就是之字形打印矩阵,之前博客也有写。。。再写一次还是得扣挺久,毕竟是一道medium的题
思路: 在左上角的元素开始,有两个指针一个一直往右,到最右后再往下,一个一直往下,到最下边后再往右, 两个指针连接的对角线就是每次要打印的数值;同时一个布尔常量控制是从左下往右上打印还是相反方向
/**
* @param {number[][]} matrix
* @return {number[]}
*/
var findDiagonalOrder = function(matrix) {
if(matrix.length === 0) {
return [];
}
let res = [];
let printLevel = function (m, tR, tC, dR, dC, f) {
if (!f) {
// 左下向右上打印
while(dR !== tR - 1) {
res.push(m[dR--][dC++]);
}
} else {
// 右上向左下角打印
while(tR !== dR + 1) {
res.push(m[tR++][tC--]);
}
}
}
let tr = 0, dr = 0, tc = 0, dc = 0;
let fromUp = false; // 该布尔量指示遍历的方向是从右上往左下的对角线还是从左下往右上方向对角线方向遍历
let endRow = matrix.length - 1;
let endCol = matrix[0].length - 1;
while(tr !== endRow + 1) {
printLevel(matrix, tr, tc, dr, dc, fromUp);
tr = tc === endCol ? tr + 1 : tr;
tc = tc === endCol ? tc : tc + 1;
dc = dr === endRow ? dc + 1 : dc;
dr = dr === endRow ? dr : dr + 1;
fromUp = !fromUp; // 方向变换
}
return res;
};