合并果子(堆模板)

#include
#include
#include
#include
using namespace std;
const int maxn=1e5+10;
int cnt,h[maxn],a[maxn],n;
void put(int x){
    h[++cnt]=x;
    int now=cnt,nxt;
    while(now>1){
        nxt=now>>1;
        if(h[now]>=h[nxt]) return ;
        swap(h[now],h[nxt]);
        now=nxt;
    }
}
int get(){
    int res=a[1];
    a[1]=a[cnt--];
    int now=1,nxt;
    while(now*2<=cnt){
        nxt=now*2;
        if(nxt+1<=cnt&&h[nxt+1]         if(a[now]<=a[nxt]) break;
        swap(a[now],a[nxt]);
        now=nxt;
    }
    return res;
}
int main(){
    long long s=0;
    cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        put(a[i]);      
    }
    for(int i=1;i         int b=get(); int c=get();
        cout<         int cur=c+b;
        cout<         s+=b+c;
        put(cur);
    }
    cout<     return 0;
}

你可能感兴趣的:(数据结构,noip冲刺复习)