欢迎来我的个人博客浏览这篇文章哦~(还有背景音乐可以听!)点这里!
最近在刷代码随想录,于是碰到了这道题,翻转二叉树理应很简单,只要在遍历时交换每个节点的左右孩子各一次就行了,但卡哥确说前中后与层序遍历,其中唯独中序遍历不方便,会将某些节点翻转两次,于是我盯着二叉树的图一直在脑中模拟中序遍历的翻转,遍历了几遍发现,这不是没啥问题吗?
直到我看到了某位仁兄写的中序遍历代码,才恍然大悟,问题不是出在思路上,而是代码的具体实现上,果然实践才能出真知啊!
代码如下
class Solution {
public TreeNode invertTree(TreeNode root) {
if (null == root) {
return root;
}
invertTree(root.left);
TreeNode tmp = root.left;
root.left = root.right;
root.right = tmp;
invertTree(root.left);
return root;
}
}
“没错,两次递归调用的都是当前节点的左孩子。”
我们假设当前节点为root,并且我们已经递归完root.left了,那接下来就会交换root.left与root.right,再接下来我们应该就要递归“原来”的root.right了,但由于root.left与root.right已经交换过位置,所以“原来”的root.right就是现在的root.left。
下面是代码随想录的链接:
0226.翻转二叉树
以及启发我的大哥的链接:
力扣226.翻转二叉树(中序遍历)