Leetcode：617. 合并二叉树（C++）

news2025/1/12 23:00:32

问题描述：

实现代码与解析：

递归：

原理思路：

迭代：

原理思路：

问题描述：

给你两棵二叉树： root1 和 root2 。

想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。

返回合并后的二叉树。

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

示例 1：

输入：root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出：[3,4,5,5,4,null,7]

示例 2：

输入：root1 = [1], root2 = [1,2]
输出：[2,2]

实现代码与解析：

递归：

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) 
    {
        //都为空的时候
        if(root1==NULL&&root2==NULL) return NULL;
        //root1为空时，root2为空时
        if(root1==NULL) return root2;
        if(root2==NULL) return root1;
        root2->val+=root1->val;//都不为空时
        root2->left=mergeTrees(root1->left,root2->left);
        root2->right=mergeTrees(root1->right,root2->right);
        return root2;
    }
};

原理思路：

简单题，用递归来做，首先就是来判断结点合并的情况，显然只有三种情况：

1、两个树对应结点都为空，我们直接返回空即可。

2、两个树对应结点只有一个为空，我们就返回不空的那个结点。

3、两个树对应结点都不为空，我们返回两个结点相加的结点。

然后递归合并每个结点即可。

因为判断其中一个结点为空时的代码也同时解决了两个结点都为空的情况，所以下面这行的判断是可以不用写的：

//都为空的时候
if(root1==NULL&&root2==NULL) return NULL;

还有其实这里我们直接用了第二棵树做为返回（第一棵树也可以），也可以创建新结点来作为返回，如下：

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) 
    {
        //都为空的时候
        if(root1==NULL&&root2==NULL) return NULL;
        //root1为空时，root2为空时
        if(root1==NULL) return root2;
        if(root2==NULL) return root1;
        TreeNode* root=new TreeNode(root1->val+root2->val);//新建的结点
        root->left=mergeTrees(root1->left,root2->left);
        root->right=mergeTrees(root1->right,root2->right);
        return root;
    }
};

但其实，里面还是包含了原先两个树的部分引用，因为我们这里是用的指针连接，在我们合并时的第二种情况下（一个对应结点为空时），不空的结点直接返回的原结点，其连接的结点也一同被连接过去了，我们这里只给需要相加的结点创建了新结点。

迭代：

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if (root1 == NULL) return root2;
        if (root2 == NULL) return root1;
        queue<TreeNode*> que;
        que.push(root1);
        que.push(root2);
        while(!que.empty()) {
            TreeNode* node1 = que.front(); que.pop();
            TreeNode* node2 = que.front(); que.pop();
            node1->val += node2->val;//val相加
            // 若两棵树左节点都不为空，加入队列
            if (node1->left != NULL && node2->left != NULL) 
            {
                que.push(node1->left);
                que.push(node2->left);
            }
            // 若两棵树右节点都不为空，加入队列
            if (node1->right != NULL && node2->right != NULL) 
            {
                que.push(node1->right);
                que.push(node2->right);
            }
            // 若t1的左节点 为空 t2左节点不为空，就赋值过去
            if (node1->left == NULL && node2->left != NULL) 
            {
                node1->left = node2->left;
            }
            // 当t1的右节点 为空 t2右节点不为空，就赋值过去
            if (node1->right == NULL && node2->right != NULL) 
            {
                node1->right = node2->right;
            }
        }
        return root1;
    }
};