力扣(LeetCode)- 617 - 合并二叉树(简单)


1. 描述

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

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

2. 示例

输入:
  Tree 1                          Tree 2
       1                                2
      /  \                              /  \
    3   2                           1    3
   /                                   \      \
  5                                    4     7

输出:
合并后的树:
           3
         /   \
       4     5
     /   \      \
   5     4      7

3. 分析

二叉树的定义是递归进行的,所以对它的操作也是递归进行的。

在示例中,Tree 1的根节点 “1” 和 Tree 2 的根节点 “2” 合并后,得到合并树的根节点 “3”, 也就是 Tree 1的根 “1” 加上 Tree 2 的根 “2”。

然后处理 Tree 1 的根节点 “1” 的左子树和 Tree 2 的根节点 “2” 的左子树;
最后处理 Tree 1 的根节点 “1” 的右子树和 Tree 2 的根节点 “2” 的右子树。

不过需要注意的是,如果两棵子树的根节点为空,则返回空;
如果某棵子树的某个节点为空,则返回另外一棵子树的对应节点;
否则的话,需要将根节点的值相加,将相加的结果放到一棵子树上(或者新建一棵树),递归地对左子树和右子树进行同样的处理。

4. 代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    
    
    public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
    
    
    	// 如果两棵子树的根节点均为空,则返回空
        if(t1 == null && t2 == null)
            return null;
        // 若某棵子树的根节点为空,则返回不为空的根节点
        else if(t1 == null)
            return t2;
        else if(t2 == null)
            return t1;
        // 否则,两棵子树的根节点均不为空,则将值相加,放到 Tree 1 树中
        //递归地对左子树和右子树进行同样的处理
        else
        {
    
    
            t1.val += t2.val;
            t1.left = mergeTrees(t1.left, t2.left);
            t1.right = mergeTrees(t1.right, t2.right);
        }
        return t1;
    }
}

5. 验证

在这里插入图片描述

6. 出处

  1. LeeCode 617. 合并二叉树
    来源:力扣(LeetCode)
    链接:https://leetcode-cn.com/problems/merge-two-binary-trees
    著作权归领扣网络所有。

猜你喜欢

转载自blog.csdn.net/PursueLuo/article/details/108746273
今日推荐