递归算法的关键
**首先:明确函数定义,相信这个定义,不要**跳进递归。
**其次:**搞明白root节点自己要做什么,然后根据题目要求选择前中后序递归框架。
**最后:**思考出每个节点需要做什么,刷题。
习题1:翻转二叉树
首先,明确函数定义
即,交换根节点的左右节点位置,返回翻转后的节点。
其次,明确root自己做什么
他要交换它的两个子节点,故选择前序遍历
再者,明确子节点都要做什么
让每个子节点都去翻转他们的左右子节点。
最后,明确边界条件
当root == null时,证明执行到了叶子节点,终止递归,直接返回。
TreeNode invertTree(TreeNode root) {
//判断root是否为空,为空直接返回。
if (root == null) {
return root;
}
//选择前序遍历
TreeNode temp = root.left;
root.left = root.right;
root.right = temp;
//左右节点都要干什么,他们要交换他们的左右节点
invertTree(root.left);
invertTree(root.right);
//返回翻转后的树,根节点
return root;
}
习题2:填充每个二叉树的右侧指针
![img](https://assets.leetcode.com/uploads/2019/02/14/116_sample.png)
二叉树问题的难点就在于,如何把题目的要求细化成每个节点要做的事情。
试探思路1:
像上一题一样,将任务分给一个节点,让他的左子节点指向右子节点,右子节点指向空,但是我们很快就会发现一个问题,如果两个相邻子节点并不属于一个双亲结点,那么他们就无法相连。这也是递归的难点,如何细化每个节点要做的事情。
解决思路:
细化节点任务,让两个节点承担任务。
明确root自己做什么
root 自己将自己的左子节点和右子节点连接起来。
左子节点和右子节点都要将他们的子节点连接起来,并且,也要将不属于一个根节点的同层节点连接起来。
明确子节点要做什么
子节点做和父节点一样的事。
Node connect(Node root) {
//临界条件,若root = null 返回root
if (root == null) {
return root;
}
connectTwoNode(root.left,root.right);
return root;
}
void connectTwoNode(Node node1,Node node2) {
//只要node1 和 node2 有一个是空的,不需要再指向,直接保持原状指向null即可。
if (node1 == null || node2 == null) {
}
//前序遍历,将node1 指向 node2
node1.next = ndoe2;
//递归将 node1 的 左子节点指向它的右子节点
connectTwoNode(node1.left,node2.left);
//将 node2 的 左子节点指向有字节点
connectTwoNode(node2.left,node2.right);
//将 node1 的右子节点指向 node2 的左子节点,解决不同根节点的问题。
connectTwoNode(node1.right,node2.left);
}
习题3:将二叉树展开为链表
![img](https://assets.leetcode.com/uploads/2021/01/14/flaten.jpg)
解题思路:
首先,明确递归函数定义
目标,将它的左右子树转化为一条链表,让左子树代替根节点的右子树,将原右子树链到新右子树的后面。
传入一个节点,函数里对这个节点的左右子树进行链表化。所以采用后序遍历。
明确root自己做什么
将它的左右子节点向上文所说的那样进行合并。
明确边界条件
root == null 返回root
左子树为null 直接返回右子树
右子树为null 直接链入左子树
public void flatten(TreeNode root) {
if (root == null) {
return;
}
//后序遍历,先将根节点的左右子树链表化
flatten(root.left);
faltten(root.right);
//记录链表化的左右子树
TreeNode left = root.left;
TreeNode right = root.right;
//将左节点置空
root.left = null;
//用左子树替换掉右子树
root.right = left;
TreeNode p = root;
//将左子树作为根节点的右子树,将右子树链到原左子树的后面
while (p.right != null) {
p = p.right;
}
p.right = right;
}
总结:
递归算法的关键要明确函数的定义,相信这个定义,而不要跳进递归细节。
写二叉树的算法题,都是基于递归框架的,我们先要搞清楚 root
节点它自己要做什么,然后根据题目要求选择使用前序,中序,后续的递归框架。
二叉树题目的难点在于如何通过题目的要求思考出每一个节点需要做什么,这个只能通过多刷题进行练习了。
如果本文讲的三道题对你有一些启发,请三连,数据好的话东哥下次再来一波手把手刷题文,你会发现二叉树的题真的是越刷越顺手,欲罢不能,恨不得一口气把二叉树的题刷通。
原文链接:https://labuladong.gitee.io/algo/2/18/21/