2021-12-20:合并二叉树。 给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。 你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。 注意: 合并必须从两个树的根节点开始。 力扣617。

答案2021-12-20:

自然智慧即可。递归。

代码用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
}

执行结果如下: 图片


左神java代码