bzoj 1691: [Usaco2007 Dec]挑剔的美食家

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 621  Solved: 280
[Submit][Status][Discuss]

Description

与很多奶牛一样，Farmer John那群养尊处优的奶牛们对食物越来越挑剔，随便拿堆草就能打发她们午饭的日子自然是一去不返了。现在，Farmer John不得不去牧草专供商那里购买大量美味多汁的牧草，来满足他那N(1 <= N <= 100,000)头挑剔的奶牛。 所有奶牛都对FJ提出了她对牧草的要求：第i头奶牛要求她的食物每份的价钱不低于A_i(1 <= A_i <= 1,000,000,000)，并且鲜嫩程度不能低于B_i(1 <= B_i <= 1,000,000,000)。商店里供应M(1 <= M <= 100,000)种不同的牧草，第i 种牧草的定价为C_i(1 <= C_i <= 1,000,000,000)，鲜嫩程度为D_i (1 <= D_i <= 1,000,000,000)。 为了显示她们的与众不同，每头奶牛都要求她的食物是独一无二的，也就是说，没有哪两头奶牛会选择同一种食物。 Farmer John想知道，为了让所有奶牛满意，他最少得在购买食物上花多少钱。

Input

* 第1行: 2个用空格隔开的整数：N 和 M

* 第2..N+1行: 第i+1行包含2个用空格隔开的整数：A_i、B_i * 第N+2..N+M+1行: 第j+N+1行包含2个用空格隔开的整数：C_i、D_i

Output

* 第1行: 输出1个整数，表示使所有奶牛满意的最小花费。如果无论如何都无法 满足所有奶牛的需求，输出-1

Sample Input

4 7
1 1
2 3
1 4
4 2
3 2
2 1
4 3
5 2
5 4
2 6
4 4

Sample Output

12

输出说明:

给奶牛1吃价钱为2的2号牧草，奶牛2吃价钱为4的3号牧草，奶牛3分到价钱
为2的6号牧草，奶牛4选择价钱为4的7号牧草，这种分配方案的总花费是12，为
所有方案中花费最少的。

Source Gold

题解：

　　让牛和草按照鲜嫩度排序，然后对于第i头奶牛，把所有新鲜度大于它要求的价值塞到一个平衡树里，每次ANS加上当前平衡树中它要求的价值的后继，但一定要先判断一下有没有和它要求的价值正好相等的草。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
typedef long long LL;
const LL maxn=200010;
LL root,tot,N,M;
LL ANS;
LL key[maxn],siz[maxn],lc[maxn],rc[maxn];
struct COW{
    LL a,b;
}cow[maxn];
struct G{
    LL a,b;
}gra[maxn];
bool cmp(const COW&w,const COW &e){
    if(w.b>e.b) return 1;
    return 0;
}
bool cmp2(const G&w,const G &e){
    if(w.b>e.b) return 1;
    return 0;
}
void r_rotate(LL &rt){
    LL k=lc[rt];
    lc[rt]=rc[k];
    rc[k]=rt;
    siz[k]=siz[rt];
    siz[rt]=siz[lc[rt]]+siz[rc[rt]]+1;
    rt=k;
}  
void l_rotate(LL &rt){
    LL k=rc[rt];
    rc[rt]=lc[k];
    lc[k]=rt;
    siz[k]=siz[rt];
    siz[rt]=siz[lc[rt]]+siz[rc[rt]]+1;
    rt=k;
}
void MAINTAIN(LL &rt,bool flag){
    if(flag==false){
        if(siz[lc[lc[rt]]]>siz[rc[rt]]) r_rotate(rt);
        else if(siz[rc[lc[rt]]]>siz[rc[rt]]){
            l_rotate(lc[rt]);
            r_rotate(rt);
        }
        else return;
    }
    else{
        if(siz[rc[rc[rt]]]>siz[lc[rt]]) l_rotate(rt);
        else if(siz[lc[rc[rt]]]>siz[lc[rt]]){
            r_rotate(rc[rt]);
            l_rotate(rt);
        }
        else return;
    }
    MAINTAIN(lc[rt],0); MAINTAIN(rc[rt],1); 
    MAINTAIN(rt,1); MAINTAIN(rt,0);
}
void insert(LL &rt,LL v){
    if(rt==0){ 
        rt=++tot; 
        key[rt]=v;
        lc[rt]=rc[rt]=0; siz[rt]=1;
        return ;
    }
    siz[rt]++;
    if(v<=key[rt]) insert(lc[rt],v);
    else insert(rc[rt],v);
    MAINTAIN(rt,v>key[rt]);
}
LL Delete(LL &rt,LL v){
    LL ans;
    siz[rt]--;
    if(v==key[rt]||(v<key[rt]&&lc[rt]==0)||(v>key[rt]&&rc[rt]==0)){
        ans=key[rt];
        if(lc[rt]==0||rc[rt]==0) rt=lc[rt]+rc[rt];
        else key[rt]=Delete(lc[rt],key[rt]+1);
        return ans;
    }
    if(v<key[rt]) ans=Delete(lc[rt],v);
    else ans=Delete(rc[rt],v);
    return ans;
}
LL succ(LL &rt,LL v){//返回比v大的最小的数 
    if(rt==0) return v;
    if(v>=key[rt]) return succ(rc[rt],v);
    else{
        LL ans=succ(lc[rt],v);
        if(ans==v) return key[rt];
        return ans;
    }
}
bool find(LL &rt,LL v){//查找是否有 key值为 v的节点 
    if(rt==0) return false;
    else if(v==key[rt]) return true;
    else if(v<key[rt]) return find(lc[rt],v);
    else if(v>key[rt]) return find(rc[rt],v);
}

int main(){
    scanf("%lld%lld",&N,&M);
    if(M<N){
        printf("-1");
        return 0;
    }
    for(LL i=1;i<=N;i++) scanf("%lld%lld",&cow[i].a,&cow[i].b);
    for(LL i=1;i<=M;i++) scanf("%lld%lld",&gra[i].a,&gra[i].b);
    sort(cow+1,cow+N+1,cmp); sort(gra+1,gra+M+1,cmp2);
    for(LL i=1,j=1;i<=N;i++){
        while(gra[j].b>=cow[i].b&&j<=M)
            insert(root,gra[j++].a);
        if(siz[root]==0){printf("-1"); return 0;}
        if(find(root,cow[i].a)==true){
            ANS+=cow[i].a;
            Delete(root,cow[i].a);
        }
        else{
            LL num=succ(root,cow[i].a);
            ANS+=num; 
            Delete(root,num);
        }
    }
    printf("%lld",ANS);
    return 0;
}