题目
给你一棵二叉树的根节点
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; }