思路: 题目描述的就是我们打印每一层的数据,但是偶数层的数据从右向左写入,很容易想到层次遍历二叉树,然后对遍历结果中奇数行进行翻转(因为index从0开始)即可
/**
* @param {TreeNode} root
* @return {number[][]}
*/
var levelOrder = function(root) {
let res = []
let queue = []
if(!root){ return res}
queue.push(root)
while(queue.length){
let len = queue.length
let temp = []
for(let i=0;i<len;i++){
let node = queue.shift()
temp.push(node.val)
if(node.left){
queue.push(node.left)
}
if(node.right){
queue.push(node.right)
}
}
res.push(temp)
}
for(let i =0;i<res.length;i++){
if((i+1)%2 === 0){
res[i] = res[i].reverse()
}
}
return res
};
还有一种解法是就是奇数行正常遍历,偶数行进行从右到左遍历。我们可以设置一个标志位level判断当前行是奇数行还是偶数行,如果是奇数行,那下一行就是偶数行,我们可以将当前节点的子节点从左到右入栈,然后出栈的时候赋给队列就是从右到左。具体代码如下:
function Print(pRoot) {
if (pRoot == null) {
return [];
}
let queue = [],
temp = [],
res = [],
level = 0,
isEven = true,
isBePrinted = 1;
queue.push(pRoot);
while (queue.length) {
let node = queue.shift();
//判断奇、偶行,奇数行
//奇数行直接push(node.val);,偶数行就unshift(node.val)
if (isEven) {
temp.push(node.val);
} else {
temp.unshift(node.val);
}
//行数加一
if (node.left) {
queue.push(node.left);
level++;
}
if (node.right) {
queue.push(node.right);
level++;
}
//当前被操作行数已遍历完
isBePrinted--;
//处理行数、奇偶、temp置空
if (isBePrinted === 0) {
res.push(temp);
temp = [];
isBePrinted = level;
level = 0;
isEven = !isEven;
}
}
return res;
}