← 返回题目页

A0499 合并果子2

标程

#include <bits/stdc++.h>
using namespace std;

// 合并果子:每次合并两堆最小的,代价是两堆重量之和。
// 用小根堆不断取出最小的两堆、合并后放回。

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n;
    cin >> n;
    priority_queue<long long, vector<long long>, greater<long long>> pq;
    for (int i = 0; i < n; i++) {
        long long x;
        cin >> x;
        pq.push(x);
    }
    long long ans = 0;
    while (pq.size() > 1) {
        long long a = pq.top(); pq.pop();
        long long b = pq.top(); pq.pop();
        long long c = a + b;
        ans += c;
        pq.push(c);
    }
    cout << ans << "\n";
    return 0;
}

动画演示