首页 > 其他 > 详细

洛谷P1090 合并果子

时间:2019-02-13 21:10:43      阅读:163      评论:0      收藏:0      [点我收藏+]

题目

 

初看这题,一开始脑子抽筋,排序一边就直接逐步累加求和,结果过了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;
}

优先队列放光芒!

洛谷P1090 合并果子

原文:https://www.cnblogs.com/Benjamin-cpp/p/10371762.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!