1. 程式人生 > 其它 >1369:合併果子(fruit)

1369:合併果子(fruit)

【題目描述】

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

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

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

例如有33種果子,數目依次為11,22,99。可以先將

11、22堆合併,新堆數目為33,耗費體力為33。接著,將新堆與原先的第三堆合併,又得到新的堆,數目為1212,耗費體力為1212。所以多多總共耗費體力=3+12=15=3+12=15。可以證明15為最小的體力耗費值。

【輸入】

兩行,第一行是一個整數nn(1n300001≤n≤30000),表示果子的種類數。第二行包含nn個整數,用空格分隔,第ii個整數ai1ai20000ai(1≤ai≤20000)是第ii種果子的數目。

【輸出】

一行,這一行只包含一個整數,也就是最小的體力耗費值。輸入資料保證這個值小於231231。

【輸入樣例】

3
1 2 9

【輸出樣例】

15

【提示】

【樣例2輸入】

10
3 5 1 7 6 4 2 5 4 1

【樣例2輸出】

120

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

void show(const vector<int> &a)
{
    for (int i = 0; i < a.size(); i++) {
        cout << a[i] << " ";
    }
    cout << endl;
}

int main()
{
    // freopen("1.txt", "r", stdin);
    int n;
    cin >> n;
    vector
<int> a(n); for (int i = 0; i < n; i++) { cin >> a[i]; } // show(a); int sum = 0; greater<int> cmp = greater<int>(); make_heap(a.begin(), a.end(), cmp); while (a.size() > 1) { pop_heap(a.begin(), a.end(), cmp); int m1 = a.back(); a.pop_back(); pop_heap(a.begin(), a.end(), cmp); int m2 = a.back(); a.pop_back(); int m = m1 + m2; a.push_back(m); push_heap(a.begin(), a.end(), cmp); sum += m; } cout << sum; return 0; }