1002. 合并二叉树

Problem:

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

例如:
         T1                        T2
 
          1                         2                            
         /  \                       /  \                           
        3   2                     1   3                       
       /                            \    \                      
      5                             4   7  
合并后:
            3
           /  \
          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;
    }
};                        

实现如下: ```c #include <stdio.h> #include <stdlib.h> struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; }; struct TreeNode* mergeTrees(struct TreeNode* t1, struct TreeNode* t2) { if (t1 == NULL) return t2; if (t2 == NULL) return t1; t1->val += t2->val; t1->left = mergeTrees(t1->left, t2->left); t1->right = mergeTrees(t1->right, t2->right); return t1; } int main() { // 构造两棵树 struct TreeNode *t1 = (struct TreeNode*)malloc(sizeof(struct TreeNode)); struct TreeNode *t2 = (struct TreeNode*)malloc(sizeof(struct TreeNode)); t1->val = 1; t1->left = (struct TreeNode*)malloc(sizeof(struct TreeNode)); t1->left->val = 3; t1->left->left = (struct TreeNode*)malloc(sizeof(struct TreeNode)); t1->left->left->val = 5; t1->left->left->left = NULL; t1->left->left->right = NULL; t1->left->right = NULL; t1->right = (struct TreeNode*)malloc(sizeof(struct TreeNode)); t1->right->val = 2; t1->right->left = NULL; t1->right->right = NULL; t2->val = 2; t2->left = (struct TreeNode*)malloc(sizeof(struct TreeNode)); t2->left->val = 1; t2->left->left = NULL; t2->left->right = (struct TreeNode*)malloc(sizeof(struct TreeNode)); t2->left->right->val = 4; t2->left->right->left = NULL; t2->left->right->right = NULL; t2->right = (struct TreeNode*)malloc(sizeof(struct TreeNode)); t2->right->val = 3; t2->right->left = NULL; t2->right->right = (struct TreeNode*)malloc(sizeof(struct TreeNode)); t2->right->right->val = 7; t2->right->right->left = NULL; t2->right->right->right = NULL; // 合并两棵树 struct TreeNode *t = mergeTrees(t1, t2); // 输出合并后的树 printf("%d\n", t->val); printf("%d %d\n", t->left->val, t->right->val); printf("%d %d %d %d\n", t->left->left->val, t->left->right->val, t->right->left->val, t->right->right->val); return 0; } ``` 该程序的输出结果为: ``` 3 4 5 5 4 7 0 ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值