bzoj1691 [Usaco2007 Dec]挑剔的美食家(treap)

本文介绍了一种使用Treap数据结构解决特定排序和查询问题的方法。通过按b值排序并利用Treap进行插入和删除操作,可以高效地找到每个元素的后继。适用于需要快速查找后继元素的场景。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

按b排序以后,就是求后继,用treap即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,rt,ls[N],rs[N],v[N],sz[N],cnt[N],owo=0,rnd[N];ll ans=0;
struct data{
    int a,b;
    friend bool operator<(data a,data b){return a.b>b.b;} 
}a[N],b[N];
inline void update(int p){
    sz[p]=sz[ls[p]]+sz[rs[p]]+cnt[p];
}
inline void rturn(int &y){int x=ls[y];ls[y]=rs[x];rs[x]=y;update(y);update(x);y=x;}
inline void lturn(int &y){int x=rs[y];rs[y]=ls[x];ls[x]=y;update(y);update(x);y=x;}
inline void ins(int &p,int x){
    if(!p){p=++owo;sz[p]=cnt[p]=1;v[p]=x;rnd[p]=rand();return;}
    if(v[p]==x){cnt[p]++;sz[p]++;return;}
    if(x<v[p]){ins(ls[p],x);if(rnd[ls[p]]<rnd[p]) rturn(p);}
    else {ins(rs[p],x);if(rnd[rs[p]]<rnd[p]) lturn(p);}
}
inline int getsucc(int x){
    int res=inf,p=rt;
    while(p){
        if(v[p]==x){res=x;break;}
        if(v[p]>x) res=v[p],p=ls[p];
        else p=rs[p];
    }return res;
}
inline void del(int &p,int x){
    if(v[p]==x){
        if(cnt[p]>1) cnt[p]--,sz[p]--;
        else if(ls[p]*rs[p]==0) p=ls[p]+rs[p];
        else if(rnd[ls[p]]<rnd[rs[p]]) rturn(p),del(p,x);
        else lturn(p),del(p,x);return;
    }sz[p]--;
    if(x<v[p]) del(ls[p],x);
    else del(rs[p],x);
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();srand(20000712);
    for(int i=1;i<=n;++i) a[i].a=read(),a[i].b=read();
    for(int i=1;i<=m;++i) b[i].a=read(),b[i].b=read();
    sort(a+1,a+n+1);sort(b+1,b+m+1);int now=1;
    for(int i=1;i<=n;++i){
        while(now<=m&&b[now].b>=a[i].b) ins(rt,b[now].a),++now;
        int succ=getsucc(a[i].a);if(succ==inf){puts("-1");return 0;}
        ans+=succ;del(rt,succ);
    }printf("%lld\n",ans);
    return 0;
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值