Go語言實現紅黑樹


紅黑樹是一種具有穩定結構和高度平衡的二叉搜尋樹,能夠自平衡。它們有利於高效地進行插入、刪除和搜尋操作。在本文中,我們將深入探討如何在Go語言中實現紅黑樹。在第一個示例中,我們將直接構建樹,而在第二個示例中,我們將使用結構體來構建樹。

解釋

紅黑樹是一種自平衡二叉搜尋樹,在插入和刪除操作期間,透過確保二叉搜尋樹中的每個節點都被指定為紅色或黑色來確保平衡。這種獨特的特性最終導致樹遵循五個關鍵屬性。

  • 每個節點的顏色都是黑色或紅色。

  • 根節點的顏色為黑色。

  • 每個葉節點(特別是NIL節點)的顏色為黑色。

  • 任何紅色節點都必須有兩個黑色的子節點。

  • 從任何節點到其後代葉節點的任何路徑上,黑色節點的數量必須相同。

語法

func (t *RedBlackTree) insert(root, node *Node) *Node

語法定義了一個名為insert的函式,該函式在新增新節點時用於保持紅黑樹的特性。為此,它需要根節點和要插入的節點作為輸入。

演算法

  • 首先建立一個普通的二叉搜尋樹。

  • 新插入的節點被著色為紅色。

  • 為了保持紅黑樹的五個屬性,我們確保在必要時執行旋轉和重新著色。

  • 插入或旋轉後,不能有連續的紅色節點。

  • 如果發生違規,我們需要根據需要應用旋轉和重新著色以恢復屬性。

示例1

在這個例子中,我們將透過定義一個結構體,透過顏色編碼和宣告節點之間的父子關係,直接在Go語言中實現一個紅黑樹。我們需要將建立的根節點包含在RedBlackTree結構體中。節點插入是透過遞迴insert方法實現的,該方法進一步實現了父指標調整。使用inOrder方法,透過進行中序遍歷來檢視節點資料和顏色。main函式突出顯示一個紅黑樹例項,最終產生相應的中序遍歷結果。

package main
import "fmt"
type Color int
const (
   Red   Color = 0
   Black Color = 1
)
type Node struct {
   data   int
   color  Color
   parent *Node
   left   *Node
   right  *Node
}
type RedBlackTree struct{ root *Node }
func NewRedBlackTree() *RedBlackTree { return &RedBlackTree{} }
func (t *RedBlackTree) insert(root, node *Node) *Node {
   if root == nil {
      return node
   }
   if node.data < root.data {
      root.left = t.insert(root.left, node)
      root.left.parent = root
   } else if node.data > root.data {
      root.right = t.insert(root.right, node)
      root.right.parent = root
    }
   return root
}
func (t *RedBlackTree) inOrder(node *Node) {
   if node == nil {
      return
   }
   t.inOrder(node.left)
   fmt.Printf("%d (%s) ", node.data, func(c Color) string {
      if c == Red {
         return "R"
      }
      return "B"
   }(node.color))
   t.inOrder(node.right)
}
func main() {
   tree := NewRedBlackTree()
   data := []int{10, 20, 30, 15, 5}
   for _, d := range data {
      tree.root = tree.insert(tree.root, &Node{data: d, color: Red})
      fmt.Printf("Inserted: %d\n", d)
   }
   fmt.Println("\nIn-order Traversal:")
   tree.inOrder(tree.root)
}

輸出

Inserted: 10
Inserted: 20
Inserted: 30
Inserted: 15
Inserted: 5

In-order Traversal:
5 (R) 10 (R) 15 (R) 20 (R) 30 (R)

示例2

在這個例子中,我們將使用名為`Node`的結構體間接地用Go語言實現紅黑樹,該結構體表示包含顏色屬性的元素。我們定義了`newNode`函式來建立節點,然後透過`insert`函式遞迴地將這些節點新增到樹中,仔細考慮它們的色彩值,並在此過程中建立父子關係。出於遍歷目的,有一個`inOrde­rTraversal`函式執行樹的左-根-右遍歷。此函式顯示節點資料和顏色。`main`函式初始化一棵樹並插入特定的資料點,最後進行相應的中序遍歷。

package main
import "fmt"
type Color int
const (
   Red   Color = 0
   Black Color = 1
)
type Node struct {
   data   int
   color  Color
   parent *Node
   left   *Node
   right  *Node
}
func newNode(data int, color Color, parent, left, right *Node) *Node {
        	return &Node{data: data, color: color, parent: parent, left: left, right: right}
}
func insert(root *Node, data int) *Node {
   if root == nil {
      return newNode(data, Red, nil, nil, nil)
   }
   if data < root.data {
      root.left = insert(root.left, data)
      root.left.parent = root
   } else if data > root.data {
       root.right = insert(root.right, data)
       root.right.parent = root
     }
   return root
}
func inOrderTraversal(root *Node) {
   if root == nil {
      return
   }
   inOrderTraversal(root.left)
   fmt.Printf("%d (%s) ", root.data, colorToString(root.color))
   inOrderTraversal(root.right)
}
func colorToString(color Color) string {
   if color == Red {
      return "R"
   }
   return "B"
}
func main() {
   var root *Node
   data := []int{10, 20, 30, 15, 5}
   for _, d := range data {
      root = insert(root, d)
      fmt.Printf("Inserted: %d\n", d)
   }
   fmt.Println("\nIn-order Traversal:")
   inOrderTraversal(root)
}

輸出

Inserted: 10
Inserted: 20
Inserted: 30
Inserted: 15
Inserted: 5
 
In-order Traversal:
5 (R) 10 (R) 15 (R) 20 (R) 30 (R)

實際應用

  • Linux核心的完全公平排程器(CFS): Linux核心中使用的完全公平排程器(CFS)在任務管理和組織中起著至關重要的作用。CFS利用紅黑樹資料結構為每個任務分配優先順序,從而能夠精確地確定下一個要執行的任務,同時長期保持所有作業的公平處理。

  • 拼寫檢查器和自動完成建議: 提供即時建議和拼寫檢查需要高效的資料結構。紅黑樹有助於儲存單詞列表和字典,其平衡的樹結構允許快速搜尋以獲得準確的拼寫和建議。

結論

透過顏色編碼的節點自平衡二叉搜尋樹(稱為紅黑樹)來維護高效的操作和平衡的高度。在本文中,我們提供了在Go語言中實現紅黑樹的簡化程式,並說明了插入過程,演示了資料結構平衡對於最佳效能的重要性。它能夠支援關鍵操作同時保持平衡,使紅黑樹成為資料庫、記憶體分配和語言庫中的首選。

更新於:2023年10月18日

101次瀏覽

啟動您的職業生涯

完成課程獲得認證

開始學習
廣告