合并果子

 1 #include<iostream>
 2 #include<algorithm> 
 3 using namespace std;
 4 int a[100001];
 5 int tot=0;
 6 int main()
 7 {
 8     int n;
 9     cin>>n;
10     for(int i=1;i<=n;i++)
11     {
12         cin>>a[i];
13     }
14     make_heap(a+1,a+n+1,greater<int>());//建立小根堆 
15     int m=n;//取出n的大小，方便进行维护 
16     for(int i=1;i<n;i++)
17     {
18         int a1=a[1];//取出最小元素 
19         pop_heap(a+1,a+m+1,greater<int>());//删除最小的并维护 
20         int a2=a[1];//其次小的元素 
21         pop_heap(a+1,a+m,greater<int>());//取出其次小的并进行维护 ， 
22         int sum=a1+a2;//算出需要的体力 
23         tot=tot+sum;
24         a[m-1]=sum;//将生成的元素再次放回 
25         push_heap(a+1,a+m,greater<int>());//再次维护 
26         m--;
27     }
28         
29     cout<<tot;
30     return 0;
31 }