以最大堆为例,实现堆的初始化、删除、插入等基本操作。
以数组的形式保存堆,则对堆的 i 节点而言,其父节点是 (i-1)/2,子节点是 2*i+1 和 2*i+2 。
void Swap(int &a,int &b)
{
int temp=a;
a=b;
b=temp;
}
//数组maxheap第n-1位(含)前的数字已经是最大堆,现在往堆中加入数组第n位的数据,使得堆的大小加1
bool MaxHeapFixup(vector<int> &maxheap,int n)
{
if(n>=maxheap.size())
return false;
//若父节点小于子节点,则交换两者数值
for(int i=(n-1)/2;i>=0 && maxheap[i]<maxheap[n];n=i,i=(n-1)/2)
Swap(maxheap[n],maxheap[i]);
return true;
}
//将第i位的数据自上而下,“下沉”到合理位置
bool MaxHeapFixdown(vector<int> &maxheap,int i)
{
if(2*i+1>=maxheap.size())//没有子节点,无需下沉
return false;
int child=2*i+1;
while(child<maxheap.size())
{
if(child+1<maxheap.size() && maxheap[child+1]>maxheap[child])//找出最大的孩子节点
child++;
if(maxheap[i]<maxheap[child])
Swap(maxheap[i],maxheap[child]);
else
break;
i=child;
child=2*i+1;
}
return true;
}
//删除最大堆的最大值,即删除数组第一个数
bool MaxHeapDeleteNumber(vector<int> &maxheap)
{
if(maxheap.empty())
return false;
Swap(maxheap[0],maxheap[maxheap.size()-1]);//交换头尾两个数
maxheap.erase(maxheap.end()-1);//删除最后一个数
MaxHeapFixdown(maxheap,0);
return true;
}
//初始化最大堆方法1
bool InitMaxHeap1(vector<int> &a)
{
if(a.empty())
return false;
for(int i=1;i<a.size();i++)
MaxHeapFixup(a,i);
return true;
}
//初始化最大堆方法2(推荐)
bool InitMaxHeap2(vector<int> &a)
{
if(a.empty())
return false;
for(int i=a.size()/2-1;i>=0;i--)
MaxHeapFixdown(a,i);
return true;
}