「这是我参与11月更文挑战的第18天,活动详情查看:2021最后一次更文挑战」
617. 合并二叉树
题目描述
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
示例 1:
输入:
Tree 1 Tree 2
1 2
/ \ / \
3 2 1 3
/ \ \
5 4 7
输出:
合并后的树:
3
/ \
4 5
/ \ \
5 4 7
复制代码
注意: 合并必须从两个树的根节点开始。
递归法
前序遍历 -- 递归法
- 1、递归函数的参数与返回值
- 参数:两个树的节点
- 返回值:新树的节点
- 2、递归终止条件
- 两个树的节点一个为空一个不为空,返回不为空的节点
- 两个树的节点都为空,返回空
- 两个树的节点都不为空,返回新节点
- 3、单层遍历的逻辑
- 根节点
- 左节点
- 右节点
代码
合并后的树使用新的空间
// 递归法
class Solution
{
public:
TreeNode *mergeTrees(TreeNode *root1, TreeNode *root2)
{
return traversal(root1, root2);
}
private:
TreeNode *traversal(TreeNode *root1, TreeNode *root2)
{
// 判断是否有节点为空
// 返回不为空的节点
// 全为空的时候返回NULL
if (root1 == NULL && root2 != NULL)
{
return root2;
}
else if (root1 != NULL && root2 == NULL)
{
return root1;
}
else if (root1 == NULL && root2 == NULL)
{
return NULL;
}
// 全都非空
// 定义新的节点,节点值为两个节点值的和
TreeNode *node = new TreeNode(root1->val + root2->val);
// 左
node->left = traversal(root1->left, root2->left);
// 右
node->right = traversal(root1->right, root2->right);
return node;
}
};
复制代码
原地合并
class Solution{
public:
TreeNode *mergeTrees(TreeNode *root1,TreeNode *root2){
// 如果第一棵树节点为空,则返回第二棵树的节点
if(root1 ==NULL){
return root2;
}
if(root2 == NULL){
return root1;
}
// 在第一棵树上进行原地修改
// 中
root1->val += root2->val;
// 左
root1->left = mergeTrees(root1->left,root2->left);
// 右
root1->right = mergeTrees(root1->right,root2->right);
return root1;
}
};
复制代码
迭代法
思路同递归
代码
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if (t1 == NULL) return t2;
if (t2 == NULL) return t1;
queue<TreeNode*> que;
que.push(t1);
que.push(t2);
while(!que.empty()) {
TreeNode* node1 = que.front();
que.pop();
TreeNode* node2 = que.front();
que.pop();
// 此时两个节点一定不为空,val相加
node1->val += node2->val;
// 如果两棵树左节点都不为空,加入队列
if (node1->left != NULL && node2->left != NULL) {
que.push(node1->left);
que.push(node2->left);
}
// 如果两棵树右节点都不为空,加入队列
if (node1->right != NULL && node2->right != NULL) {
que.push(node1->right);
que.push(node2->right);
}
// 当t1的左节点 为空 t2左节点不为空,就赋值过去
if (node1->left == NULL && node2->left != NULL) {
node1->left = node2->left;
}
// 当t1的右节点 为空 t2右节点不为空,就赋值过去
if (node1->right == NULL && node2->right != NULL) {
node1->right = node2->right;
}
}
return t1;
}
};
复制代码