继续刷题的一天,今天写的题是关于深/广度优先搜索的,下面是题目和链接:
题目:给你两棵二叉树: root1 和 root2 。
想象一下,当你将其中一棵覆盖到另一棵之上时,两棵树上的一些节点将会重叠(而另一些不会)。你需要将这两棵树合并成一棵新二叉树。合并的规则是:如果两个节点重叠,那么将这两个节点的值相加作为合并后节点的新值;否则,不为 null 的节点将直接作为新二叉树的节点。
返回合并后的二叉树。
注意: 合并过程必须从两个树的根节点开始。
示例:
输入:root1 = [1,3,2,5], root2 = [2,1,3,null,4,null,7]
输出:[3,4,5,5,4,null,7]
来源:力扣(LeetCode)
链接: https://leetcode.cn/problems/merge-two-binary-trees
下面是解题思路:先序遍历树1,如果树1或树2某个结点为空,那该结点合并后就是树2或树1对应结点的值(注意对应关系),遇到树1不为空的结点将其值更新为树1和树2这一节点的值之和,下面是具体代码实现:
class Solution {
public:
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
if(root1==NULL){
return root2;
}
if(root2==NULL){
return root1;
}
//先合并根结点的值;
root1->val = root1->val + root2->val;
//递归合并左右子树;
root1->left = mergeTrees(root1->left,root2->left);
root1->right = mergeTrees(root1->right,root2->right);
return root1;
}
};