【leetcode】617. 合并二叉树（merge-two-binary-trees）（dfs）[简单]

链接

https://leetcode-cn.com/problems/merge-two-binary-trees/

耗时

解题：25 min
题解：5 min

题意

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

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

思路

按规则合并当前的两个节点，再依相同规则合并节点的左右子树。唯一需要注意的是当前节点为空时的处理，详见代码。

时间复杂度：$O(n)$

AC代码

/**
 * 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 == NULL && t2 == NULL) return NULL;
        
        TreeNode *t1l = NULL, *t1r = NULL, *t2l = NULL, *t2r = NULL;
        int t1v = 0, t2v = 0;
        if(t1 != NULL) {
            t1l = t1->left;
            t1r = t1->right;
            t1v = t1->val;
        }
        if(t2 != NULL) {
            t2l = t2->left;
            t2r = t2->right;
            t2v = t2->val;
        }
        
        TreeNode* root = new TreeNode(t1v+t2v);
        root->left = mergeTrees(t1l, t2l);
        root->right = mergeTrees(t1r, t2r);
            
        return root;
    }
};