1. 程式人生 > >合併果子加強版(C++ 貪心)

合併果子加強版(C++ 貪心)

23:合併果子
總時間限制: 1000ms 記憶體限制: 65535kB
描述
在一個果園裡,多多已經將所有的果子打了下來,而且按果子的不同種類分成了不同的堆。多多決定把所有的果子合成一堆。
每一次合併,多多可以把兩堆果子合併到一起,消耗的體力等於兩堆果子的重量之和。可以看出,所有的果子經過n-1次合併之後,就只剩下一堆了。多多在合併果子時總共消耗的體力等於每次合併所耗體力之和。
因為還要花大力氣把這些果子搬回家,所以多多在合併果子時要儘可能地節省體力。假定每個果子重量都為1,並且已知果子的種類數和每種果子的數目,你的任務是設計出合併的次序方案,使多多耗費的體力最少,並輸出這個最小的體力耗費值。
例如有3種果子,數目依次為1,2,9。可以先將1、2堆合併,新堆數目為3,耗費體力為3。接著,將新堆與原先的第三堆合併,又得到新的堆,數目為12,耗費體力為12。所以多多總共耗費體力=3+12=15。可以證明15為最小的體力耗費值。

輸入
輸入包括兩行,第一行是一個整數n(1<=n<=10000),表示果子的種類數。第二行包含n個整數,用空格分隔,第i個整數ai(1<=ai<=20000)是第i種果子的數目。
輸出
輸出包括一行,這一行只包含一個整數,也就是最小的體力耗費值。輸入資料保證這個值小於2的31次方。
樣例輸入

3 
1 2 9 

樣例輸出

15

提示
n≤10000
a[i]≥1且a[i]∈INT

/*
        由於本題資料較大,所以所有資料都要開成long long型別(雖然可能有點多餘)
    本題貪心思想為:先將每堆果子的重量從小到大的排序,然後合併時,將每兩堆合併的
    結果進行比較,在進行累加,就能保證最少。
*/
#include<algorithm> #include<iostream> #include<cstdio> using namespace std; bool cmp(const int x,const int y) { return x>y; } long long i,n,s,k,a[10005]; int main() { scanf("%lld",&n); for(i=1;i<=n;i++) scanf("%lld",&a[i]); sort(a+1,a+n+1,cmp); //從小到大排序
for(i=n;i>1;i--) { a[i-1]+=a[i]; s+=a[i-1]; //合併 k=i-1; while(k!=1&&a[k]>a[k-1]) //每堆合併後進行比較 { swap(a[k],a[k-1]);//如果大了,就交換 k--; } } printf("%lld",s);輸出s(答案) return 0; }