性质
堆是一棵完全二叉树,因此可用数组进行模拟。
设数组为heap[size+1],第一个元素为控制边界用,赋值以很小的数。这样上调过程中不用另加判断。
arr[i]的左右子女分别为arr[2*i]和arr[2*i+1],arr[i]的父亲为arr[i/2],注意此时根节点从1开始。
结构定义、初始化
typedef struct {
int capacity;
int size;
int *elements;
} heap;
heap* initHeap(int capacity){
heap *h = (heap*)malloc(sizeof(heap));
h->capacity = capacity;
h->size = 0;
h->elements = (int*)malloc(sizeof(int)*(capacity+1));
h->elements[0] = 0x80000000;
return h;
}
插入、建堆
void insertHeap(heap *h, int ele){
if(h->size==h->capacity){
cout<< "heap is full!"<< endl;
return;
}
h->elements[++h->size] = ele;
int i=h->size;
while(ele<h->elements[i/2]){
h->elements[i] = h->elements[i/2];
i /= 2;
}
h->elements[i] = ele;
}
取得并删除最大/小的元素
int delMinFromHeap(heap *h){
if(0==h->size){
cout<< "heap is empty"<< endl;
return -1;
}
int min = h->elements[1], index, tmp;
h->elements[1] = h->elements[h->size];
h->size--;
int i = 1;
while(i*2<=h->size){
index = 2*i;
if(i*2+1<=h->size && h->elements[2*i]>h->elements[2*i+1]){
index = 2*i+1;
}
if(h->elements[i]>h->elements[index]){
tmp = h->elements[i];
h->elements[i] = h->elements[index];
h->elements[index] = tmp;
i = index;
}else{
break;
}
}
return min;
}