[ 二叉树 ] 合并二叉树

最新推荐文章于 2025-03-22 15:53:38 发布

原创最新推荐文章于 2025-03-22 15:53:38 发布 · 135 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树

二叉树专栏收录该内容

25 篇文章

订阅专栏

这篇博客介绍了如何在LeetCode上的617题中合并两个二叉树。作者提供了两种解决方案，分别是递归和层次遍历的方法。递归方案通过直接相加节点值并递归处理子节点来合并树；非递归方案则使用队列进行层次遍历，处理节点并更新子节点结构。这两种方法都能有效地合并两棵树，确保没有子树对当前树的影响。

617. 合并二叉树 - 力扣（LeetCode） (leetcode-cn.com)

合并二叉树

递归

不存在子树对当前树的影响，遍历方式任意

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if (!root1) return root2;
        if (!root2) return root1;
    	// 注意返回值可能是原root1， 可以是root2
    	// 所以每次需要接收返回值，不能省略
        root1->val += root2->val;
        root1->left = mergeTrees(root1->left, root2->left);
        root1->right = mergeTrees(root1->right, root2->right);
        return root1;
    }
};

非递归

层次遍历

class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        // 将root2 加到 root1 上 完成 合并
        if (!root1) return root2;
        if (!root2) 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;

            // 如果root1和root2 的子节点都存在，那么正常入列
            if (node1->left && node2->left) {
                que.push(node1->left);
                que.push(node2->left);
            }
            if (node1->right && node2->right) {
                que.push(node1->right);
                que.push(node2->right);
            }
            // 如果root1孩子不存在 而root2的孩子存在，那么把root2的孩子变为root1的孩子
            if (!node1->left && node2->left) node1->left = node2->left;
            if (!node1->right && node2->right) node1->right = node2->right;

            // 至于root1孩子存在，而root2孩子不存在的情况，则不需要操作
        }
        return root1;
    }
};