hdu 1540 线段树(区间合并入门)

点击打开链接

#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
typedef long long ll;
const int N=5e4+20;
int s
;
struct node{
int l,r;
int ls,rs,ms;//左端,右端,最大连续,ms区间最大连续
}a[N<<2];
void build(int l,int r,int rt)
{
a[rt].l=l;
a[rt].r=r;
a[rt].ls=a[rt].rs=a[rt].ms=r-l+1;//
if(l!=r)
{
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
}
}
void pushup(int rt)
{
a[rt].ls=a[rt<<1].ls;//左区间等于从左子树左端开始
a[rt].rs=a[rt<<1|1].rs;
a[rt].ms=max(max(a[rt<<1].ms,a[rt<<1|1].ms),a[rt<<1].rs+a[rt<<1|1].ls);//左子树右端连续~右子树左端起连续

if(a[rt<<1].ls == a[rt<<1].r-a[rt<<1].l+1)//左子树区间满了的话，父亲左区间要加上右孩子的左区间
a[rt].ls += a[rt<<1|1].ls;
if(a[rt<<1|1].rs == a[rt<<1|1].r-a[rt<<1|1].l+1)//同理
a[rt].rs += a[rt<<1].rs;
}
void update(int rt,int p,int op)
{
if(a[rt].l==a[rt].r)
{
if(op==1)
a[rt].ls=a[rt].rs=a[rt].ms=1;//
else
a[rt].ls=a[rt].rs=a[rt].ms=0;//

return;
}
int mid=(a[rt].l+a[rt].r)>>1;
if(p<=mid)//更新p所在的一段
update(rt<<1,p,op);
else
update(rt<<1|1,p,op);

pushup(rt);
}
int query(int rt,int p)
{
if(a[rt].l==a[rt].r||a[rt].ms==0||a[rt].ms==a[rt].r-a[rt].l+1)
{
return a[rt].ms;
}
int mid=(a[rt].l+a[rt].r)>>1;
if(p<=mid)//查询p所在的哪一段
{
if(p>=a[rt<<1].r-a[rt<<1].rs+1)//p在左子树右区间的内,和右子树可能有相接
{
return query(rt<<1,p)+query(rt<<1|1,mid+1);
}
else
{
return query(rt<<1,p);
}
}
else
{
if(p<=a[rt<<1|1].l+a[rt<<1|1].ls-1)//同理
{
return query(rt<<1|1,p)+query(rt<<1,mid);
}
else
{
return query(rt<<1|1,p);
}

}
}
int main()
{
int n,m;
while(cin>>n>>m)
{
build(1,n,1);
int x;
int top=0;
while(m--)
{

char op[2];
scanf("%s",op);
if(op[0]=='D')
{
scanf("%d",&x);
s[top++]=x;
update(1,x,0);//0:del
}
else if(op[0]=='Q')
{
scanf("%d",&x);
printf("%d\n",query(1,x));
}
else
{
if(x>0)
{
x=s[--top];
update(1,x,1);//1:restore
}
}
}
}
return 0;
}