Particle Accelerator II

#include <iostream>
#include <queue>
#include <vector>

using namespace std;

int main() {
    long long n;
    cin >> n;

    vector<long long> masses(n);
    for (int i = 0; i < n; ++i) {
        cin >> masses[i];
    }

    priority_queue<long long, vector<long long>, greater<long long>> pq(masses.begin(), masses.end());

    long long totalCost = 0;

    while (pq.size() > 1) {
        long long x = pq.top(); pq.pop();
        long long y = pq.top(); pq.pop();

        long long cost = x * y;
        totalCost += cost;

        pq.push(x + y);
    }

    cout << totalCost << endl;

}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED