天天看點

JavaScript:617. 合并二叉樹(遞歸合并)

題目說明

給定兩個二叉樹,想象當你将它們中的一個覆寫到另一個上時,兩個二叉樹的一些節點便會重疊。

你需要将他們合并為一個新的二叉樹。合并的規則是如果兩個節點重疊,那麼将他們的值相加作為節點合并後的新值,否則不為 NULL 的節點将直接作為新二叉樹的節點。

示例 1:

輸入: 
	Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7                  
輸出: 
合并後的樹:
	     3
	    / \
	   4   5
	  / \   \ 
	 5   4   7
注意: 合并必須從兩個樹的根節點開始。
           

解題思路一

  1. 又是一道二叉樹的題,題目要求合并二叉樹,即相同位置的val值相加,不同位置的val值互相替換。
  2. 一般思路是建立一個新樹,去取兩樹之和。
  3. 但是題目沒有要求不能改變原來的兩棵樹,那麼我們以t1樹為基準,觀察t2樹。
  4. 若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)
    }
}
           

同樣的我們使用遞歸的方法去周遊。

繼續閱讀