其他分享
首页 > 其他分享> > 617.合并二叉树

617.合并二叉树

作者:互联网

题目

给你两棵二叉树: root1 和 root2 。想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。返回合并后的二叉树。注意: 合并过程必须从两个树的根节点开始。

示例1

解题思路

//递归
class Solution {
public:
    
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(!root1&&!root2)return 0;//考虑最底层,都为空返回null
        TreeNode* root=new TreeNode(0);
        root->left=mergeTrees(root1?root1->left:0,root2?root2->left:0);//中间,处理root左右子树
        root->right=mergeTrees(root1?root1->right:0,root2?root2->right:0);
      
        if(root1)root->val+=root1->val;
        if(root2)root->val+=root2->val;//上层,更新root的value

        return root;  //返回给更上一层
    }
};
//层序遍历
class Solution {
public:
    TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
        if(root1==nullptr||root2==nullptr){
            return root1==nullptr?root2:root1;
        }
        TreeNode* r1;
        TreeNode* r2;
        queue<TreeNode*> Q;
        Q.push(root1);
        Q.push(root2);

        while(Q.size()>0){//因为这题的BFS不需要区分层数,所以不用采取 sz=Q.size()的方法得到一层中的结点个数从而控制循环次数,这种情况下需要使用两个while
            r1=Q.front(); Q.pop();//r1为树1的结点
            r2=Q.front(); Q.pop();//r2为树2中的结点
            r1->val+=r2->val;
            if(r1->left!=nullptr&&r2->left!=nullptr){//只把需要计算的结点压入队列中
                Q.push(r1->left);
                Q.push(r2->left);
            }
            else if(r1->left==nullptr&&r2->left!=nullptr){
                r1->left=r2->left;//说明 r1->left及其下面都为空,直接将r2->left 下面的整颗子树引用过来即可
            }

            if(r1->right!=nullptr&&r2->right!=nullptr){
                Q.push(r1->right);
                Q.push(r2->right);
            }
            else if(r1->right==nullptr&&r2->right!=nullptr){
                r1->right=r2->right;
            }

        }
        return root1;
        
    }
};

标签:r1,r2,nullptr,合并,617,二叉树,left,root1,root2
来源: https://www.cnblogs.com/IPurpleU/p/16074843.html