【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+1;
    t[p].a=2;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(l,mid,p<<1);
    build(mid+1,r,p<<1|1);
}

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

void pushup(int l,int r,int p)
{
    int ls=p<<1; int rs=ls+1;
    if(t[ls].a==2) t[p].l=t[ls].l+t[rs].l;
     else t[p].l=t[ls].l;
    if(t[rs].a==2) 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==1&&t[rs].a==1) t[p].a=1;
     else if(t[ls].a==2&&t[rs].a==2) t[p].a=2;
      else t[p].a=3;
}

int query(int l,int r,int x,int p)
{
    if(l==r) return l;
    pushdown(l,r,p);
    int ls=p<<1;
    int rs=ls+1;
    int mid=(l+r)>>1;
    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+1;
      else return query(mid+1,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==1)
        {
            t[p].l=t[p].r=t[p].s=0;
        }
        if(v==2)
        {
            t[p].l=t[p].r=t[p].s=r-l+1;
        }
        return;
    }
    if(t[p].a==v) return;
    pushdown(l,r,p);
    int mid=(l+r)>>1;
    int ls=p<<1;
    int rs=ls+1;
    if(x<=mid) update(l,mid,x,y,v,ls);
    if(y>mid) update(mid+1,r,x,y,v,rs);
    pushup(l,r,p);    
}
    
int main()
{

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

var t:array[1..200000]of record
                          l,r,s:longint;
                         end;
    tag:array[1..200000]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+1;
 tag[p]:=2;
 if l=r then exit;

 mid:=(l+r)>>1;
 build(l,mid,p<<1);
 build(mid+1,r,p<<1+1);
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<<1; rs:=ls+1;
 mid:=(l+r)>>1;
 if tag[p]=1 then
 begin
  t[ls].l:=0; t[ls].r:=0; t[ls].s:=0; tag[ls]:=1;
  t[rs].l:=0; t[rs].r:=0; t[rs].s:=0; tag[rs]:=1;
 end;
 if tag[p]=2 then
 begin
  len:=mid-l+1;
  t[ls].l:=len; t[ls].r:=len; t[ls].s:=len; tag[ls]:=2;
  len:=r-(mid+1)+1;
  t[rs].l:=len; t[rs].r:=len; t[rs].s:=len; tag[rs]:=2;
 end;
end;

procedure pushup(l,r,p:longint);
var ls,rs:longint;
begin
 ls:=p<<1; rs:=ls+1;
 if tag[ls]=2 then t[p].l:=t[ls].l+t[rs].l
  else t[p].l:=t[ls].l;
 if tag[rs]=2 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]=1)and(tag[rs]=1) then tag[p]:=1
  else if (tag[ls]=2)and(tag[rs]=2) then tag[p]:=2
   else tag[p]:=3;
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<<1; rs:=ls+1;
 mid:=(l+r)>>1;
 if t[ls].s>=v then query:=query(l,mid,v,p<<1)
  else if t[ls].r+t[rs].l>=v then query:=mid-t[ls].r+1
   else query:=query(mid+1,r,v,p<<1+1);
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=1 then
  begin
   t[p].l:=0; t[p].r:=0; t[p].s:=0;
  end;
  if v=2 then
  begin
   len:=y-x+1;
   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)>>1;
 if (x>=l)and(y<=mid) then update(l,mid,x,y,v,p<<1)
  else if (x>mid)and(y<=r) then update(mid+1,r,x,y,v,p<<1+1)
   else
   begin
    update(l,mid,x,mid,v,p<<1);
    update(mid+1,r,mid+1,y,v,p<<1+1);
   end;
 pushup(l,r,p);
end;


begin

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

end.