HDU 1540 Tunnel Warfare(区间合并)【线段树】

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 using namespace std;
 5 
 6 #define Lson rt<<1,l,mid
 7 #define Rson rt<<1|1,mid+1,r
 8 const int N=50000+4;
 9 int len[N<<2],llen[N<<2],rlen[N<<2];
10 int n,m;
11 
12 void Pushup(int rt,int length){    //更新维护每个节点的三个值
13     llen[rt]=llen[rt<<1];   //首先当前节点的最长连续前缀==左子节点的最长连续前缀     
14     rlen[rt]=rlen[rt<<1|1];
15     /*-- 进一步更新 --*/   
16     if(llen[rt<<1]==(length-(length>>1)))llen[rt]+=llen[rt<<1|1]; //如果左子节点的最长连续前缀为整个左子区间，那么本节点的前缀还要加上右子区间的最长前缀
17     if(rlen[rt<<1|1]==(length>>1))rlen[rt]+=rlen[rt<<1];          //同理
18     /*-- 最终更新本节点的最长连续长度  --*/
19     len[rt]=max(max(len[rt<<1],len[rt<<1|1]),rlen[rt<<1]+llen[rt<<1|1]); 
20 }
21 
22 void build(int rt,int l,int r){
23     if(l==r){
24         len[rt]=llen[rt]=rlen[rt]=1;
25         return;
26     }
27     int mid=(l+r)>>1;
28     build(Lson);
29     build(Rson);
30     Pushup(rt,r-l+1);
31 }
32 
33 void update(int rt,int l,int r,int loc,int c){   //单点更新
34     if(l==r){
35         len[rt]=llen[rt]=rlen[rt]=c; 
36         return;
37     }
38     int mid=(l+r)>>1;
39     if(loc<=mid)update(Lson,loc,c);
40     if(loc>mid)update(Rson,loc,c);
41     Pushup(rt,r-l+1);
42 }
43 
44 int query(int rt,int l,int r,int loc){
45     if(l==r)return len[rt];
46     int mid=(l+r)>>1;
47     if(loc<=mid){
48         if(loc+rlen[rt<<1]>mid)return rlen[rt<<1]+llen[rt<<1|1]; //如果loc在左子区间的最长后缀和右子区间的最长前缀中，直接输出这两个前后缀之和即可
49         else return query(Lson,loc);  //否则的话，继续向左子节点查询
50     }
51     else{
52         if(mid+llen[rt<<1|1]>=loc)return rlen[rt<<1]+llen[rt<<1|1];//如果loc在左子区间的最长后缀和右子区间的最长前缀中，直接输出这两个前后缀之和即可
53         else return query(Rson,loc);
54     }
55 }
56 
57 int main(){
58     while(scanf("%d %d",&n,&m)!=EOF){
59         build(1,1,n);
60         int tot=0,stk[N+10]; //stk模拟栈
61         while(m--){
62             char str[10];
63             scanf("%s",str);
64             if(str[0]==‘D‘){
65                 int cal;scanf("%d",&cal);
66                 stk[++tot]=cal;
67                 update(1,1,n,cal,0);
68             }
69             else if(str[0]==‘R‘){
70                 int cal=stk[tot--];
71                 update(1,1,n,cal,1);
72             }
73             else{
74                 int cal;scanf("%d",&cal);
75                 printf("%d\n",query(1,1,n,cal));
76             }
77         }
78     }
79     return 0;
80 }