洛谷[1090]合并果子

@合并果子

题目大意

  有n堆果子，第i堆果子数量为C[i]。每次合并操作能合并任意两堆果子a与b，消耗的能量为C[a]+C[b]。
  目标是不断进行合并操作将所有果子合并为一堆，求出如何才能使消耗的能量最小。
 

输入输出

输入第一行为果子堆数n
输入第二行有n个整数，表示第i堆果子的数量

输出有一行，消耗的最小能量（保证结果小于2^31）
 

算法讨论

  这题应该使用贪心来解，证明如下：
  设果子编号为1-n，并按顺序合并，则：
    第一堆果子被合并了n-1+0次
    第二堆果子被合并了n-2+1次
    第三堆果子被合并了n-3+1次
    …你可以用手指头模拟试试…
    第n堆果子被合并了n-n+1次
  所以 当i>1时 第i堆果子被合并了n-i+1次
     当i=1时 第i堆果子被合并了n-i+0次
  可见，合并次序越靠前的果子堆被合并的次数越多，造成的能量消耗也就越多
  而合并次序越靠后的果子堆被合并的次数越少，造成的能量消耗也就越少
  那么：如果果子堆a的果子数量大于果子堆b的果子数量，且a在b前被合并，我们就应当将他们的合并顺序对调
  不断执行上述操作，我们就会得出一个果子数量递增的序列
  因此，我们应该按照果子数量从小到大合并果子堆
 

数据结构

  既然我们要不断地取出果子堆中最小的两堆进行合并，那么我们就可以用二叉堆（优先队列）来存储果子堆，这样时间复杂度就降到了O(nlogn)。
 

代码实现

#include <iostream> 
#include <cstdio>
#include <queue>
using namespace std;

struct Int{
    int n;
    bool operator < (const Int & a) const{
        return n > a.n; //重载运算符
    }
}node;

int main(){

    priority_queue<Int> que; //用来存果子的小根堆
    int n, Result=0;
    scanf("%d", &n);
    for(int i=0; i<n; i++){
        scanf("%d", &node.n);
        que.push(node); //推入所有果子堆
    }

    while(que.size()>1){
        node.n = que.top().n;  que.pop(); //合并
        node.n += que.top().n; que.pop(); //果子
        que.push(node); //推入新果子堆
        Result += node.n; //计算消耗的能量
    }
    printf("%d", Result);

    return 0;
}