UVA 11136 - Hoax or what(STL / 最大最小堆)

题目链接 https://cn.vjudge.net/problem/UVA-11136

【题意】
沃尔玛超市有n天做活动,每天有k笔交易. 每次交易都将含有交易额的小票放在一个箱子中抽奖. 每天从箱子中抽取最大和最小交易额的小票,并支出它们两个差的金额的钱给最大交易额的顾客. 求n天沃尔玛超市总共支出.(1<=n<=5000, 0<=k<=1e5, 小票金额不超过1e6)

【思路】
用multiset或者最大最小堆维护最值即可

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

int n,k;
multiset<int> s;

int main(){
    while(scanf("%d",&n)==1 && n){
        s.clear();
        ll ans=0;
        while(n--){
            scanf("%d",&k);
            while(k--){
                int x;
                scanf("%d",&x);
                s.insert(x);
            }
            if(s.size()>=2){
                ans+=(ll)*s.rbegin()-(ll)*s.begin();
                multiset<int>::iterator it=s.end();
                --it;
                s.erase(s.begin());
                s.erase(it);
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1000005;

int n,k;
int c[maxn];
priority_queue<int,vector<int>,greater<int> > minq;
priority_queue<int> maxq;

int main(){
    while(scanf("%d",&n)==1 && n){
        while(minq.size()) minq.pop();
        while(maxq.size()) maxq.pop();
        memset(c,0,sizeof(c));
        ll ans=0;
        while(n--){
            scanf("%d",&k);
            while(k--){
                int x;
                scanf("%d",&x);
                ++c[x];
                minq.push(x);
                maxq.push(x);
            }
            while(minq.size() && c[minq.top()]==0) minq.pop();
            while(maxq.size() && c[maxq.top()]==0) maxq.pop();
            if(minq.size() && maxq.size()){
                ans+=(ll)maxq.top()-(ll)minq.top();
                --c[maxq.top()];
                --c[minq.top()];
                maxq.pop();
                minq.pop();
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}
posted @ 2018-08-16 09:24  不想吃WA的咸鱼  阅读(137)  评论(0编辑  收藏  举报