最大堆参考资料:堆和堆排序
实现一个最大堆,堆的建立可以使用数组或者一个一个添加,可以通过提取堆顶元素实现降序排序,而原位排序则可以实现升序排序,这里使用原位排序。
MaxHeap.h
#pragma once
#include<iostream>
using namespace std;
class MaxHeap
{
private:
int maxsize;//记录堆允许的最大容量
int count;//记录堆目前已插入元素个数
int* a;//堆使用数组保存,记录数组指针
void shiftUp(int n);//向上交换元素,保持堆特性,不需向外暴露
void shiftDown(int n);//向下交换元素
public:
MaxHeap(int size);//建立空堆,元素通过后期一一添加
MaxHeap(int*pa, int n);//根据已有数组建立堆
void add(int value);//插入元素
int extractTop();//提取堆顶元素
void sort();/*对数组排序,但是会提取出堆内所有元素,
也就是这个堆理论上已经空了,但是堆的数组存储的元素已经有序*/
};
MaxHeap.cpp
#include"MaxHeap.h"
MaxHeap::MaxHeap(int size)
{
count = 0;
maxsize = size;
a = new int[size];//申请堆空间
}
void MaxHeap::add(int value)
{
if (count >= maxsize)
{
cout << "堆已满,操作失败!" << endl;
return;
}
a[count] = value;
count++;//注意增加堆已用容量
shiftUp(count - 1);//保持堆的特性
for (int i = 0; i < count; i++)
cout << a[i] << " ";
cout << endl;
}
void MaxHeap::shiftUp(int n)
{
if ((n>0)&&(a[n] > a[(n - 1) / 2]))//递归出口,当子节点不大于父节点或者n等于0时,结束递归
{
int temp = a[n];
a[n] = a[(n - 1) / 2];
a[(n - 1) / 2] = temp;
shiftUp((n - 1) / 2);
}
}
int MaxHeap::extractTop()
{
int val = a[0];
a[0] = a[count - 1];
count--;
shiftDown(0);
return val;
}
void MaxHeap::shiftDown(int n)
{
if (n * 2 + 2 < count)//一般情况:左右子节点都存在
{
int temp;
if ((a[n * 2 + 1] >= a[n * 2 + 2]) && (a[n] < a[n * 2 + 1]))
{
temp = a[n];
a[n] = a[n * 2 + 1];
a[n * 2 + 1] = temp;
shiftDown(n * 2 + 1);
}
if ((a[n * 2 + 1] < a[n * 2 + 2]) && (a[n] < a[n * 2 + 2]))
{
temp = a[n];
a[n] = a[n * 2 + 2];
a[n * 2 + 2] = temp;
shiftDown(n * 2 + 2);
}
}
if ((n * 2 + 1 == count - 1)&& a[n] < a[n * 2 + 1])//只存在左子节点
{
int temp;
temp = a[n];
a[n] = a[n * 2 + 1];
a[n * 2 + 1] = temp;
shiftDown(n * 2 + 1);
}
}
MaxHeap::MaxHeap(int*pa, int n)
{
a = pa;
maxsize = n;
count = 0;
for (int i = 0; i < n; i++)
add(a[i]);/*注意,这里是原位传递,最大堆是在原有数组空间上建立的,
传递入堆的a[i]会对i位置及i之前的位置重新排序,但是不会影响i后面的数据;
注意理论上的堆与实际的数据存储之间的关系*/
}
void MaxHeap::sort()
{
cout << "排序:" << endl;
int n = count;
for (int i = n - 1; i >= 0; i--)
{
a[i] = extractTop();/*在提取堆顶元素的方法中,对i个元素在理论上的堆内已经被清空,因此可以用于原位存储排序*/
}
for (int i = 0; i < n; i++)
{
cout << a[i] << " ";
}
cout << endl;
}
Main.cpp(客户程序验证)
#include"MaxHeap.h"
#include<time.h>//用于产生随机数
int main()
{
srand((unsigned)time(NULL));
MaxHeap h(10);
for (int i = 0; i < 10; i++)
{
int temp = rand() % 100;//产生100以内的随机数
cout << temp << endl;
h.add(temp);
}
h.sort();
system("pause");
return 0;
}