P1090 合并果子
本题要用到堆
一个小根堆
每次取出两堆,合并成一堆,为了让多多花费体力最少,我们要尽量少的重复大堆的合并,因此每次合并完以后,要把新的一坨放到堆里排一排,维护一个堆
有必要强调一下这个合并的操作:
(1)取出最小的一个(或一坨)果子 x
(2)再取出最小的一个(或一坨)果子 y
(3)合并为一坨 x+y
(4)体力值自然要加上 x+y 了
(5)把 x+y 放到堆里维护一下这个堆,保证下次从堆中取出小的一个或一坨
【代码】:
#include<bits/stdc++.h> using namespace std; int n,guo,heap_size,ans=0,ans1=0,x,y; int heap[10001]; void put(int d) //小根堆入堆 { int now,next; heap[++heap_size]=d; now=heap_size; while(now>1) { next=now>>1; if(heap[now]>=heap[next]) break; else swap(heap[now],heap[next]); now=next; } }
int get() //小根堆出堆 { int now,next,res; res=heap[1]; heap[1]=heap[heap_size--]; now=1; while(now*2<=heap_size) { next=now*2; if(next<heap_size&&heap[next+1]<heap[next]) next++; if(heap[now]>heap[next]) swap(heap[now],heap[next]); now=next; } return res; }
int main() { scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&guo); put(guo); } for(int i=1;i<n ;i++) //每次把两堆果子合为一堆,一共需要合并n-1次 { x=get(); //取出一堆小果子 y=get(); //再取出一堆小果子 ans+=x+y; //体力等于搬运这两个之和 put(x+y); //把这两个放进堆 } cout<<ans; return 0; }