Stories

Detail Return Return

可視化圖解算法29:合併二叉樹 - Stories Detail

1. 題目

描述

已知兩顆二叉樹,將它們合併成一顆二叉樹。合併規則是:都存在的結點,就將結點值加起來,否則空的位置就由另一個樹的結點來代替。例如:

數據範圍:樹上節點數量滿足 0 ≤n≤500,樹上節點的值一定在32位整型範圍內。

進階:空間複雜度 O(1) ,時間複雜度 O(n)

示例1

輸入:

{1,3,2,5},{2,1,3,#,4,#,7}

返回值:

{3,4,5,5,4,#,7}

示例2

輸入:

{1},{}

返回值:

{1}

2. 解題思路

二叉樹tree1與tree2的合併,我們可以將tree2合併到tree1,採用遞歸方法。具體操作為:

​   tree1的節點值為兩顆二叉樹節點值的和:t1.val=t1.val+t2.val;

​   tree1的新生成的左子樹為:tree1的左子樹和tree2的左子樹合併的結果;

​   tree1的新生成的右子樹為:tree1的右子樹和tree2的右子樹合併的結果。

先來看看二叉樹的合併是否滿足遞歸條件:

可以看出,二叉樹的合併滿足遞歸的兩個條件,因此可以通過遞歸來實現二叉樹的合併。我們將tree2合併到tree1,採用遞歸方法。具體操作為:

​   tree1的節點值為兩顆二叉樹節點值的和:t1.val=t1.val+t2.val;

​   tree1的新生成的左子樹為:tree1的左子樹和tree2的左子樹合併的結果;

​   tree1的新生成的右子樹為:tree1的右子樹和tree2的右子樹合併的結果。

對應的遞推公式如下:

有了遞推公式,就可以很方便的寫出代碼。tree1與tree2合併之後,新生成的二叉樹如下所示:

如果文字描述的不太清楚,你可以參考視頻的詳細講解。

  • Python版本:https://www.bilibili.com/cheese/play/ep1372109
  • Java版本:https://www.bilibili.com/cheese/play/ep1367348
  • Golang版本:https://www.bilibili.com/cheese/play/ep1364773

3. 編碼實現

核心代碼如下:

/**
 * 代碼中的類名、方法名、參數名已經指定,請勿修改,直接返回方法規定的值即可
 *
 *
 * @param t1 TreeNode類
 * @param t2 TreeNode類
 * @return TreeNode類
 */
func mergeTrees(t1 *TreeNode, t2 *TreeNode) *TreeNode {
    // write code here
    // 2. 遞歸終止條件
    if t1 == nil && t2 == nil {
        return nil
    }
    // 2.2 當前節點為空,則返回另外一個節點
    if t1 == nil && t2 != nil {
        return t2
    }
    if t1 != nil && t2 == nil {
        return t1
    }

    // 1. 問題分解(遞推公式)
    // 1.1 root節點處理(將t2合併到t1),兩個節點的值求和並賦給t1
    t1.Val = t1.Val + t2.Val
    // 1.2 合併t1、t2的左節點
    t1.Left = mergeTrees(t1.Left, t2.Left)
    // 1.3 合併t1、t2的右節點
    t1.Right = mergeTrees(t1.Right, t2.Right)
    return t1
}

具體完整代碼你可以參考下面視頻的詳細講解。

  • Python版本:https://www.bilibili.com/cheese/play/ep1372109
  • Java版本:https://www.bilibili.com/cheese/play/ep1367348
  • Golang版本:https://www.bilibili.com/cheese/play/ep1364773

4.小結

二叉樹tree1與tree2的合併,我們可以將tree2合併到tree1,採用遞歸方法。具體操作為:

​   tree1的節點值為兩顆二叉樹節點值的和:t1.val=t1.val+t2.val;

​   tree1的新生成的左子樹為:tree1的左子樹和tree2的左子樹合併的結果;

​   tree1的新生成的右子樹為:tree1的右子樹和tree2的右子樹合併的結果。

《數據結構與算法》深度精講課程正式上線啦!七大核心算法模塊全解析:

  ✅   鏈表

  ✅   二叉樹

  ✅   二分查找、排序

  ✅   堆、棧、隊列

  ✅   回溯算法

  ✅   哈希算法

  ✅   動態規劃

無論你是備戰筆試面試、提升代碼效率,還是突破技術瓶頸,這套課程都將為你構建紮實的算法思維底座。🔥立即加入學習打卡,與千名開發者共同進階!

  • Python編碼實現:https://www.bilibili.com/cheese/play/ss897667807
  • Java編碼實現:https://www.bilibili.com/cheese/play/ss161443488
  • Golang編碼實現:https://www.bilibili.com/cheese/play/ss63997
對於二叉樹的相關算法,我們總結了一套【可視化+圖解】方法,依據此方法來解決相關問題,算法變得易於理解,寫出來的代碼可讀性高也不容易出錯。具體也可以參考視頻詳細講解。

今日佳句:十年身事各如萍,白首相逢淚滿纓。

user avatar yuanfang_648a85b26d85e Avatar jinyeyoudianerliang Avatar aloudata Avatar hoistthecolorsandsteptotherail Avatar shoushoudeniupai Avatar gvenusleo Avatar jidcoo Avatar
Favorites 7 users favorite the story!
Favorites

Add a new Comments

Some HTML is okay.