【POJ3667】Hotel（线段树）

题意：有n个依次编号的元素，要求维护以下两个操作：

1.询问整个数列中是否有长度>=x的连续的一段未被标记的元素，若无输出0，若有输出最小的开始编号ans并将[ans,ans+x-1]标记

2.将[x,x+y-1]其中的元素取消标记（如果有）

n,m<=5e4

思路：线段树区间合并

记录从左、右边开始最长连续长度，一段最长连续长度，以及标记状态

Pascal跑的比C++快……

 #include<cstdio>
 #include<cstring>
 #include<string>
 #include<cmath>
 #include<iostream>
 #include<algorithm>
 #include<map>
 #include<set>
 #include<queue>
 #include<vector>
 using namespace std;
 typedef long long ll;
 typedef unsigned int uint;
 typedef unsigned long long ull;
 typedef pair<int,int> PII;
 typedef vector<int> VI;
 #define fi first
 #define se second
 #define MP make_pair
 #define N   210000
 #define M   7010
 #define eps 1e-8
 #define pi  acos(-1)
 #define oo  1e9
 #define MOD 10007

 struct node
 {
     int l,r,a,s;
 }t[N];

 void build(int l,int r,int p)
 {
     t[p].l=t[p].r=t[p].s=r-l+;
     t[p].a=;
     if(l==r) return;
     int mid=(l+r)>>;
     build(l,mid,p<<);
     build(mid+,r,p<<|);
 }

 void pushdown(int l,int r,int p)
 {
     int ls=p<<; int rs=ls+;
     int mid=(l+r)>>;
     if(t[p].a==)
     {
         t[ls].l=t[ls].r=t[ls].s=t[rs].l=t[rs].r=t[rs].s=;
         t[ls].a=t[rs].a=;
     }
     if(t[p].a==)
     {
         t[ls].l=t[ls].r=t[ls].s=mid-l+;
         t[rs].l=t[rs].r=t[rs].s=r-mid;
         t[ls].a=t[rs].a=;
     }
 }

 void pushup(int l,int r,int p)
 {
     int ls=p<<; int rs=ls+;
     if(t[ls].a==) t[p].l=t[ls].l+t[rs].l;
      else t[p].l=t[ls].l;
     if(t[rs].a==) t[p].r=t[rs].r+t[ls].r;
      else t[p].r=t[rs].r;
     t[p].s=max(max(t[ls].s,t[rs].s),t[ls].r+t[rs].l);
     if(t[ls].a==&&t[rs].a==) t[p].a=;
      else if(t[ls].a==&&t[rs].a==) t[p].a=;
       else t[p].a=;
 }

 int query(int l,int r,int x,int p)
 {
     if(l==r) return l;
     pushdown(l,r,p);
     int ls=p<<;
     int rs=ls+;
     int mid=(l+r)>>;
     if(t[ls].s>=x) return query(l,mid,x,ls);
      else if(t[ls].r+t[rs].l>=x) return mid-t[ls].r+;
       else return query(mid+,r,x,rs);
 }

 void update(int l,int r,int x,int y,int v,int p)
 {
     if(x<=l&&r<=y)
     {
         t[p].a=v;
         if(v==)
         {
             t[p].l=t[p].r=t[p].s=;
         }
         if(v==)
         {
             t[p].l=t[p].r=t[p].s=r-l+;
         }
         return;
     }
     if(t[p].a==v) return;
     pushdown(l,r,p);
     int mid=(l+r)>>;
     int ls=p<<;
     int rs=ls+;
     if(x<=mid) update(l,mid,x,y,v,ls);
     if(y>mid) update(mid+,r,x,y,v,rs);
     pushup(l,r,p);
 }

 int main()
 {

     int n,m;
     scanf("%d%d",&n,&m);
     build(,n,);
     for(int i=;i<=m;i++)
     {
         int x;
         scanf("%d",&x);
         if(x==)
         {
             int y;
             scanf("%d",&y);
             if(t[].s<y) printf("0\n");
              else
              {
                  int k=query(,n,y,);
                 printf("%d\n",k);
                 update(,n,k,k+y-,,);
              }
         }
          else
          {
              int y,z;
              scanf("%d%d",&y,&z);
              update(,n,y,y+z-,,);
          }
     }
     return ;
 }

 var t:array[..]of record
                           l,r,s:longint;
                          end;
     tag:array[..]of longint;
     n,m,i,x,y,z,k:longint;

 procedure build(l,r,p:longint);
 var mid:longint;
 begin
  t[p].s:=r-l+;
  tag[p]:=;
  if l=r then exit;

  mid:=(l+r)>>;
  build(l,mid,p<<);
  build(mid+,r,p<<+);
 end;

 function max(x,y:longint):longint;
 begin
  if x>y then exit(x);
  exit(y);
 end;

 procedure pushdown(l,r,p:longint);
 var ls,rs,mid,len:longint;
 begin
  ls:=p<<; rs:=ls+;
  mid:=(l+r)>>;
  if tag[p]= then
  begin
   t[ls].l:=; t[ls].r:=; t[ls].s:=; tag[ls]:=;
   t[rs].l:=; t[rs].r:=; t[rs].s:=; tag[rs]:=;
  end;
  if tag[p]= then
  begin
   len:=mid-l+;
   t[ls].l:=len; t[ls].r:=len; t[ls].s:=len; tag[ls]:=;
   len:=r-(mid+)+;
   t[rs].l:=len; t[rs].r:=len; t[rs].s:=len; tag[rs]:=;
  end;
 end;

 procedure pushup(l,r,p:longint);
 var ls,rs:longint;
 begin
  ls:=p<<; rs:=ls+;
  if tag[ls]= then t[p].l:=t[ls].l+t[rs].l
   else t[p].l:=t[ls].l;
  if tag[rs]= then t[p].r:=t[rs].r+t[ls].r
   else t[p].r:=t[rs].r;
  t[p].s:=max(t[ls].s,t[rs].s);
  t[p].s:=max(t[p].s,t[ls].r+t[rs].l);
  if (tag[ls]=)and(tag[rs]=) then tag[p]:=
   else if (tag[ls]=)and(tag[rs]=) then tag[p]:=
    else tag[p]:=;
 end;

 function query(l,r,v,p:longint):longint;
 var mid,ls,rs:longint;
 begin
  if l=r then exit(l);
  pushdown(l,r,p);
  ls:=p<<; rs:=ls+;
  mid:=(l+r)>>;
  if t[ls].s>=v then query:=query(l,mid,v,p<<)
   else if t[ls].r+t[rs].l>=v then query:=mid-t[ls].r+
    else query:=query(mid+,r,v,p<<+);
 end;

 procedure update(l,r,x,y,v,p:longint);
 var mid,len:longint;
 begin
  if (l=x)and(r=y) then
  begin
   tag[p]:=v;
   if v= then
   begin
    t[p].l:=; t[p].r:=; t[p].s:=;
   end;
   if v= then
   begin
    len:=y-x+;
    t[p].l:=len; t[p].r:=len; t[p].s:=len;
   end;
   exit;
  end;
  if tag[p]=v then exit;
  pushdown(l,r,p);
  mid:=(l+r)>>;
  if (x>=l)and(y<=mid) then update(l,mid,x,y,v,p<<)
   else if (x>mid)and(y<=r) then update(mid+,r,x,y,v,p<<+)
    else
    begin
     update(l,mid,x,mid,v,p<<);
     update(mid+,r,mid+,y,v,p<<+);
    end;
  pushup(l,r,p);
 end;

 begin

  readln(n,m);
  build(,n,);
  for i:= to m do
  begin
   read(x);
   if x= then
   begin
    read(y);
    if t[].s<y then writeln()
     else
     begin
      k:=query(,n,y,);
      writeln(k);
      update(,n,k,k+y-,,);
     end;
   end;
   if x= then
   begin
    read(y,z);
    update(,n,y,y+z-,,);
   end;
  end;

 end.