【LeetCode 617】合并二叉树

最新推荐文章于 2024-09-20 15:09:03 发布

CUG-GZ

最新推荐文章于 2024-09-20 15:09:03 发布

阅读量157

点赞数

分类专栏： LeetCode 文章标签：二叉树算法 leetcode 数据结构 JavaScript

本文链接：https://blog.csdn.net/qq_42033567/article/details/108755457

版权

LeetCode 专栏收录该内容

105 篇文章 9 订阅

订阅专栏

题目描述

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

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

示例 1:

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

注意: 合并必须从两个树的根节点开始。

解题思路

我们这里可以使用递归的方式来计算，保持t1不便，将t2的节点往t1上加就可以了。

时间复杂度为O(n)

代码实现

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} t1
 * @param {TreeNode} t2
 * @return {TreeNode}
 */
var mergeTrees = function(t1, t2) {
    if(!t1 && t2){
        return t2
    }
    if(t1 && !t2 || !t1 && !t2){
        return t1
    }

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