【两次过】Lintcode 1126. 合并两棵二叉树

最新推荐文章于 2022-12-22 14:48:33 发布

原创最新推荐文章于 2022-12-22 14:48:33 发布 · 351 阅读

0 ·

CC 4.0 BY-SA版权

lintcode 同时被 2 个专栏收录

393 篇文章

订阅专栏

二叉树

67 篇文章

订阅专栏

本文详细解析了二叉树合并算法，通过前序遍历的方式，实现两个二叉树节点值的叠加，生成新的二叉树。文章提供了完整的代码示例，帮助读者深入理解算法的实现细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给出两个二叉树，并想象当你把其中一个覆盖另一个时，两棵树的一些节点重叠，而其他节点则不重叠。

您需要将它们合并到一个新的二叉树中。合并规则是，如果两个节点重叠，则将节点值加起来作为合并节点的新值。否则，非空节点将用作新树的节点。

样例

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

注意事项

合并过程必须从两个树的根节点开始。

解题思路：

前序遍历，依次相加即可。

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */

public class Solution {
    /**
     * @param t1: the root of the first tree
     * @param t2: the root of the second tree
     * @return: the new binary tree after merge
     */
    public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
        // Write your code here
        if(t1==null && t2==null)
            return null;
        else if(t1 == null)
            return t2;
        else 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;
    }
}

二刷：

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */

public class Solution {
    /**
     * @param t1: the root of the first tree
     * @param t2: the root of the second tree
     * @return: the new binary tree after merge
     */
    public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
        // Write your code here
        TreeNode root = null;
        
        if(t1 == null && t2 == null)
            return root;
        else if(t1 == null)
            root = new TreeNode(t2.val);
        else if(t2 == null)
            root = new TreeNode(t1.val);
        else
            root = new TreeNode(t1.val + 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;
    }
}