题目:从前序与中序遍历序列构造二叉树
题目描述:
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/
9 20
/
15 7
分析:
代码:
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func buildTree(preorder []int, inorder []int) *TreeNode {
if len(preorder)==0 || len(inorder)==0{
return nil
}
if len(preorder)!=len(inorder){
return nil
}
n:=len(preorder)
root:=&TreeNode{
Val:preorder[0],
}
for i:=0;i<n;i++{
if preorder[0]==inorder[i]{
root.Left=buildTree(preorder[1:i+1],inorder[0:i])
root.Right=buildTree(preorder[i+1:n],inorder[i+1:n])
}
}
return root
}
题目:从中序与后序遍历序列构造二叉树
题目描述:
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
返回如下的二叉树:
3
/
9 20
/
15 7
分析:与上题的解题思路没有什么差别,只是后序是左右根的形式,所以每次取后序中的最后一个数据,注意边界问题
代码:
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func buildTree(inorder []int, postorder []int) *TreeNode {
if len(inorder)==0 || len(postorder)==0{
return nil
}
if len(inorder)!=len(postorder){
return nil
}
n:=len(inorder)
root :=&TreeNode{
Val:postorder[n-1],
}
for i:=0;i<n;i++{
if inorder[i]==postorder[n-1]{
root.Left=buildTree(inorder[0:i],postorder[0:i])
root.Right=buildTree(inorder[i+1:n],postorder[i:n-1])
}
}
return root
}