树的前中后序是递归的重要应用,也是后面大量高频算法问题的基础,我们这里就来好好研究一下
关卡名 | 理解树的前中后序遍历,以及如何体现递归思想的 | 我会了✔️ | |
---|---|---|---|
内容 | 1.理解树的前中后序遍历具体是如何工作的 | √ | |
2.如果没理解,就画图再理解一下 | √ |
1. 深入理解前中后序遍历
深度优先遍历有前中后三种情况,大部分人看过之后就能写出来,很遗憾大部分人只是背下来的,稍微变换一下就废了。
我们再从二叉树的角度看递归,每次遇到递归,都按照前面说的四步来写,可以更好地写出正确的递归算法。通过二叉树可以非常方便的理解递归,递归只处理当前这一层和下一层之间的关系,并不关心下层和下下层之间的关系,这就像老子只管养好儿子,至于孙子怎么样,那是儿子的事,你也不能瞎掺合。具体来说,我们总结了四条行之有效的方法来分解递归:
●第一步:从小到大递推
●第二步:分情况讨论,明确结束条件
●第三步:组合出完整方法
●第四步:想验证,则从大到小画图推演
我们接下来就一步步看怎么做:
第一步:从小到大递推,分情况讨论
我们就以这个二叉树为例:
3
/
9 20
/
15 7
我们先选一个最小的子树:
20
/
15 7
假如20为head,则此时前序访问顺序应该是:
void visti1(){
list.add(root);//20被访问
root.left; // 继续访问 15
root.right; // 继续访问 7
}
然后再向上访问,看node(3)的情况:
void visit2(){
list.add(root);//3被访问
root.left;//继续访问,得到9
root.right; //继续访问,得到20
}
这里的20是一个子树的父节点,访问方式与上面的visit1()一样,所以我们可以直接合并到一起就是:
void visit2(){
list.add(root);//3被访问
root.left;//继续访问,得到9
root.right; //继续访问,得到20
}
这就是我们期待的递归方法。
第二步:分情况讨论,明确结束条件
上面有了递归的主体,但是这个递归什么时候结束呢?很明显应该是root=null的时候。一般来说链表和二叉树问题的终止条件都包含当前访问的元素为null。有些题目结束条件比较复杂,此时最好的方式就是先将所有可能的结束情况列举出来,然后整理一下就行了,这个我们后面在具体题目里再看。
第三步:组合出完整方法
到此为止,我们就能将完整代码写出来了,同时为了方便区分,我们将方法名换成preorder:
public void preorder(TreeNode root, List res) {
if (root == null) {
return;
}
res.add(root.val);
preorder(root.left, res);
preorder(root.right, res);
}
第四步 从大到小 画图推演写完之后对不对呢?递归的方法是很难调试的,即使对的,你也可能晕,这里介绍一种简单有效的验证方法——调用过程图法。我们可以画个过程图看一下,因为是两个递归函数,如果比较复杂,我们可以少画几组。
递归的特征是“不撞南墙不回头”,一定是在执行到某个root=null了才开始返回,下图中的序号就是递归的完整过程:
从图中可以看到,当root的一个子树为null的时候还是会执行递归,进入之后发现root==null了,然后就开始返回。这里我们要特别注意res.add()的时机对不对,将其进入顺序依次写出来就是需要的结果。该过程明确之后再debug就容易很多,刚开始学习递归建议多画几次,熟悉之后就不必再画了。
前序遍历写出来之后,中序和后序遍历就不难理解了,中序是左中右,后序是左右中。代码如下:
public static void inOrderRecur(TreeNode head) {
if (head == null) {
return;
}
inOrderRecur(head.left);
System.out.print(head.value + " ");
inOrderRecur(head.right);
}
再看后序的:
public static void postOrderRecur(TreeNode head) {
if (head == null) {
return;
}
postOrderRecur(head.left);
postOrderRecur(head.right);
System.out.print(head.value + " ");
}
另外需要注意一点,面试,以及LeetCode里提供的方法可能不能直接用来递归,需要我们再创建一个方法,例如:
LeetCode144 二叉树前序遍历问题,此时给的函数preorderTraversal()就难以直接递归,我们可以自己再创建一个:
class Solution {
public:
void preorder(TreeNode *root, vector &res) {
if (root == nullptr) {
return;
}
res.push_back(root->val);
preorder(root->left, res);
preorder(root->right, res);
}
vector preorderTraversal(TreeNode *root) {
vector res;
preorder(root, res);
return res;
}
};
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List preorderTraversal(TreeNode root) {
List res = new ArrayList();
preorder(root,res);
return res;
}
public void preorder(TreeNode root,List res){
if(root == null){
return;
}
res.add(root.val);
preorder(root.left,res);
preorder(root.right,res);
}
}
class Solution:
def preorderTraversal(self, root) :
def preorder(root):
if not root:
return
res.append(root.val)
preorder(root.left)
preorder(root.right)
res = list()
preorder(root)
return res
2. 通关文牒
本关讲解了树的前中后序遍历的基本问题,难度不算大,看懂即过关,先热身,后面有大招等着你