7-29 修理牧場(25 分)

7-29 修理牧場(25 分)

農夫要修理牧場的一段柵欄,他測量了柵欄,發現需要N塊木頭,每塊木頭長度爲整數Li個長度單位,於是他購買了一條很長的、能鋸成N塊的木頭,即該木頭的長度是Li的總和。

但是農夫自己沒有鋸子,請人鋸木的酬金跟這段木頭的長度成正比。爲簡單起見,不妨就設酬金等於所鋸木頭的長度。例如,要將長度爲20的木頭鋸成長度爲8、7和5的三段,第一次鋸木頭花費20,將木頭鋸成12和8;第二次鋸木頭花費12,將長度爲12的木頭鋸成7和5,總花費爲32。如果第一次將木頭鋸成15和5,則第二次鋸木頭花費15,總花費爲35(大於32)。

請編寫程序幫助農夫計算將木頭鋸成N塊的最少花費。

輸入格式:

輸入首先給出正整數N104),表示要將木頭鋸成N塊。第二行給出N個正整數(50),表示每段木塊的長度。

輸出格式:

輸出一個整數,即將木頭鋸成N塊的最少花費。

輸入樣例:

8
4 5 1 2 1 3 1 1

輸出樣例:

49


就是每次取最小的兩個數,可以用哈夫曼樹來寫,但是有點麻煩,優先隊列是最簡單的。

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
long long ans;
int N;
int main()
{
	cin>>N;
	priority_queue<int,vector<int>,greater<int> > q;
	for(int i=0;i<N;i++)
	{
		int a;
		cin>>a;
		q.push(a);
	}
	while(q.size()>1)
	{
		int x=q.top();
		q.pop();
		int y=q.top();
		q.pop();
		q.push(x+y);
		ans+=x+y;
	}
	cout<<ans<<endl;
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章