力扣617题:合并二叉树
题目描述
给你两棵二叉树: root1 和 root2 。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
输入输出样例
输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出:[3,4,5,5,4,null,7]
输入:root1 = [1], root2 = [1,2]
输出:[2,2]
Tips
- 两棵树中的节点数目在范围
[0, 2000]
内 -104 <= Node.val <= 104
解法一:使用递归DFS
//使用递归进行实现,深度优先搜索
TreeNode *mergeTrees(TreeNode*root1,TreeNode*root2)
{
TreeNode *res=dfs(root1,root2);
return res;
}
TreeNode * dfs(TreeNode *root1,TreeNode *root2)
{
if(!root1)
{
return root2;
}
if(!root2)
{
return root1;
}
TreeNode *tempNode=new TreeNode(root1->val+root2->val);
tempNode->left=dfs(root1->left,root2->left);
tempNode->right=dfs(root1->right,root2->right);
return tempNode;
}
解法二:使用迭代BFS
//使用迭代进行实现,广度优先搜索
//当两棵树的左节点不为null,就将他们放入队列中
//只要两棵树的右节点都不为null了,也将他们放入队列中
TreeNode *mergeTrees2(TreeNode*root1,TreeNode*root2)
{
//当两颗树有一个为空,则返回另一个树
if(!root1||!root2)
{
return root1==nullptr?root2:root1;
}
queue<TreeNode*>que;
//将两棵树的根节点入队
que.push(root1);
que.push(root2);
//进行层序遍历
while(!que.empty())
{
TreeNode* tempNode1=que.front();
que.pop();
TreeNode* tempNode2=que.front();
que.pop();
//在root1上添加结点
tempNode1->val+=tempNode2->val;
//当tempNode1和tempNode2的左子树都不为空,放入队列中
//若tempNode1的左子树为空,便将tempNode2的左子树挂到tempNode1的左子树上面
if(tempNode1->left&&tempNode2->left)
{
que.push(tempNode1->left);
que.push(tempNode2->left);
}
else if(!tempNode1->left)
{
tempNode1->left=tempNode2->left;
}
//同理
if(tempNode1->right&&tempNode2->right)
{
que.push(tempNode1->right);
que.push(tempNode2->right);
}
else if(!tempNode1->right)
{
tempNode1->right=tempNode2->right;
}
}
return root1;
}