題目說明
給定兩個二叉樹,想象當你将它們中的一個覆寫到另一個上時,兩個二叉樹的一些節點便會重疊。
你需要将他們合并為一個新的二叉樹。合并的規則是如果兩個節點重疊,那麼将他們的值相加作為節點合并後的新值,否則不為 NULL 的節點将直接作為新二叉樹的節點。
示例 1:
輸入:
Tree 1 Tree 2
1 2
/ \ / \
3 2 1 3
/ \ \
5 4 7
輸出:
合并後的樹:
3
/ \
4 5
/ \ \
5 4 7
注意: 合并必須從兩個樹的根節點開始。
解題思路一
- 又是一道二叉樹的題,題目要求合并二叉樹,即相同位置的val值相加,不同位置的val值互相替換。
- 一般思路是建立一個新樹,去取兩樹之和。
- 但是題目沒有要求不能改變原來的兩棵樹,那麼我們以t1樹為基準,觀察t2樹。
- 若t1樹對應t2樹的位置都有值,則相加,若t1無,t2有,則t2替換t1節點,若t1有,t2無則無需操作。
代碼實作一
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} t1
* @param {TreeNode} t2
* @return {TreeNode}
*/
let mergeTrees = (t1, t2) => {
if (t1 && t2) {
mergeNode(t1,t2)
}
return t1 || t2;
};
let mergeNode = (t1, t2, root) => {
t1.val += t2.val;
if (t1.left === null) {
t1.left = t2.left
} else if( t2.left !== null) {
mergeNode(t1.left,t2.left)
}
if (t1.right === null) {
t1.right = t2.right
} else if (t2.right !== null) {
mergeNode(t1.right,t2.right)
}
}
同樣的我們使用遞歸的方法去周遊。