该题来自POJ3253
另外两道可用优先队列解决的贪心算法:http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1163
http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1191
原题如下:
一位老木匠需要将一根长的木棒切成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
分析:可以这样思考,因为每次切割的花费是当前切割总长。所以,肯定希望先把比较长的木板切了,比较短的留在后头。
那么逆向分析,把最终要完成的木板拿出最短的两块,合成一块长木板,加入队列。该木板的长度也就是需要的花费。依次类推直到队列为空。
代码如下
#include<cstdio>
#include<iostream>
#include<cstring>
#include <vector>
#include <algorithm>
#include <queue>
#define MAXN 50010
using namespace std;
priority_queue<int,vector<int>,greater<int> >que;
int a[MAXN];
int main()
{
int n;
long long cost=0;
cin>>n;
for(int i=0;i<n;i++)
{
cin>>a[i];
que.push(a[i]);
}
while(que.size()!=1)
{
int a=que.top();
que.pop();
int b=que.top();
que.pop();
int c=a+b;
cost+=c;
que.push(c);
}
printf("%lld\n",cost);
}