Description
一位老木匠需要将一根长的木棒切成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体力。显然,后者比前者更省体力。
那么,木匠至少要花费多少体力才能完成切割任务呢?
Input
第1行:1个整数N(2 <= N <= 50000)
第2 - N + 1行:每行1个整数Li(1 <= Li <= 1000)。
Output
输出最小的体力消耗。
Input示例
3
3
4
5
Output示例
19
解题思路
首先逆推,每次取两个数组合起来,和为t(正向即为将长度为t的木棒切分成这两段)。根据哈夫曼思想,每次取最小的两个,然后从总体中删去这两个元素,并将其和放回总体中,由于有n 段所以切n-1 刀之后的总和为最终结果。使用优先队列来维护。
代码实现
#include<bits/stdc++.h>
using namespace std;
#define IO ios::sync_with_stdio(false);\
cin.tie(0);\
cout.tie(0);
const int maxn=107;
const int maxm=1e4+7;
priority_queue<int,vector<int>,greater<int> >qu;
int main()
{
IO;
int n,t;
cin>>n;
for(int i=0; i<n; i++)
{
cin>>t;
qu.push(t);
}
int ans=0;
for(int i=1; i<n; i++)
{
int t1=qu.top();
qu.pop();
int t2=qu.top();
qu.pop();
ans+=t1+t2;
qu.push(t1+t2);
}
cout<<ans<<endl;
return 0;
}