1002. 合并二叉树

最新推荐文章于 2024-08-30 16:50:50 发布

jzylf

最新推荐文章于 2024-08-30 16:50:50 发布

阅读量314

点赞数

CC 4.0 BY-SA版权

分类专栏： algorithms 文章标签： algorithms Sicily

本文链接：https://blog.youkuaiyun.com/jzylf/article/details/79000986

algorithms 专栏收录该内容

26 篇文章

订阅专栏

本文介绍了一种合并两棵二叉树的算法，通过将相同位置的节点值相加来生成新的二叉树。提供了完整的代码实现，并说明了输入输出的具体形式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Problem:

输入两个二叉树T1和T2，要求对T1和T2进行合并. 合并是指将二叉树同一位置节点上的数求和，作为合并后二叉树相应位置节点的数值. 如果某个位置上只有一个二叉树有节点，则合并后的二叉树对应位置上的节点上的数值就等于这个节点上的数值.

例如：

T1 T2

1 2

/ \ / \

3 2 1 3

/ \ \

5 4 7

合并后：

/ \

4 5

/ \ \

5 4 7

T1和T2的结点数均不多于100000，每个结点的数值在1和1000之间.

请为下面的Solution类实现解决上述问题的merge函数，函数的两个参数T1和T2分别代表两个二叉树的根节点，函数返回值为合并后的二叉树的根节点.

/**

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* merge(TreeNode* T1, TreeNode* T2) {

}

};

注意：你只需要提交Solution类的代码，你在本地可以编写main函数测试程序，但不需要提交main函数的代码，也不需要提交TreeNode的定义. 注意不要修改类和函数的名称.

Solution:

class Solution {
public:
    TreeNode* merge(TreeNode* T1, TreeNode* T2) {
        TreeNode* T3;
        if (T1 != NULL && T2 != NULL)
        {
            T3 = new TreeNode(T1->val + T2->val);
            T3->left = merge(T1->left, T2->left);
            T3->right = merge(T1->right, T2->right);
        }
        if(T1 == NULL && T2 != NULL)
        {
            T3 = new TreeNode(T2->val);
            T3->left = merge(NULL, T2->left);
            T3->right = merge(NULL, T2->right);
        }
        if (T1 != NULL && T2 == NULL)
        {
            T3 = new TreeNode(T1->val);
            T3->left = merge(T1->left, NULL);
            T3->right = merge(T1->right,NULL);
        }
        if (T1 == NULL && T2 == NULL)
        {
            T3 = NULL;
        }
        return T3;
    }
};