二叉堆是一棵满二叉树,父节点的值大于子节点。

用数组存储二叉堆,假如一个节点的下标为k,那么这个节点的左孩子就是2k,右孩子就是2k+1。

用Go语言实现如下:

package heap

import (
    "fmt"
)

// MaxHeap 大顶堆
type MaxHeap struct {
    data []int
}

// Init 初始化
func (heap *MaxHeap) Init(maxSize int) {
    heap.data = make([]int, 0, maxSize)
    heap.data = append(heap.data, 0)
}

// Size 堆的长度
func (heap MaxHeap) Size() (size int) {
    size = len(heap.data) - 1
    return
}

// IsEmpty 堆是否为空
func (heap MaxHeap) IsEmpty() bool {
    return heap.Size() == 0
}

// Insert 插入一个元素
func (heap *MaxHeap) Insert(item int) {
    heap.data = append(heap.data, item)
    heap.shiftUp(heap.Size())
}

// Extract 删除一个元素
func (heap *MaxHeap) Extract() int {
    item := heap.data[1]
    heap.data[1], heap.data[heap.Size()] = heap.data[heap.Size()], heap.data[1]
    heap.data = heap.data[:len(heap.data)-1]
    if !heap.IsEmpty() {
        heap.shiftDown(1)
    }
    return item
}

// shiftUp 向上调整堆的结构
// 优化:不用每次都交换
func (heap *MaxHeap) shiftUp(k int) {
    temp := heap.data[k]
    for k > 1 && heap.data[k/2] < temp {
        heap.data[k/2] = heap.data[k]
        k /= 2
    }
    heap.data[k] = temp
}

// shiftDown 向下调整堆的结构
// 优化:不用每次都交换
func (heap *MaxHeap) shiftDown(k int) {
    temp := heap.data[k]
    for k*2 <= heap.Size() {
        j := 2 * k
        if j+1 <= heap.Size() && heap.data[j+1] > heap.data[j] {
            j = j + 1
        }
        if temp >= heap.data[j] {
            break
        }
        heap.data[k] = heap.data[j]
        k = j
    }
    heap.data[k] = temp
}

// Show 显示堆的内容
func (heap MaxHeap) Show() {
    for i, e := range heap.data {
        if i == 0 {
            continue
        } else {
            fmt.Println(e)
        }
    }
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 二叉树 满二叉树 国内教程定义:一个二叉树,如果每一个层的结点数都达到最大值,则这个二叉树就是满二叉树。也就是说,...
    简_爱SimpleLove阅读 4,313评论 0 3
  • 0.目录 1.优先队列ADT 2.几种实现 3.二叉堆 4.d-堆 5.左式堆 6.斜堆 7.二项队列 8.斐波那...
    王侦阅读 3,192评论 1 2
  • 更详细的讲解和代码调试演示过程,请点击链接 做过系统编程的人都知道,几乎任何系统都会提供一种时钟机制,也就是Set...
    望月从良阅读 827评论 0 1
  • 数据结构与算法--优先队列和堆排序 在某些数据处理的例子中,总数据量太大,无法排序(甚至无法全部装进内存)。例如,...
    sunhaiyu阅读 1,053评论 0 2
  • 我们总是这样问自己 如果当时选择了另外一条路 会不会不一样 会不会没有那么多遗憾 可是 现状是无从改变的 即使 我...
    祎言阅读 216评论 0 0