hdu 1540 Tunnel Warfare(线段树区间合并)

hdu 1540 Tunnel Warfare

记录每个节点的最大左连续值、最大右连续值、最大连续值，向上更新的是常规的区间合并处理方式

关键是想到如何去查询，如果查询节点落在左儿子节点的右连续段中，则代表右儿子节点的左连续段也要算上，同样的如果查询节点落在右儿子的左连续段中，则代表左儿子的右连续段也要算上

#include<cstdio>
#include<cstring>
#include<stack>
using namespace std;
#define MAXN 50005
#define lch p<<1
#define rch p<<1|1
#define mid (t[p].l+t[p].r)>>1
struct node
{
int l,r;
int lmx,mx,rmx;
int getlen() {return r-l+1;}
}t[MAXN*4];
void construct(int l,int r,int p)
{
t[p].l=l,t[p].r=r;
t[p].lmx=t[p].rmx=t[p].mx=t[p].getlen();
if(l==r) return ;
int m=mid;
construct(l,m,lch);
construct(m+1,r,rch);
}
int max(int a,int b) {return a>b?a:b;}
void pushup(int p)
{
t[p].mx=max(t[lch].mx,t[rch].mx);
t[p].mx=max(t[p].mx,t[lch].rmx+t[rch].lmx);

t[p].lmx=t[lch].lmx;
if(t[lch].mx==t[lch].getlen()) t[p].lmx+=t[rch].lmx;

t[p].rmx=t[rch].rmx;
if(t[rch].mx==t[rch].getlen()) t[p].rmx+=t[lch].rmx;
}
void modify(int x,int p,int v)
{
if(t[p].l==t[p].r)
{
t[p].lmx=t[p].rmx=t[p].mx=v;
return ;
}
int m=mid;
if(x<=m) modify(x,lch,v);
else modify(x,rch,v);
pushup(p);
}
int query(int x,int p)
{
if(t[p].l==t[p].r||
t[p].mx==0||
t[p].mx==t[p].getlen())
return t[p].mx;

int m=mid;
if(x<=m)
{
if(t[lch].r-t[lch].rmx+1<=x)
return query(x,lch)+query(m+1,rch);
else return query(x,lch);
}
else
{
if(t[rch].l+t[rch].lmx-1>=x)
return query(x,rch)+query(m,lch);
else return query(x,rch);
}
}
stack<int> stc;
int main()
{
int n,m,x;
char op[3];
while(scanf("%d%d",&n,&m)!=EOF)
{
construct(1,n,1);
while(!stc.empty()) stc.pop();
while(m--)
{
scanf("%s",op);
if(op[0]=='D')
{
scanf("%d",&x);
modify(x,1,0);
stc.push(x);
}
else if(op[0]=='Q')
{
scanf("%d",&x);
printf("%d\n",query(x,1));
}
else
{
if(!stc.empty())
{
modify(stc.top(),1,1);
stc.pop();
}
}
}
}
return 0;
}