HDU 1540 Tunnel Warfare 【線段樹--最大連續區間】

【題目地址】

  http://acm.hdu.edu.cn/showproblem.php?pid=1540

 

【題目大意】

  題目意思爲有一系列村莊，然後對一些村莊進行破壞或修復。讓你求與某村莊直接或間接相連且都沒有破壞的村莊數量。

 

【解題思路】

/************************************************************************ 
  
此題線段樹最經典的部分是更新部分，也是這題最難的部分，要考慮全面 
 
線段樹結構體定義兩個變量  Lx，Rx  
 
分別表示從左節點開始的長度，從右節點開始的長度 

 

更新代碼是關鍵：
if(tb[rt<<1].Lx==mid-L+1)tb[rt].Lx=tb[rt<<1].Lx+tb[rt<<1|1].Lx;
 else tb[rt].Lx=tb[rt<<1].Lx;

 

if(tb[rt<<1|1].Rx==R-mid)tb[rt].Rx=tb[rt<<1|1].Rx+tb[rt<<1].Rx;
 else tb[rt].Rx=tb[rt<<1|1].Rx;
        
************************************************************************/

 

【代碼】

#include <cstdio> const int N=50005; #define max(a,b) a>b?a:b struct Node { int Lx,Rx; int value; }tb[N<<2]; void Build(int L,int R,int rt) { if(L==R){ tb[rt].Lx=tb[rt].Rx=1; tb[rt].value=0; return; } int mid=(L+R)>>1; Build(L,mid,rt<<1); Build(mid+1,R,rt<<1|1); tb[rt].Lx=tb[rt].Rx=R-L+1; } void update(int x,int y,int L,int R,int rt){ if(R==L){ tb[rt].value=y; if(y)tb[rt].Lx=tb[rt].Rx=0; else tb[rt].Lx=tb[rt].Rx=1; return; } int mid=(L+R)>>1; if(x<=mid)update(x,y,L,mid,rt<<1); else update(x,y,mid+1,R,rt<<1|1); if(tb[rt<<1].Lx==mid-L+1)tb[rt].Lx=tb[rt<<1].Lx+tb[rt<<1|1].Lx; else tb[rt].Lx=tb[rt<<1].Lx; if(tb[rt<<1|1].Rx==R-mid)tb[rt].Rx=tb[rt<<1|1].Rx+tb[rt<<1].Rx; else tb[rt].Rx=tb[rt<<1|1].Rx; } int query(int x,int L,int R,int rt){ if (L==R){ return tb[rt].value==0; } int mid=(R+L)>>1; if (x<=mid){ if(x>mid-tb[rt<<1].Rx)return tb[rt<<1].Rx+tb[rt<<1|1].Lx; else return query(x,L,mid,rt<<1); } else{ if(x<=mid+tb[rt<<1|1].Lx) return tb[rt<<1].Rx+tb[rt<<1|1].Lx; else query(x,mid+1,R,rt<<1|1); } } int main() { int n,m,i,v; int stack[N],top=0; char op; while(scanf("%d%d",&n,&m)!=EOF) { Build(1,n,1); getchar(); for (i=1,top=0;i<=m;i++) { scanf("%c",&op); if (op=='D') { scanf("%d",&v); stack[top++]=v; update(v,1,1,n,1); } else if(op=='R'){ if (top==0)continue; update(stack[--top],0,1,n,1); } else{ scanf("%d",&v); printf("%d/n",query(v,1,n,1)); } getchar(); } } return 0; }