今天这道题还是有点思路的…(难得)
题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
输入描述
前序遍历序列{1,2,4,7,3,5,6,8}
中序遍历序列{4,7,2,1,5,3,8,6}
输出描述
重建好的二叉树。
题目描述
这道题之所以简单,是因为在这里并未出现重复的数字,而且,对于这一类题目,直接考虑使用递归的方法解决即可。既然要重建树,那么首先肯定得有树的样子:
type TreeNode struct {
Val int
Left *TreeNode
Right *TreeNode
}
既然是使用递归方法,那么第一步,我们要找到递归的出口是什么,对于这道题,递归的主要作用就是根据先序遍历和中序遍历找到每个节点应在的位置,再通过TreeNode
的数据结构将其连接起来。那么当执行递归的数组为空的时候,即可以人为所有元素都找到了自己的位置,此时即为递归的出口:
func RebuildTree(preIndex, inIndex []int) *TreeNode {
if len(preIndex) != len(inIndex) || len(preIndex) == 0 {
return nil
}
rootVal := preIndex[0]
// 从中序遍历中找到根节点所在位置
for i:=0; i<len(inIndex); i++ {
if inIndex[i] == rootVal {
rootIndex == i
}
}
// 执行递归
left := RebuildTree(preIndex[1:rootIndex+1], inIndex[:rootIndex])
right := RebuildTree(preIndex[rootIndex+1:], inIndex[rootIndex+1:])
return &TreeNode{rootVal, left, right}
}
这样就可以实现对树的重建操作了。