[SHOI2009] 会场预约 - Treap

Description

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

A操作：有一个新的预约是从“start日”到“end日”，并且拒绝掉所有与它相冲突的预约。执行这个操作的时候，你的系统应当返回为了这个新预约而拒绝掉的预约个数，以方便QQ与自己的记录相校对。

B操作：请你的系统返回当前的仍然有效的预约的总数。

Input&&Output

Input

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

Output

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

Sample

Input

6
A 10 15
A 17 19
A 12 17
A 90 99
A 11 12
B

Output

0
0
2
0
1
2

Solution

两颗Treap维护开始时间和结束时间，每次插入新节点，从开始时间不断找开始树上的后继，直到大于结束时间，结束时间类似。另外一项操作处理覆盖当前区间的区间。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<vector>
#include<algorithm>
#define maxn 200005
using namespace std;
struct node{
    int v,w;
    int l,r;
    int k,size;
    node()
    {
        l=r=size=k=0;
    }
}t[maxn<<1];
int b[100002],e[100002];
char c;
int l,r,op,ed,cnt,size,ans,n;
int New(int &p,int x)
{
    p=++cnt;
    t[p].v=x;
    t[p].w=rand();
    t[p].size=t[p].k=1;
}
void pushup(int p)
{
    t[p].size=t[t[p].l].size+t[t[p].r].size+t[p].k;
}
void turnl(int &p)
{
    int q=t[p].r;
    t[p].r=t[q].l;t[q].l=p;
    p=q;
    pushup(t[p].l);
    pushup(p);
}
void turnr(int &p)
{
    int q=t[p].l;
    t[p].l=t[q].r;t[q].r=p;
    p=q;
    pushup(t[p].r);
    pushup(p);
}
void Insert(int &p,int x)
{
    if(p==0)
    {
        New(p,x);
        return;
    }
    t[p].size++;
    if(t[p].v==x)
    {
        t[p].k++;
    }
    else if(t[p].v<x)
    {
        Insert(t[p].r,x);
        if(t[t[p].r].w<t[p].w)turnl(p);
    }
    else
    {
        Insert(t[p].l,x);
        if(t[t[p].l].w<t[p].w)turnr(p);
    }
}
void Delete(int &p,int x)
{
    if(p==0)return;
    if(t[p].v==x)
    {
        if(t[p].k>1){
            t[p].k--;
            t[p].size--;
            return;
        }
        if(t[p].l*t[p].r==0)p=t[p].l+t[p].r;
        else if(t[t[p].r].w>t[t[p].l].w)
            turnr(p),Delete(p,x);
        else turnl(p),Delete(p,x);
    }
    else if(x>t[p].v)
    {
        t[p].size--;Delete(t[p].r,x);
    }
    else
    {
        t[p].size--;Delete(t[p].l,x);
    }
}
 int rank(int p,int x)
 {
    if(p==0)return 0;
    if(x==t[p].v)
        return t[t[p].l].size+1;
    else if(x>t[p].v)
        return rank(t[p].r,x)+t[t[p].l].size+t[p].k;
    else return rank(t[p].l,x);
 }
void pre(int p,int x)
 {
    if(p==0)return;
    else if(x>=t[p].v)
    {
        ans=p;
        pre(t[p].r,x);
    }
    else pre(t[p].l,x);
 }
void suf(int p,int x)
 {
    if(p==0)return;
    else if(x<=t[p].v)
    {
        ans=p;
        suf(t[p].l,x);
    }
    else suf(t[p].r,x);
 }
int main()
{
    srand(time(NULL));
    scanf("%d",&n);
    for(int i=1;i<=n;++i)
    {
        cin>>c;
        if(c=='A'){
            scanf("%d%d",&l,&r);
            int cur=r,count=0;
            ans=0;
            while(1)
            {
                ans=0;
                pre(ed,cur);
                if(ans!=0){
                    if(t[ans].v<l) break;
                    cur=t[ans].v;
                    Delete(ed,t[ans].v);
                    Delete(op,e[t[ans].v]);
                    count++;
                }
                else break;
            }
            ans=0;cur=l;
            while(1)
            {
                ans=0;
                suf(op,cur);
                if(ans!=0){
                    if(t[ans].v>r) break;
                    cur=t[ans].v;
                    Delete(op,t[ans].v);
                    Delete(ed,b[t[ans].v]);
                    count++;

                }
                else break;
            }
            ans=0;cur=l;
            while(1)
            {
                ans=0;
                suf(op,cur);
                if(ans!=0){
                    if(t[ans].v>r)break;
                    else {
                        Delete(op,t[ans].v);
                        Delete(ed,b[t[ans].v]);
                        count++;
                    }
                }
                else break;
            }
            ans=0;
            pre(op,l);
            if(ans!=0&&b[t[ans].v]>=r){
                Delete(op,t[ans].v);
                Delete(ed,b[t[ans].v]);
                count++;
            }
            Insert(op,l);
            Insert(ed,r);
            b[l]=r;
            e[r]=l;
            printf("%d\n",count);
        }
        else {
            printf("%d\n",t[ed].size);
        }
    }
    return 0;
}

原文地址：https://www.cnblogs.com/nishikino-curtis/p/8581339.html