链接
https://leetcode-cn.com/problems/merge-two-binary-trees/
耗时
解题:25 min
题解:5 min
题意
给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。
你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则 不为 NULL 的节点 将直接作为新二叉树的节点。
思路
按规则合并当前的两个节点,再依相同规则合并节点的左右子树。唯一需要注意的是当前节点为空时的处理,详见代码。
时间复杂度: O ( n ) O(n) O(n)
AC代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) {
if(t1 == NULL && t2 == NULL) return NULL;
TreeNode *t1l = NULL, *t1r = NULL, *t2l = NULL, *t2r = NULL;
int t1v = 0, t2v = 0;
if(t1 != NULL) {
t1l = t1->left;
t1r = t1->right;
t1v = t1->val;
}
if(t2 != NULL) {
t2l = t2->left;
t2r = t2->right;
t2v = t2->val;
}
TreeNode* root = new TreeNode(t1v+t2v);
root->left = mergeTrees(t1l, t2l);
root->right = mergeTrees(t1r, t2r);
return root;
}
};