开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第30天,点击查看活动详情
题目
给你一棵二叉树的根节点
root,翻转这棵二叉树,并返回其根节点。
输入: root = [4,2,7,1,3,6,9]
输出: [4,7,2,9,6,3,1]
思路一
我们可以使用递归进行实现,先判断当前形参root是否为null,如果是则直接返回null,否则就继续往下执行,我们在创建一个left变量和right变量,用于接受调用自身函数传值为root左节点和root右节点的返回值,然后我们将返回值重新赋值给形参root的左节点和形参root的右节点,最后返回root形参即可
/**
* @param {TreeNode} root
* @return {TreeNode}
*/
var invertTree = function(root) {
if (root == null) {
return null
}
let left = invertTree(root.left)
let right = invertTree(root.right)
root.left = right
root.right = left
return root
};
思路二
我们这里可以使用到深度优先搜索,用栈的另一种写法的后序遍历来实现,我们把要处理的节点放入栈后然后放入一个空指针作为标记,如果出栈元素为空,则说明该节点是需要处理的节点,我们先创建一个st变量,他是一个空栈,然后再判断当前的root是否不等于null,如果不是则将root使用push方法添加到st栈中,然后使用循环进行变量,循环条件为当前的st变量的长度不为0,在循环中我们声明一个node变量用于存储栈顶的元素,然后判断node是否为null,如果不是null,就不是需要处理的节点,然后按照中子节点,右子节点,左子节点的顺序使用push方法添加到st栈中,如果为空则删除空节点,然后再删除需要处理的节点,在交换其左右子树,最后将形参root返回出去即可
var invertTree = function(root) {
var st = [];
if (root !== null) st.push(root);
while (st.length !== 0) {
var node = st[st.length - 1];
if (node !== null) {
st.pop();
st.push(node);
st.push(null);
if (node.right) st.push(node.right);
if (node.left) st.push(node.left);
} else {
st.pop();
node = st.pop();
var temp = node.left;
node.left = node.right;
node.right = temp;
}
}
return root;
}