合并二叉树-LeetCode

题目:

给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
在这里插入图片描述
注意: 合并必须从两个树的根节点开始。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-two-binary-trees

思路:

  1. 直接使用先序遍历的递归法,对左右子树进行判断和叠加,这里使用了没有影响原二叉树的递归法。
  2. 或者使用队列实现的广度优先的迭代法,直接在原二叉树t1上操作,对满足不为空的的左右子结点进行叠加,否则直接将t2点对应结点进行赋值

代码(递归法):

/**
 * 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 t1==null?t2:t1;
        // 先合并根结点
        TreeNode root = new TreeNode((t1==null?0:t1.val)+(t2==null?0:t2.val));
        // 递归合并左右子树
        root.left = mergeTrees(t1==null?null:t1.left, t2==null?null:t2.left);
        root.right = mergeTrees(t1==null?null:t1.right, t2==null?null:t2.right);
        return root;
    }
}

代码(迭代法):

import java.util.LinkedList;
class Solution {
    public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
        if(t1==null || t2==null) return t1==null?t2:t1;
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(t1);
        queue.add(t2);
        while(queue.size()>0){
            TreeNode r1 = queue.remove();
            TreeNode r2 = queue.remove();
            // 直接在t1上操作,合并值
            r1.val += r2.val;
            // 存入左子树
            if(r1.left!=null && r2.left!=null){
                queue.add(r1.left);
                queue.add(r2.left);
            }
            // 如果左子树为空直接赋值
            if(r1.left==null){
                r1.left = r2.left;
            }
            if(r1.right!=null && r2.right!=null){
                queue.add(r1.right);
                queue.add(r2.right);
            }
            if(r1.right==null){
                r1.right = r2.right;
            }
        }
        return t1;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值