617. 合并二叉树 简单

给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。

示例 1:

输入: 
    Tree 1                     Tree 2
          1                         2
         / \                       / \
        3   2                     1   3
       /                           \   \
      5                             4   7
输出: 
合并后的树:
         3
        / \
       4   5
      / \   \ 
     5   4   7

注意: 合并必须从两个树的根节点开始。

代码参考:

package main

import "fmt"

func main() {
    t1 := &TreeNode{
        Val:   1,
        Left:  &TreeNode{Val: 3, Left: &TreeNode{Val: 5}},
        Right: &TreeNode{Val: 2},
    }
    t2 := &TreeNode{
        Val:   2,
        Left:  &TreeNode{Val: 1, Right: &TreeNode{Val: 4}},
        Right: &TreeNode{Val: 3, Right: &TreeNode{Val: 7}},
    }
    t3 := mergeTrees(t1, t2)
    fmt.Println(t3.Val) // 3
    // also ok
}

// DP
func mergeTrees(t1 *TreeNode, t2 *TreeNode) *TreeNode {
    if t1 == nil {
        return t2
    }
    if t2 == nil {
        return t1
    }
    return &TreeNode{
        Val:   t1.Val + t2.Val,
        Left:  mergeTrees(t1.Left, t2.Left),
        Right: mergeTrees(t1.Right, t2.Right),
    }
}
最后编辑: kuteng  文档更新时间: 2021-06-05 10:16   作者:kuteng