九度OJ題目1107：搬水果

題目1107：搬水果

時間限制：1 秒

內存限制：32 兆

特殊判題：否

題目描述：

    在一個果園裏，小明已經將所有的水果打了下來，並按水果的不同種類分成了若干堆，小明決定把所有的水果合成一堆。每一次合併，小明可以把兩堆水果合併到一起，消耗的體力等於兩堆水果的重量之和。當然經過 n‐1 次合併之後，就變成一堆了。小明在合併水果時總共消耗的體力等於每次合併所耗體力之和。

    假定每個水果重量都爲 1，並且已知水果的種類數和每種水果的數目，你的任務是設計出合併的次序方案，使小明耗費的體力最少，並輸出這個最小的體力耗費值。例如有 3 種水果，數目依次爲 1，2，9。可以先將 1，2 堆合併，新堆數目爲3，耗費體力爲 3。然後將新堆與原先的第三堆合併得到新的堆，耗費體力爲 12。所以小明總共耗費體力=3+12=15，可以證明 15 爲最小的體力耗費值。

輸入：

    每組數據輸入包括兩行,第一行是一個整數 n(1<=n<=10000),表示水果的種類數，如果 n 等於 0 表示輸入結束，且不用處理。第二行包含 n 個整數，用空格分隔，第 i 個整數(1<=ai<=1000)是第 i 種水果的數目。

輸出：

對於每組輸入，輸出一個整數並換行，這個值也就是最小的體力耗費值。輸入數據保證這個值小於 2^31。

樣例輸入：

3
9 1 2
0

樣例輸出：

15

來源：

2011年吉林大學計算機研究生機試真題

#include <stdio.h>
#include <queue>//使用優先隊列
using namespace std;
priority_queue<int,vector<int>,greater<int> > q;//定義一個小頂堆

int main(){
	int n;
	while(scanf("%d",&n)!=EOF&&n!=0){ //輸入n
		while(q.empty()==false)  q.pop();//清空堆中元素
		for(int i=0;i<n;i++){ //輸入n個葉子結點權值
			int x;
			scanf("%d",&x);
			q.push(x); //將權值放入堆中
		}
		int ans=0; //保存答案
		while(q.size()>1){ //當堆中元素大於1個
			int a=q.top();
			q.pop();
			int b=q.top();
			q.pop(); //取出堆中兩個最小元素
			ans+=a+b; //累加權值
			q.push(a+b);//將該雙親結點的權值放回堆中
		}
		printf("%d\n",ans); //輸出答案
	}
	return 0;
} 

/************************************************************** 
    Problem: 1107 
    User: zpy
    Language: C++ 
    Result: Accepted 
    Time:30 ms 
    Memory:1024 kb 
****************************************************************/