洛谷P2161 [SHOI2009]会场预约【Treap】

时空限制 1000ms / 128MB

题目描述

PP大厦有一间空的礼堂，可以为企业或者单位提供会议场地。这些会议中的大多数都需要连续几天的时间（个别的可能只需要一天），不过场地只有一个，所以不同的会议的时间申请不能够冲突。也就是说，前一个会议的结束日期必须在后一个会议的开始日期之前。所以，如果要接受一个新的场地预约申请，就必须拒绝掉与这个申请相冲突的预约。一般来说，如果PP大厦方面事先已经接受了一个会场预约，例如从10日到15日，就不会在接受与之相冲突的预约，例如从12日到17日。不过，有时出于经济利益，PP大厦方面有时会为了接受一个新的会场预约，而拒绝掉一个甚至几个之前预订的预约。于是，礼堂管理员QQ的笔记本上笔记本上经常记录着这样的信息：本题中为方便起见，所有的日期都用一个整数表示。例如，如果一个为期10天的会议从“90日”开始到“99日”，那么下一个会议最早只能在“100日”开始。最近，这个业务的工作量与日俱增，礼堂的管理员QQ希望参加SHTSC的你替他设计一套计算机系统，方便他的工作。这个系统应当能执行下面两个操作： A操作：有一个新的预约是从“start日”到“end日”，并且拒绝掉所有与它相冲突的预约。执行这个操作的时候，你的系统应当返回为了这个新预约而拒绝掉的预约个数，以方便QQ与自己的记录相校对。 B操作：请你的系统返回当前的仍然有效的预约的总数。

输入格式：

输入文件的第一行是一个整数n，表示你的系统将接受的操作总数。接下去n行每行表示一个操作。每一行的格式为下面两者之一： “A start end”表示一个A操作； “B”表示一个B操作。

输出格式：

输出文件有n行，每行一次对应一个输入。表示你的系统对于该操作的返回值

说明

N< = 200000
1< = Start End < = 100000

题目分析

每个预约以结束日期为第一关键字，开始日期为第二关键字升序维护
每个A操作(strat,end)，找到当前预约中结束日期大于等于start且最小的预约(若有多个结束日期相同的预约就先找其中开始日期小的)
若这个预约的开始日期小于end，则冲突，累计答案并删除这个预约
反复这个操作直到没有结束日期大于等于start的预约为止，再插入新的预约
B操作直接输出treap的大小即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
typedef long long lt;
#define pir pair<int,int>
#define mkp(x,y) make_pair(x,y)int read()
{int x=0,f=1;char ss=getchar();while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}return x*f;
}const int inf=1e9;
const int maxn=200010;
int n,m;
int rt,ch[maxn][2],rnd[maxn],cnt;
int size[maxn],st[maxn],ed[maxn];void update(int p)
{int lc=ch[p][0],rc=ch[p][1];size[p]=size[lc]+size[rc]+1;
}void rotate(int &p,int d)
{int k=ch[p][d^1];ch[p][d^1]=ch[k][d];ch[k][d]=p;update(p); update(k);p=k;
}void ins(int &p,int x,int y)
{if(!p){p=++cnt; st[p]=x; ed[p]=y;size[p]=1; rnd[p]=rand(); return;}int d=0;if(y==ed[p]) d=x<st[p]?0:1;else d=y<ed[p]?0:1;ins(ch[p][d],x,y);if(rnd[ch[p][d]]<rnd[p]) rotate(p,d^1);update(p);
}int qnxt(int p,int x)
{if(!p) return 0;int d=x<ed[p]?0:1;if(ed[p]==x){int tt=qnxt(ch[p][0],x);if(ed[tt]!=ed[p]) return p;else return st[tt]<ed[p]?tt:p;}else if(d==1) return qnxt(ch[p][1],x);else{int tt=qnxt(ch[p][0],x);if(ed[p]==ed[tt]) return st[p]<st[tt]?p:tt;return ed[p]<ed[tt]?p:tt;}
}void del(int &p,int x,int y)
{if(!p) return;if(st[p]==x&&ed[p]==y){if(!ch[p][0]) p=ch[p][1];else if(!ch[p][1]) p=ch[p][0];else{int d=rnd[ch[p][0]]<rnd[ch[p][1]]?1:0;rotate(p,d); del(ch[p][d],x,y);}}else if(y==ed[p]){int d=x<st[p]?0:1;del(ch[p][d],x,y);}else{int d=y<ed[p]?0:1;del(ch[p][d],x,y);}if(p) update(p);
}int main()
{n=read(); st[0]=ed[0]=inf;while(n--){char ss[5];scanf("%s",&ss);if(ss[0]=='A'){int x=read(),y=read(),ans=0;while(1){int tt=qnxt(rt,x);if(st[tt]<=y) del(rt,st[tt],ed[tt]),ans++;else break;}ins(rt,x,y);printf("%d\n",ans);}else if(ss[0]=='B') printf("%d\n",size[rt]);}return 0;
}

顺便还写了个set版，虽然是慢了一点，但明显好码一百倍=_=

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
#include<cmath>
using namespace std;
typedef long long lt;
#define IT set<node>::iterator int read()
{int x=0,f=1;char ss=getchar();while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}return x*f;
}const int inf=1e9;
const int maxn=200010;
int n,m;
struct node
{int L,R;node(int ll,int rr): L(ll),R(rr) {}bool operator < (const node& tt)const { return R==tt.R?L<tt.L:R<tt.R;}
};
set<node> st;int main()
{n=read();while(n--){char ss[5];scanf("%s",&ss);if(ss[0]=='A'){int x=read(),y=read(),ans=0;while(!st.empty()){IT it=st.lower_bound(node(-1,x));if(it==st.end()) break;if(it->L<=y) st.erase(it),ans++;else break;}st.insert(node(x,y));printf("%d\n",ans);}else if(ss[0]=='B') printf("%d\n",st.size());}return 0;
}