信息学奥赛一本通题目解析：1836：【04NOIP提高组】合并果子

【题目描述】

在一个果园里，多多已经将所有的果子打了下来，而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。

每一次合并，多多可以把两堆果子合并到一起，消耗的体力等于两堆果子的重量之和。可以看出，所有的果子经过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种果子的数目。

【输出】

一行，这一行只包含一个整数，也就是最小的体力耗费值。输入数据保证这个值小于231。

【输入样例】

3 
1 2 9

【输出样例】

15

【提示】

【数据规模】

对于30％的数据，保证有n<=1000：

对于50％的数据，保证有n<=5000；

对于全部的数据，保证有n<=10000。

【解题思路】

典型的贪心算法问题，可以通过使用优先队列（最小堆）来高效解决。优先队列可以保证每次合并都是选择当前最轻的两堆果子进行合并，这样可以保证总的体力消耗最小。以下是具体的解题步骤和思路：

1. 初始化优先队列：将所有果子的数目添加到一个优先队列（最小堆）中。优先队列的特点是能够保证每次取出的都是队列中最小的元素。

2. 合并果子：每一次从优先队列中取出两个最小的元素（即两堆最轻的果子），将它们的和作为新的一堆果子的重量，并将这个和作为这次合并的体力消耗加到总体力消耗上。然后，将新的果子重量再次放入优先队列中。重复这个过程，直到优先队列中只剩下一个元素，这个过程中累积的体力消耗就是多多合并果子所需的最小体力消耗。

3. 输出结果：当优先队列中只剩下一个元素时，循环结束，此时累积的体力消耗就是所求的最小体力消耗。

【代码实现】

#include <iostream>
#include <queue>
#include <vector>
#include <functional> // 用于 std::greater

using namespace std;

int main() {
    int n; // 果子的种类数
    cin >> n;
    priority_queue<int, vector<int>, greater<int>> pq; // 最小堆

    // 读入每种果子的数目，并加入优先队列
    for (int i = 0; i < n; ++i) {
        int weight;
        cin >> weight;
        pq.push(weight);
    }

    int totalCost = 0;
    // 当队列中至少有两个元素时进行合并操作
    while (pq.size() > 1) {
        int a = pq.top(); pq.pop();
        int b = pq.top(); pq.pop();

        int cost = a + b; // 这次合并的体力消耗
        totalCost += cost; // 累积到总体力消耗中

        pq.push(cost); // 将合并后的果子重量放回优先队列中
    }

    // 输出总体力消耗
    cout << totalCost << endl;

    return 0;
}