617. 合并二叉树 - 力扣
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
示例 1:
输入:
Tree 1 Tree 2
1 2
/ \ / \
3 2 1 3
/ \ \
5 4 7
输出:
合并后的树:
3
/ \
4 5
/ \ \
5 4 7
解题思路:
a
问 答 如何合并? 如果相对于的节点都有值,值相加.如果左边为空,右边存在节点,直接让左边连接右边的节点即可. 如何取相应节点? 运用递归, 终止条件 直到两个为空或者右边的树为空. 递归条件 左孩子或者右孩子都存在就持续递归. 单层递归 把相对于的值相加,然后判断是否满足递归条件,如果满足持续递归.直到结束位置.
- 到这里就已经把递归理顺了,只需要把传进来的节点值相加,然后看情况是继续递归还是连接节点或者返回.实现代码如下:
代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
void addTrees(TreeNode* root1, TreeNode* root2)
{
//两个值相加,能进来说明两个都存在
root1->val += root2->val;
//如果两边的左孩子都存在
if (root1->left && root2->left)
addTrees(root1->left, root2->left);
//如果两边的右孩子都存在
if (root1->right && root2->right)
addTrees(root1->right, root2->right);
//如果左边不存在但右边存在
if (!root1->left && root2->left)
root1->left = root2->left;
if (!root1->right && root2->right)
root1->right = root2->right;
//右边不存在但左边存在直接返回即可.
}
TreeNode* mergeTrees(TreeNode* root1, TreeNode* root2) {
//两个根节点均为空
if (!root1 && !root2)
return root1;
//有一个为空
if(!root1)
return root2;
if(!root2)
return root1;
addTrees(root1, root2);
return root1;
}
};
总结:
这道题的递归并不难想,只要想明白递归的终止条件和递归条件即可.