LeetCode - Merge Two Binary Tree

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.

大体意思是:将两个二叉树合并,相同位置的节点值相加

我的思路:遍历两个树,将相同位置的值相加,重组新的树。

我的代码
public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
        if(t1==null&&t2==null){
            return null;
        }
        TreeNode newTree = null;
        if(t1==null){
            newTree = new TreeNode(t2.val);
            newTree.left = mergeTrees(null, t2.left);
            newTree.right = mergeTrees(null, t2.right);
        }else if(t2==null){
            newTree = new TreeNode(t1.val);
            newTree.left = mergeTrees(t1.left, null);
            newTree.right = mergeTrees(t1.right, null);
        }else{
            newTree = new TreeNode(t2.val+t1.val);
            newTree.left = mergeTrees(t1.left, t2.left);
            newTree.right = mergeTrees(t1.right, t2.right);
        }

        return newTree;
    }

思路没什么问题,但是在判断层面做了一些多余的问题。比如:在t1==null之后直接返回t2就可以,不必要再做其他操作。

更改后代码:

public TreeNode mergeTrees1(TreeNode t1, TreeNode t2) {
        if (t1 == null) {
            return t2;
        }
        if (t2 == null) {
            return t1;
        }
        TreeNode newTree = new TreeNode(t2.val + t1.val);
        newTree.left = mergeTrees(t1.left, t2.left);
        newTree.right = mergeTrees(t1.right, t2.right);

        return newTree;
    }

更改后的代码更简洁,运行速度也提升了。

总结:问题要想得透彻点,不急于下手写代码。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值