堆
如果有一个关键码的集合K = {k0,k1, k2,…,kn-1},把它的所有元素按完全二叉树的顺序存储方式存储
在一个一维数组中,并满足:Ki <= K2i+1 且 Ki<= K2i+2 (Ki >= K2i+1 且 Ki >= K2i+2) i = 0,1,2…,则称为
小堆(或大堆)。将根节点最大的堆叫做最大堆或大根堆,根节点最小的堆叫做最小堆或小根堆。
●性质
- 堆中某个节点的值总是不大于或不小于其父节点的值;
- 堆总是一棵完全二叉树
●堆的实现及堆排序
common.h
#ifndef _COMMON_H_
#define _COMMOM_H_
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
#include <stdbool.h>
#include <memory.h>
#include <string.h>
#define ElemType char
void Swap(ElemType *a, ElemType *b)
{
ElemType temp = *a;
*a = *b;
*b = temp;
}
#endif /*_COMMOM_H_*/
heap.h
#ifndef _HEAP_H_
#define _HEAP_H_
#include"common.h"
#define HeapDataType int
typedef struct Heap
{
HeapDataType *base;
int capacity;
int size;
}Heap;
void HeapInit(Heap *php, int n);
void MinHeapInsert(Heap *php, HeapDataType x);
HeapDataType MinHeapRemove(Heap *php);
bool HeapEmpty(Heap *php);
void HeapPrint(Heap *php);
void HeapSort(Heap *php);
void AdjustUp(HeapDataType *base, int start);
void AdjustDown(HeapDataType *base, int start, int n);
//初始化
void HeapInit(Heap *php, int n)
{
assert(php != NULL);
php->base = (HeapDataType*)malloc(sizeof(HeapDataType)* n);
assert(php->base != NULL);
php->capacity = n;
php->size = 0;
}
bool HeapEmpty(Heap *php)
{
assert(php != NULL);
return php->size == 0;
}
//插入元素
void MinHeapInsert(Heap *php, HeapDataType x)
{
assert(php != NULL);
if (php->size < php->capacity)
{
php->base[php->size] = x;
//调整为小堆结构 从下往上调整
AdjustUp(php->base, php->size);
php->size++;
}
}
//小堆
HeapDataType MinHeapRemove(Heap *php)
{
assert(php != NULL);
assert(php->size > 0);
int heaptop_val = php->base[0];
php->size--;
php->base[0] = php->base[php->size];
AdjustDown(php->base, 0, php->size);
return heaptop_val;
}
void HeapPrint(Heap *php)
{
for (int i = 0; i<php->size; ++i)
printf("%d ", php->base[i]);
printf("\n");
}
//向上插入
void AdjustUp(HeapDataType *base, int start)
{
int j = start;
int i = (j - 1) / 2;
HeapDataType tmp = base[j];
while (j > 0)
{
if (tmp < base[i])
{
base[j] = base[i]; //覆盖
j = i;
i = (j - 1) / 2;
}
else
break;
}
base[j] = tmp;
}
//向下插入
void AdjustDown(HeapDataType *base, int start, int n)
{
int i = start;
int j = 2 * i + 1;
while (j < n)
{
if (j + 1<n && base[j]>base[j + 1]) //找出左右子树较小的值
j++;
if (base[i] > base[j])
{
Swap(&base[i], &base[j]);
i = j;
j = 2 * i + 1;
}
else
break;
}
}
//堆排序
void HeapSort(Heap *php, int ar[], int n)
{
for (int i = 0; i<n; ++i)
php->base[i] = ar[i];
php->size = n;
//调整成小堆
int curpos = n / 2 - 1; //找到最后一个分支
while (curpos >= 0)
{
AdjustDown(php->base, curpos, n);
curpos--;
}
//排序
int end = n - 1;
while (end > 0)
{
Swap(&php->base[0], &php->base[end]);
AdjustDown(php->base, 0, end);
end--;
}
//HeapPrint(php);
int k = php->size - 1;
for (int i = 0; i<n; ++i)
{
ar[i] = php->base[k--];
}
}
#endif /* _HEAP_H_ */