优先队列(大小堆)

代码主要实现了小顶堆的创建、插入、删除、向上调整、向下调整、遍历、排序

#include <iostream>
#include <assert.h>
using namespace std;

template <typename T>
class MinHeap
{
public:
    MinHeap(int sz = DefaultSize)
    {
        maxsize = sz > DefaultSize ? sz:DefaultSize;
        heap = new T [maxsize];
        cursize = 0;
    }
    ~MinHeap()
    {
        delete[]heap;
        heap = NULL;
        maxsize = cursize = 0;
    }
public:
    void Insert(T &x)
    {
        if(cursize >= maxsize)
            return;
        heap[cursize] = x;
        ShiftUp(cursize);
        cursize++;
    }
    T RemoveHeap()
    {
       assert(cursize != 0);
       T key = heap[0];
       heap[0] = heap[cursize-1];
       cursize--;
       ShiftDown(0);
       return key;
    }
    void ShowHeap()const
    {
        for (int i = 0; i < cursize; ++i) {
            cout<<heap[i]<<" ";
        }
        cout<<endl;
    }
    void Sort()
    {
        while (cursize != 0){
            cout<<RemoveHeap()<<" ";
        }
    }
protected:
    void ShiftUp(int pos)
    {
        int i = (pos-1)/2;//最后分支  i 是 j parent
        int j = pos;
        int tmp = heap[j];
        while(j > 0)
        {
            if(tmp < heap[i])
            {
                heap[j] = heap[i];
                j = i;
                i = (j-1)/2;
            }
            else
                break;
        }
        heap[j] = tmp;
    }
    void ShiftDown(int pos)
    {
        int i = pos;
        int j = 2*i+1;
        int tmp = heap[i];
        while(j < cursize)
        {
            if(j+1<cursize && heap[j]>heap[j+1])
                j++;
            if(heap[j] < tmp)
            {
                heap[i] = heap[j];
                i = j;
                j = 2*i+1;
            }
            else
                break;
        }
        heap[i] = tmp;
    }
private:
    enum {DefaultSize = 10};
    T *heap;
    int cursize;
    int maxsize;
};
int main ()
{
    int arr[] = {49,38,65,97,13,27};
    MinHeap<int >mh;
    int n = sizeof(arr) / sizeof(int );
    for (int i = 0; i < n; ++i) {
        mh.Insert(arr[i]);
    }
    mh.ShowHeap();
    mh.Sort();
    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值