持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第11天,点击查看活动详情
题目
给你一棵二叉树的根节点
root,翻转这棵二叉树,并返回其根节点。
输入: root = [4,2,7,1,3,6,9]
输出: [4,7,2,9,6,3,1]
题解
我们这里可以采用递归的方式进行实现,我们首先判断传入的
root出参是否为空,为空情况下直接返回null,如果不为空则往下执行,下面是一个左右节点互换的操作,我们先声明一个tmp变量,作为互换的中转站,我们先将root出参的左节点存到tmp变量中,然后在将root出参的右节点存到root的左节点中,然后在将tmp变量赋值给root出参的右节点上,以此完成数据的互换操作,然后在将root的左节点作为参数传给自身函数进行执行,也将root的右节点作为参数传给自身函数进行执行,以此完成反转,最后返回root出参
/**
* @param {TreeNode} root
* @return {TreeNode}
*/
var invertTree = function(root) {
if (!root) return null;
let tmp=root.left;
root.left=root.right;
root.right=tmp;
invertTree(root.left);
invertTree(root.right);
return root;
};
我们这里还有另一种思路,先判断
root出参是否为空,为空返回null,然后声明一个stack变量,它是一个栈用于存储root出参,然后我们使用循环,循环的终止条件是stack变量长度为0,在循环内部我们声明一个tmp变量,用于存储反转后的值,然后去循环stack栈,在循环中我们声明一个tmpNode变量,借用该变量完成当前循环值左右节点的互换,然后判断当前值的左节点存在的情况下,就使用push方法添加到tmp变量中,右节点也如此,当循环结束后,我们在将tmp变量赋值给stack变量,最后将root出参返回出去
/**
* @param {TreeNode} root
* @return {TreeNode}
*/
var invertTree = function(root) {
if (!root) return null;
let stack = [root];
while(stack.length!=0) {
const tmp = [];
for (let item of stack) {
let tmpNode = item.left;
item.left = item.right;
item.right = tmpNode;
if (item.left) tmp.push(item.left);
if (item.right) tmp.push(item.right);
}
stack = tmp;
}
return root;
};
坚持努力,无惧未来!