洛谷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;
}