文章目录
- 中序遍历图解
- 总结
中序遍历图解
首先看下中序遍历的代码,其接受一个根结点root
作为参数,判断根节点是否为nil
,不为nil
则先递归遍历左子树。
func traversal(root *TreeNode,res *[]int) {if root == nil {return}traversal(root.Left,res)*res = append(*res,root.Val)traversal(root.Right,res)
}
我们把树根结点A
传递给它,其左结点为B
,右结点为C
。
首先我们要检查root
是否为nil
,其不为nil
,通过递归继续遍历左边的子树,将左子树传递给递归函数,该层递归函数的root
为B
,其左子树为D
右子树为E
,判断root
是否为nil
,root
不为nil
,继续将该树的左子树向下递归,该层递归函数的root
为D
,左右子树都为nil
,我们检查root
是否为nil
,root
为D
,不为nil
,继续递归遍历D
的左子树,其左子树为nil
,所以该层递归函数的root
为nil
,满足递归结束条件,执行return退出该层递归函数
。
递
到此处时的运行栈如下图所示
此时的return
会回到root
为D
的递归层,D
的左子树已经遍历完毕,我们执行下一行语句append
,
该语句会将root
的数据域加入到结构列表res
中,即访问到了D
,如下图
按照顺序继续执行,接下来将使用递归遍历D
的右子树,这里D
的右子树为nil
,所以我们传入的递归参数也为nil
,检测到root
为nil
,我们退出该层递归函数,回到调用层D
,该层的所有语句都执行完毕了。我们继续回到调用它的函数,即B
层的 traversal(root.Left,res)
语句处
继续执行后序语句,执行 *res = append(*res,root.Val)
记录root
的数据域,即访问B
,再执行下一条语句
递归访问B
的右子树,将E
传递给它,判断root
是否为nil
,root
为E
,不为nil
,递归调用E
的左子树,左子树为nil
,判断root
是否为nil
,为nil
,退出该层
执行下一行,将E
记录到结果中,继续遍历E
的右子树,右子树为nil
,直接退出该层递归函数,返回到了E
的递归层,E
这层也执行完毕了,返回到调用它的B
层
随即B
层也执行完了,返回到调用它的A
层 ,在该层执行下一行代码,将A
记录到结果中,继续遍历A
的右子树,A
的右子树为C
,其不为nil
,递归C
的左子树F
,F
不为nil
,递归F
的左子树,F
的左子树为nil
,即传入的root
为nil
,满足递归结束条件,开始回归上层。
返回到调用它的F
层,将F
记录到结果中。遍历F
的右子树,F
的右子树也为nil
,退出该层,到此F
这层函数执行完毕,返回到调用F
的递归层 C
,下一行语句记录C
继续下一行语句,递归C
的右子树G
,判断该层递归的root
是否为nil
,当前root
为G
,不为nil
,递归G
的左子树,左子树为nil
,满足递归结束条件,返回到调用它的G
,输出G
,递归G
的右子树,右子树为nil
满足递归结束条件,返回到调用它的G
,G
这层函数结束,返回上层到C
C
也运行完毕,返回上层到A
,A
也运行完毕,此该树递归结束,这样我们就得到了中序遍历序列
总结
再次回顾一下代码
func traversal(root *TreeNode,res *[]int) {if root == nil {return}traversal(root.Left,res)*res = append(*res,root.Val)traversal(root.Right,res)
}
从我们的遍历全流程图解来看,不难理解,每个节点都是将其左子树全部递归遍历完后,才开始遍历其右子树的,如根节点A
,是将其左子树BDE
全部遍历完后,才开始遍历右子树的,注意思考递归栈哦,这样才能真正的理解。
中序遍历理解后,前序和后续遍历是一样的道理。这时一起看下后续遍历
的代码:
func traversal(root *TreeNode,res *[]int) {if root == nil {return}traversal(root.Left,res)traversal(root.Right,res)*res = append(*res,root.Val)
}
很多同学看到递归左右子树的那两行代码,很容易陷入误区,以为那两行是"同时"在执行,认为遍历完root的左节点后,立马遍历了其右节点,这种理解是非常不对的,实际是遍历完整个左子树后,经过
归
回到当前层,此时才会开始执行当前层的traversal(root.Right,res)
语句去遍历右子树。
尤其是看到两条回溯语句写在同一行时会更容易误解,如力扣104. 二叉树的最大深度
/*** definition for a binary tree node.* type treenode struct {* val int* left *treenode* right *treenode* }*/
func max (a, b int) int {if a > b {return a}return b
}// 递归
func maxdepth(root *treenode) int {if root == nil {return 0}return max(maxdepth(root.left), maxdepth(root.right)) + 1
}
实际最后一行代码是简化版,等价于如下代码
/*** definition for a binary tree node.* type treenode struct {* val int* left *treenode* right *treenode* }*/
func max (a, b int) int {if a > b {return a}return b
}// 递归
func maxdepth(root *treenode) int {if root == nil {return 0}leftDepth := maxdepth(root.left) // 递归左子树的深度rightDepth := maxdepth(root.right) // 递归右子树的深度return max(leftDepth,rightDepth ) + 1 // 当前根节点到叶子节点的最大深度
}
此时再看此代码是不是好理解一些 ,它就是我们的后序遍历
哇!将左子树全部遍历完后,才会开始遍历右子树,最后则是根节点,然后回到上层调用栈,直到所有接地那遍历完毕,递归函数执行完毕。