合并二叉树。
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
注意: 合并必须从两个树的根节点开始。
自然智慧即可。递归。
代码用golang编写。代码如下:
package main
import "fmt"
func main() {
t1 := &TreeNode{val: 1}
t1.left = &TreeNode{val: 2}
t1.right = &TreeNode{val: 3}
t2 := &TreeNode{val: 1}
t2.left = &TreeNode{val: 3}
t2.right = &TreeNode{val: 2}
t := mergeTrees(t1, t2)
fmt.Println(t.val)
fmt.Println(t.left.val)
fmt.Println(t.right.val)
}
type TreeNode struct {
val int
left *TreeNode
right *TreeNode
}
func NewTreeNode(val int) *TreeNode {
res := &TreeNode{}
res.val = val
return res
}
// 当前,一棵树的头是t1,另一颗树的头是t2
// 请返回,整体merge之后的头
func mergeTrees(t1, t2 *TreeNode) *TreeNode {
if t1 == nil {
return t2
}
if t2 == nil {
return t1
}
// t1和t2都不是空
merge := NewTreeNode(t1.val + t2.val)
merge.left = mergeTrees(t1.left, t2.left)
merge.right = mergeTrees(t1.right, t2.right)
return merge
}
执行结果如下: