Merge Two Binary Trees

LeetCode Merge Two Binary Trees

先看题目:
Given two binary trees and imagine that when you put one of them to cover the other, some nodes of the two trees are overlapped while the others are not.

You need to merge them into a new binary tree. The merge rule is that if two nodes overlap, then sum node values up as the new value of the merged node. Otherwise, the NOT null node will be used as the node of new tree.

题目意思是说给定两个二叉树,将这两个二叉树合并成一个二叉树输出,合并原则如果同一节点都非空,则该节点的值等于原来两个二叉树该节点只之和。

Example 1:

Input: 
Tree 1                     Tree 2                  
      1                         2                             
     / \                       / \                            
    3   2                     1   3                        
   /                           \   \                      
  5                             4   7    

Output: 
Merged tree:
     3
    / \
   4   5
  / \   \ 
 5   4   7

解决这道题的思路就是先遍历两个二叉树,然后根据节点情况(有一个数为空,都不为空,都为空)进行对应操作,都为空即完成了整个遍历,将合并后的树返回即可。需要注意的是java中的树类型需要自己定义,该题初始代码已经定义,那么直接用了。具体代码如下:

/**
 * 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){
            return t2;
        }
        if(t2==null){
            return t1;
        }
        t1.val += t2.val;
        t1.left = mergeTrees(t1.left,t2.left);
        t1.right = mergeTrees(t1.right,t2.right);
        return t1;
    }
}

写一个测试类:

public class MergeTwoBinaryTrees {

public static void main(String[] args) {
    // TODO Auto-generated method stub
    //构造Tree 1
    TreeNode t1 = new TreeNode(1);
    t1.left = new TreeNode(3);
    t1.right = new TreeNode(2);
    t1.left.left = new TreeNode(5);

    //构造Tree 2
    TreeNode t2 = new TreeNode(2);
    t2.left = new TreeNode(1);
    t2.right = new TreeNode(3);
    t2.left.right = new TreeNode(4);
    t2.right.left = new TreeNode(7);

    mergeTrees(t1, t2);
    printTree(t1);
    //System.out.println(t1.val);


}

    public static TreeNode mergeTrees(TreeNode t1, TreeNode t2){

        if(t1==null){
            return t2;
        }
        if(t2==null){
            return t1;
        }
        t1.val += t2.val;
        t1.left = mergeTrees(t1.left,t2.left);
        t1.right = mergeTrees(t1.right,t2.right);
        return t1;
    }

    public static void printTree(TreeNode t){
        if(t != null){
            System.out.print(t.val+" ");
            printTree(t.left);
            printTree(t.right);
        }
    }   
}

输出结果为3 4 5 4 5 7 ,正确。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值