Question

问题是：

给定两个二叉树并想象当你把其中一个覆盖另一个时，两棵树的一些节点重叠而其他节点不重叠。

您需要将它们合并到一个新的二叉树中。合并规则是，如果两个节点重叠，则将节点值加起来作为合并节点的新值。否则，NOT null节点将用作新树的节点。

Example 1:
Input: 
    Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7                  
Output: 
Merged tree:
         3
        / \
       4   5
      / \   \ 
     5   4   7

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

我尝试解决这个leetcode问题，但总是得到一个错误的答案。

我的回答是：

**Merged tree:
         3
        / \
       4   5
      /   
     5**

似乎我丢失了节点4和7.

但是，从std :: cout创建了所有节点，但似乎没有构造树。

我非常感谢您对我的代码提出任何意见：

class Solution {

public:

TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if (t1 == NULL && t2 == NULL)
    return NULL;
else if (t1 == NULL && t2 != NULL) {

    t1 = new TreeNode(t2->val);
    cout << "vq1:" << t1->val << endl;

    mergeTrees(t1->left, t2->left);
    mergeTrees(t1->right, t2->right);
}
else if (t1 != NULL && t2 == NULL) {
    t1->val += 0;
    cout << "vu1:" << t1->val << endl;
    mergeTrees(t1->left, NULL);
    mergeTrees(t1->right, NULL);
}
else if (t1 != NULL && t2 != NULL) {
    t1->val += t2->val;
    cout << "vx1:" << t1->val << endl;

    mergeTrees(t1->left, t2->left);
    mergeTrees(t1->right, t2->right);

}
return t1;
}
};

Answer 1

您正在更新节点但不是节点的左右子节点尝试这个，

TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {

    if (t1 == NULL && t2 == NULL)
        return NULL;
    else if (t1 == NULL && t2 != NULL) {

        t1 = new TreeNode(t2->val);
        cout << "vq1:" << t1->val << endl;

        t1->left = mergeTrees(t1->left, t2->left);
        t1->right = mergeTrees(t1->right, t2->right);
    }
    else if (t1 != NULL && t2 == NULL) {
        t1->val += 0;
        cout << "vu1:" << t1->val << endl;
        t1->left = mergeTrees(t1->left, NULL);
        t1->right = mergeTrees(t1->right, NULL);
    }
    else if (t1 != NULL && t2 != NULL) {
        t1->val += t2->val;
        cout << "vx1:" << t1->val << endl;

        t1->left = mergeTrees(t1->left, t2->left);
        t1->right = mergeTrees(t1->right, t2->right);

    }
    return t1;
}

LeetCode＃617“使用C ++合并两个二叉树”

1 个答案: