LeetCode 617 合并二叉树 HERODING的LeetCode之路

给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。

在这里插入图片描述

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

解题思路:
将两个二叉树合二为一,最简单的方法就是在构建一个什么都没有的二叉树,然后将已有的两个二叉树插入进去,可以用到递归的思想,在往后递归时,要注意判断,如果出现节点为null,那么左右子树一定不存在,就无法继续递归,所以要进行一个判断,代码如下:

/**
 * 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* mergeTrees(TreeNode* t1, TreeNode* t2) {
    
    
        if(!t1 && !t2){
    
    
            return NULL;
        }
        TreeNode * root = new TreeNode(0);
        if(t1){
    
    
            root->val += t1->val;
        }
        if(t2){
    
    
            root->val += t2->val;
        }
        // 注意这里的判断
        root->left = mergeTrees(t1?t1->left:NULL, t2?t2->left:NULL);
        root->right = mergeTrees(t1?t1->right:NULL, t2?t2->right:NULL);
        return root;
    }
};

猜你喜欢

转载自blog.csdn.net/HERODING23/article/details/108745170
今日推荐