1. 程式人生 > >【LeetCode】 617. 合並二叉樹

【LeetCode】 617. 合並二叉樹

兩個 val return 將他 子節點 覆蓋 們的 二叉樹 不存在

題目

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

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

示例 1:

輸入: 
    Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7                  
輸出: 
合並後的樹:
         3
        /        4   5
      / \   \ 
     5   4   7

註意: 合並必須從兩個樹的根節點開始。

思路

使用遞歸算法將兩個二叉樹合並,合並時進行同步遍歷,剛開始可以將t1,t2看做根節點,遞歸開始以後看做子節點,對於相同位置上的一對結點t1和t2,處理策略是:

  • 若t1和t2都存在,則只需要保留其中一個結點(如t1),將另一結點的值加到此結點上即可(如t1.val += t2.val)。
  • 若t1或t2任一不存在,則合並後的二叉樹對應位置上的結點就是存在的那個了。
  • 若n1和n2都不存在,則合並後仍不存在。

代碼

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
struct TreeNode* mergeTrees(struct TreeNode* t1, struct TreeNode* t2) {
    if(t1 == NULL)
        return t2;
    if(t2 == NULL)
        return t1;
    t1->val = t1->val + t2->val;
    t1->left = mergeTrees(t1->left,t2->left);
    t1->right = mergeTrees(t1->right,t2->right);
    return t1;
    
}

【LeetCode】 617. 合並二叉樹