首頁>技術>

讀完本文,你可以去力扣拿下如下題目:

100.相同的樹

701.二叉搜尋樹中的插入操作

700.二叉搜尋樹中的搜尋

98.驗證二叉搜尋樹

-----------

透過之前的文章框架思維,二叉樹的遍歷框架應該已經印到你的腦子裡了,這篇文章就來實操一下,看看框架思維是怎麼靈活運用,秒殺一切二叉樹問題的。

二叉樹演算法的設計的總路線:明確一個節點要做的事情,然後剩下的事拋給框架。

void traverse(TreeNode root) {    // root 需要做什麼?在這做。    // 其他的不用 root 操心,拋給框架    traverse(root.left);    traverse(root.right);}

舉兩個簡單的例子體會一下這個思路,熱熱身。

1. 如何把二叉樹所有的節點中的值加一?

void plusOne(TreeNode root) {    if (root == null) return;    root.val += 1;    plusOne(root.left);    plusOne(root.right);}

2. 如何判斷兩棵二叉樹是否完全相同?

boolean isSameTree(TreeNode root1, TreeNode root2) {    // 都為空的話,顯然相同    if (root1 == null && root2 == null) return true;    // 一個為空,一個非空,顯然不同    if (root1 == null || root2 == null) return false;    // 兩個都非空,但 val 不一樣也不行    if (root1.val != root2.val) return false;    // root1 和 root2 該比的都比完了    return isSameTree(root1.left, root2.left)        && isSameTree(root1.right, root2.right);}

藉助框架,上面這兩個例子不難理解吧?如果可以理解,那麼所有二叉樹演算法你都能解決。

二叉搜尋樹(Binary Search Tree,簡稱 BST)是一種很常用的的二叉樹。它的定義是:一個二叉樹中,任意節點的值要大於等於左子樹所有節點的值,且要小於等於右邊子樹的所有節點的值。

如下就是一個符合定義的 BST:

下面實現 BST 的基礎操作:判斷 BST 的合法性、增、刪、查。其中“刪”和“判斷合法性”略微複雜。

零、判斷 BST 的合法性

這裡是有坑的哦,我們按照剛才的思路,每個節點自己要做的事不就是比較自己和左右孩子嗎?看起來應該這樣寫程式碼:

boolean isValidBST(TreeNode root) {    if (root == null) return true;    if (root.left != null && root.val <= root.left.val) return false;    if (root.right != null && root.val >= root.right.val) return false;    return isValidBST(root.left)        && isValidBST(root.right);}

但是這個演算法出現了錯誤,BST 的每個節點應該要小於右邊子樹的所有節點,下面這個二叉樹顯然不是 BST,但是我們的演算法會把它判定為 BST。

出現錯誤,不要慌張,框架沒有錯,一定是某個細節問題沒注意到。我們重新看一下 BST 的定義,root 需要做的不只是和左右子節點比較,而是要整個左子樹和右子樹所有節點比較。怎麼辦,鞭長莫及啊!

這種情況,我們可以使用輔助函式,增加函式引數列表,在引數中攜帶額外資訊,請看正確的程式碼:

boolean isValidBST(TreeNode root) {    return isValidBST(root, null, null);}​boolean isValidBST(TreeNode root, TreeNode min, TreeNode max) {    if (root == null) return true;    if (min != null && root.val <= min.val) return false;    if (max != null && root.val >= max.val) return false;    return isValidBST(root.left, min, root)         && isValidBST(root.right, root, max);}

一、在 BST 中查詢一個數是否存在

根據我們的指導思想,可以這樣寫程式碼:

boolean isInBST(TreeNode root, int target) {    if (root == null) return false;    if (root.val == target) return true;​    return isInBST(root.left, target)        || isInBST(root.right, target);}

這樣寫完全正確,充分證明了你的框架性思維已經養成。現在你可以考慮一點細節問題了:如何充分利用資訊,把 BST 這個“左小右大”的特性用上?

很簡單,其實不需要遞迴地搜尋兩邊,類似二分查詢思想,根據 target 和 root.val 的大小比較,就能排除一邊。我們把上面的思路稍稍改動:

boolean isInBST(TreeNode root, int target) {    if (root == null) return false;    if (root.val == target)        return true;    if (root.val < target)         return isInBST(root.right, target);    if (root.val > target)        return isInBST(root.left, target);    // root 該做的事做完了,順帶把框架也完成了,妙}

於是,我們對原始框架進行改造,抽象出一套針對 BST 的遍歷框架

void BST(TreeNode root, int target) {    if (root.val == target)        // 找到目標,做點什麼    if (root.val < target)         BST(root.right, target);    if (root.val > target)        BST(root.left, target);}

二、在 BST 中插入一個數

對資料結構的操作無非遍歷 + 訪問,遍歷就是“找”,訪問就是“改”。具體到這個問題,插入一個數,就是先找到插入位置,然後進行插入操作。

上一個問題,我們總結了 BST 中的遍歷框架,就是“找”的問題。直接套框架,加上“改”的操作即可。一旦涉及“改”,函式就要返回 TreeNode 型別,並且對遞迴呼叫的返回值進行接收。

TreeNode insertIntoBST(TreeNode root, int val) {    // 找到空位置插入新節點    if (root == null) return new TreeNode(val);    // if (root.val == val)    //     BST 中一般不會插入已存在元素    if (root.val < val)         root.right = insertIntoBST(root.right, val);    if (root.val > val)         root.left = insertIntoBST(root.left, val);    return root;}

三、在 BST 中刪除一個數

這個問題稍微複雜,不過你有框架指導,難不住你。跟插入操作類似,先“找”再“改”,先把框架寫出來再說:

TreeNode deleteNode(TreeNode root, int key) {    if (root.val == key) {        // 找到啦,進行刪除    } else if (root.val > key) {        root.left = deleteNode(root.left, key);    } else if (root.val < key) {        root.right = deleteNode(root.right, key);    }    return root;}

找到目標節點了,比方說是節點 A,如何刪除這個節點,這是難點。因為刪除節點的同時不能破壞 BST 的性質。有三種情況,用圖片來說明。

情況 1:A 恰好是末端節點,兩個子節點都為空,那麼它可以當場去世了。

圖片來自 LeetCode

if (root.left == null && root.right == null)    return null;

情況 2:A 只有一個非空子節點,那麼它要讓這個孩子接替自己的位置。

圖片來自 LeetCode

// 排除了情況 1 之後if (root.left == null) return root.right;if (root.right == null) return root.left;

情況 3:A 有兩個子節點,麻煩了,為了不破壞 BST 的性質,A 必須找到左子樹中最大的那個節點,或者右子樹中最小的那個節點來接替自己。我們以第二種方式講解。

圖片來自 LeetCode

if (root.left != null && root.right != null) {    // 找到右子樹的最小節點    TreeNode minNode = getMin(root.right);    // 把 root 改成 minNode    root.val = minNode.val;    // 轉而去刪除 minNode    root.right = deleteNode(root.right, minNode.val);}

三種情況分析完畢,填入框架,簡化一下程式碼:

TreeNode deleteNode(TreeNode root, int key) {    if (root == null) return null;    if (root.val == key) {        // 這兩個 if 把情況 1 和 2 都正確處理了        if (root.left == null) return root.right;        if (root.right == null) return root.left;        // 處理情況 3        TreeNode minNode = getMin(root.right);        root.val = minNode.val;        root.right = deleteNode(root.right, minNode.val);    } else if (root.val > key) {        root.left = deleteNode(root.left, key);    } else if (root.val < key) {        root.right = deleteNode(root.right, key);    }    return root;}TreeNode getMin(TreeNode node) {    // BST 最左邊的就是最小的    while (node.left != null) node = node.left;    return node;} 

刪除操作就完成了。注意一下,這個刪除操作並不完美,因為我們一般不會透過 root.val = minNode.val 修改節點內部的值來交換節點,而是透過一系列略微複雜的連結串列操作交換 root 和 minNode 兩個節點。因為具體應用中,val 域可能會很大,修改起來很耗時,而連結串列操作無非改一改指標,而不會去碰內部資料。

但這裡忽略這個細節,旨在突出 BST 基本操作的共性,以及藉助框架逐層細化問題的思維方式。

四、最後總結

透過這篇文章,你學會了如下幾個技巧:

二叉樹演算法設計的總路線:把當前節點要做的事做好,其他的交給遞迴框架,不用當前節點操心。如果當前節點會對下面的子節點有整體影響,可以透過輔助函式增長引數列表,藉助引數傳遞資訊。在二叉樹框架之上,擴展出一套 BST 遍歷框架:
void BST(TreeNode root, int target) {    if (root.val == target)        // 找到目標,做點什麼    if (root.val < target)         BST(root.right, target);    if (root.val > target)        BST(root.left, target);}
掌握了 BST 的基本操作。

8
最新評論
  • BSA-TRITC(10mg/ml) TRITC-BSA 牛血清白蛋白改性標記羅丹明
  • 那些小而美的演算法技巧:字首和/差分陣列