LeetCode 617. 合并二叉树

合并二叉树

题目描述：
给定两个二叉树，想象当你将它们中的一个覆盖到另一个上时，两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠，那么将他们的值相加作为节点合并后的新值，否则不为 NULL 的节点将直接作为新二叉树的节点。

示例 1:

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

题目分析：
合并两棵二叉树，我们以第一棵二叉树为基准，把第二棵树合并到第一棵树上面，我们用递归的方法来合并所有结点。
分三种情况来讨论：

1. 两个结点都不为空，则把第二个结点的值加到第一个结点的值上，然后我们用递归函数来更新第一个结点的左子树，再同理更新右子树（递归函数的返回值为TreeNode*）。
2. 其中一个结点为空，则返回另一个非空结点，例如t1为空，返回t2；t2为空，返回t1.
3. 两个结点都为空，则返回NULL。

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
        if(!t1 && !t2)
            return NULL;
        else if(t1 && !t2)
            return t1;
        else if(!t1 && t2)
            return t2;
        else{
            t1->val += t2->val;
            t1->left = mergeTrees(t1->left, t2->left);
            t1->right = mergeTrees(t1->right, t2->right);
            return t1;
        }
    }
};