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块的最少花费。

输入格式:

输入首先给出正整数N(≤),表示要将木头锯成N块。第二行给出N个正整数(≤),表示每段木块的长度。

输出格式:

输出一个整数,即将木头锯成N块的最少花费。

输入样例:

8
4 5 1 2 1 3 1 1

输出样例:

49

 1 #include<stdio.h>
 2 #include<queue>
 3 using namespace std;
 4 priority_queue<int, vector<int>,greater<int> > Q; //建立一个小顶堆
 5 //priority_queue<int> Q;  //默认建立一个大顶堆
 6 
 7 int main()
 8 {
 9     int n;
10     int ans,temp1,temp2;
11     int i;
12 
13     while( scanf("%d",&n)!=EOF)
14     {
15         while( Q.empty()==false) Q.pop();  //清除堆中元素
16 
17         for( i=1; i<=n; i++)
18         {
19             scanf("%d",&temp1);
20             Q.push(temp1);
21         }
22         ans = 0;
23         while( Q.size()>1)   //当堆中元素个数大于1
24         {
25             temp1 =Q.top();
26             Q.pop();
27             temp2 =Q.top();
28             Q.pop();          //取出堆中两个最小的元素
29             ans += temp1+temp2;
30             Q.push( temp1+temp2);  //将该双亲结点的权值放回堆中
31         }
32         printf("%d\n",ans);
33     }
34     return 0;
35 }

 

posted @ 2018-02-10 08:43  yuxiaoba  阅读(916)  评论(0编辑  收藏  举报