一位老木匠需要将一根长的木棒切成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体力。显然,后者比前者更省体力。
那么,木匠至少要花费多少体力才能完成切割任务呢?
收起
输入
第1行:1个整数N(2 <= N <= 50000) 第2 - N + 1行:每行1个整数Li(1 <= Li <= 1000)。
输出
输出最小的体力消耗。
输入样例
3 3 4 5
输出样例
19
这样想这个问题,从需要切成的木块当中取出最短的两个然后合并在一起,加入队列,合并起来的长度便是花费,然后依次遍历(有点类似于哈夫曼树),得到的最后花费就是总花费。
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<queue>
#include<string>
#include<iostream>
#include<vector>
using namespace std;
long long n,a[51000],sum,ans,t;
int main()
{
scanf("%lld",&n);
sum=ans=0;
priority_queue<long long,vector<long long>,greater<long long> > q;
for(int i=0; i<n; i++)
{
scanf("%lld",&a[i]);
q.push(a[i]);
}
while(q.size()!=1)
{
t=q.top();
q.pop();
ans=t+q.top();
q.pop();
sum=sum+ans;
q.push(ans);
}
printf("%lld\n",sum);
return 0;
}