https://www.luogu.com.cn/problem/P1334
思路
上来就想着,从大到小排序一下,然后每次砍掉最大长度的那一段,不就ok了吗?码之,WA了。不出意料
仔细一想,切割的时候可以从中间切开,迅速缩小长度,每次砍掉最大长度看来并不是最优解。
每次选一根木棒砍成两段,所消耗能量是两段木棒的长度和。我们用逆向思维:每次选两根木棒拼在一起,使得长度和最小,这样能量也就达到最小。这就是一个合并果子的变形而已。。。直接就优先队列冲了。
代码
#include <bits/stdc++.h>
using namespace std;
const int maxn=2e4+5;
int n;
typedef long long ll;
priority_queue<ll,vector<ll>,greater<ll> >q;
int main()
{
scanf("%d",&n);
int i,num;
for(i=1;i<=n;i++)
{
scanf("%d",&num);
q.push(num);
}
ll ans=0;
for(i=1;i<=n-1;i++)
{
ll t1=q.top();
q.pop();
ll t2=q.top();
q.pop();
ans+=t1+t2;
q.push(t1+t2);
}
printf("%lld\n",ans);
return 0;
}