洛谷 P1090 合併果子 優先隊列

洛谷 P1090 合併果子 優先隊列

https://www.luogu.com.cn/problem/P1090

題目描述
在一個果園裏，多多已經將所有的果子打了下來，而且按果子的不同種類分成了不同的堆。多多決定把所有的果子合成一堆。

每一次合併，多多可以把兩堆果子合併到一起，消耗的體力等於兩堆果子的重量之和。可以看出，所有的果子經過 n−1 次合併之後， 就只剩下一堆了。多多在合併果子時總共消耗的體力等於每次合併所耗體力之和。

因爲還要花大力氣把這些果子搬回家，所以多多在合併果子時要儘可能地節省體力。假定每個果子重量都爲 1 ，並且已知果子的種類 數和每種果子的數目，你的任務是設計出合併的次序方案，使多多耗費的體力最少，並輸出這個最小的體力耗費值。

例如有 3 種果子，數目依次爲 1 ， 2 ， 9 。可以先將 1 、 2 堆合併，新堆數目爲 3 ，耗費體力爲 3 。接着，將新堆與原先的第三堆合併，又得到新的堆，數目爲 12 ，耗費體力爲 12 。所以多多總共耗費體力 =3+12=15 。可以證明 15 爲最小的體力耗費值。

輸入格式
共兩行。
第一行是一個整數 n(1≤n≤10000) ，表示果子的種類數。

第二行包含 n 個整數，用空格分隔，第 i 個整數 ai(1≤ai≤20000) 是第 i 種果子的數目。

輸出格式
一個整數，也就是最小的體力耗費值。輸入數據保證這個值小於 2^{31}
輸入輸出樣例
輸入
3
1 2 9
輸出
15

代碼：

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

typedef long long ll;
const long long mod = 1e9 + 7;
const int INF = 0x3f3f3f3f;
const double PI = 3.141592;
const int e = 1e4;

int main()
{
    ll n;
    scanf("%lld", &n);
	//小根堆
    priority_queue<ll, vector<ll>, greater<ll> > q;
    for(int i = 0; i < n; i++){
        ll t;
        scanf("%lld", &t);
        q.push(t);
    }
    ll ans = 0;
    ll num = n - 1;
    while(num--){
        ll temp = 0;
        temp += q.top();
        q.pop();
        temp += q.top();
        q.pop();
        q.push(temp);
        ans += temp;
    }
    cout << ans << endl;
    return 0;
}