题意:农夫锯木头,将一根木头锯成要求的n段,每一次锯木头的花费为本次所锯的木头的长度,问如果锯成要求的n段需要的花费总共为多少。
贪心,数据刚开始给出n段木头,将n段木头加入队列,可以每次将最短的两段合并起来,将合并后的加入队列并累加花费,然后继续上述操作,一直到队列中剩下一根木头。
这里用到了优先队列会方便很多,每次将最小的出队。
priority_queue<int,vector<int>,greater<int> >que3;//最小值优先出列; priority_queue<int,vector<int>,less<int> >que4;//最大值优先出列;
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#define INF 9999999999
#define N 20010
typedef long long LL;
using namespace std;
priority_queue< LL,vector<LL>,greater<LL> >q;
LL wood(int n)
{
LL x,y,z,ans=0;
while(q.size()>1)
{
x=q.top();
q.pop();
y=q.top();
q.pop();
z=x+y;
q.push(z);
ans+=z;
}
return ans;
}
int main()
{
int i,n,num;
while(~scanf("%d",&n))
{
while(!q.empty())
q.pop();
for(i=0; i<n; i++)
{
scanf("%d",&num);
q.push(num);
}
if(q.size()==1)
cout<<num<<endl;
else
{
LL ans=wood(n);
cout<<ans<<endl;
}
}
return 0;
}