1.堆的基本操作:
(1)插入
(2)输出最小值
(3)删除最小值
(4)删除第k个插入的数
(5)修改第k个插入的数
堆的核心算法:down() ,up();
问题:
代码:
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=100010;
int h[N],hp[N],ph[N],size;
int n,m;
void heap_swap(int a,int b)
{
swap(h[a],h[b]);
swap(ph[hp[a]],ph[hp[b]]);
swap(hp[a],hp[b]);
}
向下排序
void down(int u)
{
int t=u;
if(2*u<=size && h[t]>h[2*u]) t=2*u;
if(2*u+1<=size && h[t]>h[2*u+1]) t=2*u+1;
if(t!=u)
{
heap_swap(t,u);
down(t);
}
}
//向上排序
void up(int u)
{
int t=u;
if(u/2 && h[t] <h[u/2]) t=u/2;
if(u!=t)
{
heap_swap(t,u);
up(t);
}
}
int main()
{
cin>>n;
while(n--)
{
char op[10];
cin>>op;
int k,x;
if(!strcmp(op,"I"))
{
cin>>x;
size++;
m++;
ph[m]=size; //第M个插入的数是size
hp[size]=m;// size对应第M个插入的数
h[size]=x;
up(size);
}
else if(!strcmp(op,"PM")) cout<<h[1]<<endl;
else if(!strcmp(op,"DM"))
{
heap_swap(1,size);
size--;
down(1);
}
else if(!strcmp(op,"D"))
{
cin>>k;
k=ph[k]; //找到第k个插入的数
heap_swap(k,size);
size--;
up(k);
down(k);
}
}
return 0;
}