pku 3667 hotel

#include<stdio.h>
#include<iostream>
using namespace std;
const int OTHER = 0;
const int FULL = 1;
const int EMPTY = -1;
const int QUERRY = 1;
const int LEAVE = 2;
const int MAX_N = 50050;
const int MAX_T = MAX_N*2;
const int root = 1;
struct node
{
int L,R;
int Lid,Rid;
int value;
int Lvalue,Rvalue;
int tag;
};
struct node T[MAX_T];
int N,M;
int times;
struct node Tmaking(int a,int b,int c,int d,int e,int f,int g,int h)
{
struct node tmp ;
tmp.L=a;
tmp.R=b;
tmp.Lid=c;
tmp.Rid=d;
tmp.value=e;
tmp.Lvalue=f;
tmp.Rvalue=g;
tmp.tag=h;
return tmp;
}
void Build(int l,int r)
{
if (l==r) {T[++times]=Tmaking(l,r,-1,-1,1,1,1,OTHER);return ;}
int mid=(l+r)/2;
int now=++times;
int lid=times+1;
Build(l,mid);
int rid=times+1;
Build(mid+1,r);
T[now]=Tmaking(l,r,lid,rid,r-l+1,r-l+1,r-l+1,OTHER);
}
void init()
{
scanf("%d %d",&N,&M);
Build(1,N);
}
void lazy_tag(int t)
{
if (T[t].tag!=OTHER)
{
int l=T[t].L;
int r=T[t].R;
int lid=T[t].Lid;
int rid=T[t].Rid;
int mid=(l+r)/2;
T[lid].tag=T[rid].tag=T[t].tag;
T[lid].value=T[lid].Lvalue=T[lid].Rvalue=(T[t].tag==FULL?0:mid-l+1);
T[rid].value=T[rid].Lvalue=T[rid].Rvalue=(T[t].tag==FULL?0:r-mid);
T[t].tag=OTHER;
}
}
void change(int t,int s,int e,int tagf)
{
int l=T[t].L;
int r=T[t].R;
int lid=T[t].Lid;
int rid=T[t].Rid;
int mid=(l+r)/2;
int atag= (tagf==FULL?0:r-l+1);
if (s<=l&&e>=r) {T[t]=Tmaking(l,r,lid,rid,atag,atag,atag,tagf);return ;}
lazy_tag(t);
if (e<=mid) change(lid,s,e,tagf);
else if (s>mid) change(rid,s,e,tagf);
else change(lid,s,mid,tagf),change(rid,mid+1,e,tagf);
int value = max(T[lid].value,T[rid].value);
value=max(value,T[lid].Rvalue+T[rid].Lvalue);
int lvalue = T[lid].Lvalue;
if (lvalue==mid-l+1) lvalue+=T[rid].Lvalue;
int rvalue = T[rid].Rvalue;
if (rvalue==r-mid) rvalue+=T[lid].Rvalue;
T[t]=Tmaking(l,r,lid,rid,value,lvalue,rvalue,OTHER);
}
int Querry(int t,int room)
{
if (T[t].value<room) return 0;
int l=T[t].L;
int r=T[t].R;
int lid=T[t].Lid;
int rid=T[t].Rid;
int mid=(l+r)/2;
lazy_tag(t);
int pos;
pos=Querry(lid,room);
if (pos) return pos;
if (T[lid].Rvalue+T[rid].Lvalue>=room) pos=mid-T[lid].Rvalue+1;
if (pos) return pos;
pos=Querry(rid,room);
if (pos) return pos;
return 0;
}
void work_put()
{
int op,a,b,i;
int pos;
for (i=1;i<=M;i++)
{
scanf("%d",&op);
if (op==QUERRY)
{
scanf("%d",&a);
pos=Querry(root,a);
printf("%d\n",pos);
if (pos) change(root,pos,pos+a-1,FULL);
}
if (op==LEAVE)
{
scanf("%d %d",&a,&b);
change(root,a,a+b-1,EMPTY);
}
}
}
int main()
{
init();
work_put();
return 0;
}