洛谷P1090 合并果子

题目

 

初看这题，一开始脑子抽筋，排序一边就直接逐步累加求和，结果过了1个点。。。

---

极度不爽， 每次加一个排序， 结果10个TLE...

---

冷静下来好好想想，就想到了一样东西：

c++党的福音：STL中的优先队列！！！

 

接下来的思路很简单，贪心。 每一次取出队首的两个数a, b（这里要重载小于号，故队首的两个值为最小的两堆）， 并成一堆， sum = a + b，ans += sum；这样每次的力气花费就累加了。

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
struct node{
    bool operator () (int &a, int &b){
        return a > b;
    }
};//node结构体的作用是重载<;
priority_queue <int, vector<int>, node> q;//定义一个优先队列
int main() {
    int n;
    scanf("%d", &n);
    for(int i = 0; i < n; i++) {
        int x;
        scanf("%d", &x);
        q.push(x); 
    }
    int ans = 0;
    int sum = 0;
    //重点来了：
    while(q.size() != 1) {
        int a = q.top();
        q.pop();
        int b = q.top();
        q.pop();
        sum = a + b;
        ans += sum;
        q.push(sum); 
    }
    printf("%d\n", ans);
    return 0;
}

优先队列放光芒！