LeetCode 617. 合并二叉树

最新推荐文章于 2021-06-14 15:24:56 发布

云胡同学

最新推荐文章于 2021-06-14 15:24:56 发布

阅读量298

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_32046111/article/details/82797438

版权

leetcode 专栏收录该内容

34 篇文章 0 订阅

订阅专栏

题目描述

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

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

示例 1:

输入: 
	Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7

               
输出: 
合并后的树:
	     3
	    / \
	   4   5
	  / \   \ 
	 5   4   7
注意: 合并必须从两个树的根节点开始。

思路

使用递归对二叉树进行合并

如果 t1，t2 两个节点都不为空，直接加起来
如果 t1 节点为空，t2 不为空，return t2
如果 t2 节点为空，t1 不为空，return t1
如果 t1，t2 节点都不为空，return null

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        if(t1 && t2)
		{
			t1->val = t1->val + t2->val;
			t1->left = mergeTrees(t1->left,t2->left);
			t1->right = mergeTrees(t1->right,t2->right);
			return t1;
		}
		else if(t1 == NULL && t2)
		{
			return t2;
		}
		else if(t1 && t2 == NULL)
		{
			return t1;
		}
		else 
		{
			return NULL;
		}
    }
};