Go语言程序:在二叉搜索树中查找floor和ceil值


二叉搜索树 (BST) 是一种二叉树,其中每个节点最多有两个子节点,通常称为左子节点和右子节点。

在这篇Go语言文章中,我们将学习如何使用递归和迭代方法在二叉搜索树中查找floor和ceil值。二叉搜索树是一种用于高效搜索、插入和删除元素的有用数据结构。

语法

func ceil(root *Node, val int) int {…}

ceil()函数用于在二叉搜索树中查找ceil值。

func floor(root *Node, val int) int {…}

floor()函数用于在二叉搜索树中查找floor值。

func floorCeil(root *Node, key int) (int, int) {…}

floorCeil()函数用于递归地在二叉搜索树中查找floor和ceil值。

算法

  • 步骤1 - 首先,我们需要导入fmt包。

  • 步骤2 - 然后,初始化一个节点结构体并在其中分配三个变量。第一个变量存储整数值,而第二个和第三个指针变量分别存储指向左节点和右节点的地址。

  • 步骤3 - 现在,创建一个insert()函数,它接受一个节点和一个要插入的值作为输入。此函数将节点值插入到合适的二叉搜索树中。

  • 步骤4 - 此函数递归地搜索插入新节点的合适位置,基于二叉搜索树的特性。

  • 步骤5 - 现在,定义一个名为ceil()的函数。它接受一个节点根和一个值val作为输入,并返回树中大于或等于val的最小值。

  • 步骤6 - 然后,定义floor()函数,它接受一个节点根和一个值val作为输入,并返回树中小于或等于val的最大值。

  • 步骤7 - 开始main()函数。在main()函数中,将多个节点插入到二叉搜索树中。

  • 步骤8 - 现在,调用ceil()和floor()函数并将整数值作为参数传递给函数。

  • 步骤9 - 此外,使用fmt.Println()函数在屏幕上打印二叉搜索树的ceil和floor值。

示例1

在这个例子中,我们将迭代地定义一个ceil()和floor()函数,用于在二叉搜索树中查找ceil和floor值。

package main

import (
   "fmt"
)

type Node struct {
   val   int
   left  *Node
   right *Node
}

func insert(root *Node, val int) *Node {
   if root == nil {
      root = &Node{val, nil, nil}
      return root
   }

   if val < root.val {
      root.left = insert(root.left, val)
   } else {
      root.right = insert(root.right, val)
   }

   return root
}

func ceil(root *Node, val int) int {
   if root == nil {
      return -1
   }

   if root.val == val {
      return root.val
   }

   if val > root.val {
      return ceil(root.right, val)
   }

   leftCeil := ceil(root.left, val)
   if leftCeil >= val {
      return leftCeil
   }

   return root.val
}

func floor(root *Node, val int) int {
   if root == nil {
      return -1
   }

   if root.val == val {
      return root.val
   }

   if val < root.val {
      return floor(root.left, val)
   	}

   rightFloor := floor(root.right, val)
   if rightFloor <= val {
      return rightFloor
   }

   return root.val
}

func main() {
   var root *Node

   root = insert(root, 10)
   insert(root, 5)
   insert(root, 15)
   insert(root, 1)
   insert(root, 8)
   insert(root, 12)
   insert(root, 18)

   fmt.Println("Floor of 9:", floor(root, 9))
   fmt.Println("Ceil of 9:", ceil(root, 9))
   fmt.Println("Floor of 11:", floor(root, 11))
   fmt.Println("Ceil of 11:", ceil(root, 11))
}

输出

Floor of 9: -1
Ceil of 9: 10
Floor of 11: -1
Ceil of 11: 12

示例2

在这个例子中,我们将递归地定义一个floorCeil()函数,用于在二叉搜索树中查找ceil和floor值。

package main

import (
   "fmt"
)

type Node struct {
   val   int
   left  *Node
   right *Node
}

func newnode(val int) *Node {
   node := &Node{}
   node.val = val
   node.left = nil
   node.right = nil
   return node
}

func floorCeil(root *Node, key int) (int, int) {
   floor := -1
   ceil := -1

   if root == nil {
      return floor, ceil
   }

   if root.val == key {
      return root.val, root.val
   }

   if root.val > key {
      ceil = root.val
      f, c := floorCeil(root.left, key)
      if f != -1 {
         floor = f
      }
      if c != -1 && c < ceil {
         ceil = c
      }
   } else {
      floor = root.val
      f, c := floorCeil(root.right, key)
      if f != -1 && f > floor {
         floor = f
      }
      if c != -1 {
         ceil = c
      }
   }

   return floor, ceil
}

func main() {
   root := newnode(8)
   root.left = newnode(4)
   root.right = newnode(12)
   root.left.left = newnode(2)
   root.left.right = newnode(6)
   root.right.left = newnode(10)
   root.right.right = newnode(14)

   key := 14
   floor, ceil := floorCeil(root, key)
   fmt.Printf("Floor of %d is %d\n", key, floor)
   fmt.Printf("Ceil of %d is %d\n", key, ceil)

   key = 11
   floor, ceil = floorCeil(root, key)
   fmt.Printf("Floor of %d is %d\n", key, floor)
   fmt.Printf("Ceil of %d is %d\n", key, ceil)

   key = 5
   floor, ceil = floorCeil(root, key)
   fmt.Printf("Floor of %d is %d\n", key, floor)
   fmt.Printf("Ceil of %d is %d\n", key, ceil)
}

输出

Floor of 14 is 14
Ceil of 14 is 14
Floor of 11 is 10
Ceil of 11 is 12
Floor of 5 is 4
Ceil of 5 is 6

结论

我们已经成功地编译并执行了一个Go语言程序,该程序使用递归和迭代方法在二叉搜索树中查找floor和ceil值,并附带两个示例。在第一个示例中,我们使用了迭代方法,在第二个示例中,我们使用了递归方法。

更新于:2023年5月10日

浏览量:172

开启你的职业生涯

通过完成课程获得认证

开始学习
广告