题意
一位老木匠需要将一根长的木棒切成N段。每段的长度分别为L1,L2,……,LN(1 <= L1,L2,…,LN <= 1000,且均为整数)个长度单位。我们认为切割时仅在整数点处切且没有木材损失。
木匠发现,每一次切割花费的体力与该木棒的长度成正比,不妨设切割长度为1的木棒花费1单位体力。例如:若N=3,L1 = 3,L2 = 4,L3 = 5,则木棒原长为12,木匠可以有多种切法,如:先将12切成3+9.,花费12体力,再将9切成4+5,花费9体力,一共花费21体力;还可以先将12切成4+8,花费12体力,再将8切成3+5,花费8体力,一共花费20体力。显然,后者比前者更省体力。
那么,木匠至少要花费多少体力才能完成切割任务呢?
解题思路
Huffman树,先将最小的两个数从集合中拿出来,然后将和放入集合,然后又继续取出两个最小的数,将和放入集合中,一直重复,直到剩下一个数。所花费的体力就是所有求的过程中产生和的和。为了可以取出两个最小的数,采用了优先队列。
参考代码
#include <iostream>
#include <queue>
using namespace std;
int main(){
priority_queue<int, vector<int>, greater<int> > q;
int n,tmp;
while (cin>>n){
for (int i=0;i<n;i++){
cin>>tmp;
q.push(tmp);
}
int ans=0;
while (!q.empty()){
int x=q.top();
q.pop();
if (q.empty()) break;
int y=q.top();
q.pop();
q.push(x+y);
ans+=(x+y);
}
cout<<ans<<endl;
}
return 0;
}