题目描述
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则不为 NULL 的节点将直接作为新二叉树的节点。
链接:https://leetcode-cn.com/problems/merge-two-binary-trees
三种情况:
- 两个二叉树对应的节点均不为空,合并后的二叉树节点为二者之和。
- 两个二叉树对应的节点有一个为空,那么合并后的二叉树的对应节点为其中的非空节点
- 两个二叉树对应的节点均为空,合并后的二叉树节点为空。
题解中没有t1 == null && t2 = =null 这条语句,其实在 if(!t1||!t2) return t1==nullptr?t2:t1; 中已经包含了该层意思。
递归版本:
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
// write code here
if(!t1&&!t2) return nullptr;
if(!t1||!t2) return t1==nullptr?t2:t1;
t1->val=t1->val+t2->val;
t1->left=mergeTrees(t1->left, t2->left);
t1->right=mergeTrees(t1->right, t2->right);
return t1;
}
};
迭代版本:
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if (t1 == NULL) return t2;
if (t2 == NULL) return t1;
queue<TreeNode*> que;//TreeNode* 结构体指针
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;
}
};