合并果子 / Fence Repair G
题目链接:ybt高效进阶4-1-1 / luogu P1090
题目大意
有一堆东西,每次你可以选两个东西,用它们大小的和的代价,把它们合并,得到一个它们大小和的东西。
然后把它们合并成一个东西所要的最小的代价。
思路
我们考虑贪心,让每次合并的费用都尽可能小。
自然想到可以每次选已有石头中最小的两个合并。
然后可以用堆维护,然后就可以了。
代码
#include<queue>
#include<cstdio>
#define ll long long
using namespace std;
int n;
priority_queue <ll, vector<ll>, greater<ll> > q;
ll ans, x, y;
int main() {
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%lld", &x);
q.push(x);
}
while (!q.empty()) {
x = q.top();
q.pop();
y = q.top();
q.pop();
ans += x + y;//代价
if (q.empty()) {//只剩一个
break;
}
q.push(x + y);//得到新的东西
}
printf("%lld", ans);
return 0;
}