617. 合併二叉樹
阿新 • • 發佈:2018-12-11
617.合併二叉樹
給定兩個二叉樹,想象當你將它們中的一個覆蓋到另一個上時,兩個二叉樹的一些節點便會重疊。
你需要將他們合併為一個新的二叉樹。合併的規則是如果兩個節點重疊,那麼將他們的值相加作為節點合併後的新值,否則不為 NULL 的節點將直接作為新二叉樹的節點。
示例 1:
輸入: Tree 1 Tree 2
1 2 / \ / \ 3 2 1 3 / \ \ 5 4 7
輸出: 合併後的樹:
3
/ \
4 5
/ \ \
5 4 7
注意: 合併必須從兩個樹的根節點開始。
分析: 先判斷t1,t2是否為空,都不為空則將值相加並遞迴計算左右子樹。
/**
* 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;
else if(t1==NULL) return t2;
else if(t2==NULL) return t1;
else
{
t1->val+=t2->val;
t1->left=mergeTrees (t1->left,t2->left);
t1->right=mergeTrees(t1->right,t2->right);
}
return t1;
}
};