首页 > 其他分享 >基于四叉树的小顶堆(最小优先队列)

基于四叉树的小顶堆(最小优先队列)

时间:2023-02-19 20:01:12浏览次数:36  
标签:tmp return 队列 when w3 c3 小顶 四叉树 节点

实现来自Go源码

 

从下往上调整堆

func siftupTimer(t []*timer, i int) bool {
   if i >= len(t) {
      return false
   }
   when := t[i].when
   tmp := t[i]
   for i > 0 { // 除非到达堆顶,否则每次和当前父节点进行比较,如果当前节点更小,那么交换
      p := (i - 1) / 4 // parent
      if when >= t[p].when {
         break
      }
      t[i] = t[p]
      t[i].i = i
      i = p
   }
   if tmp != t[i] {
      t[i] = tmp
      t[i].i = i
   }
   return true
}

从上往下调整堆

func siftdownTimer(t []*timer, i int) bool {
   n := len(t)
   if i >= n {
      return false
   }
   when := t[i].when
   tmp := t[i]
   for { // 这里的前一部分在确定哪个子节点最小,可以使用 for 循环遍历数组来优化
      c := i*4 + 1 // left child
      c3 := c + 2  // mid child
      if c >= n {
         break
      }
      w := t[c].when
      if c+1 < n && t[c+1].when < w {
         w = t[c+1].when
         c++
      }
      if c3 < n {
         w3 := t[c3].when
         if c3+1 < n && t[c3+1].when < w3 {
            w3 = t[c3+1].when
            c3++
         }
         if w3 < w {
            w = w3
            c = c3
         }
      }
      if w >= when { // 如果子节点最小值 >= 当前节点,则结束
         break
      }
      t[i] = t[c]
      t[i].i = i
      i = c
   }
   if tmp != t[i] {
      t[i] = tmp
      t[i].i = i
   }
   return true
}

不同情况的小顶堆调整
插入新元素时,放置到堆的末尾,基于这个元素向上调整堆。
删除某个元素时
1. 如果是最后一个元素,那么直接删除。
2. 否则,把最后一个元素置为当前元素,基于当前元素,先向上调整堆,再向下调整堆。

标签:tmp,return,队列,when,w3,c3,小顶,四叉树,节点
From: https://www.cnblogs.com/WJQ2017/p/17135443.html

相关文章