hdu--1540 Tunnel Warfare（线段树+区间合并）

  1 #include <iostream>
  2 #include<cstdio>
  3 #include <cstring>
  4 #include <cmath>
  5 using namespace std;
  6 const int maxn=50000+10;
  7 int num[maxn];
  8 struct note
  9 {
 10     int l,r,lm,rm,m,lazy;
 11 }a[maxn<<2];
 12 int max(int a,int b)
 13 {
 14     if(a>b)
 15         return a;
 16     else
 17         return b;
 18 }
 19 int build(int l,int r,int k)
 20 {
 21     a[k].l=l,a[k].r=r,a[k].lm=a[k].rm=a[k].m=(r-l+1);
 22     a[k].lazy=1;
 23     if(l==r)
 24         return 0;
 25         int mid=(l+r)/2;
 26     build(l,mid,k*2);
 27     build(mid+1,r,k*2+1);
 28     return 0;
 29 }
 30 int pushup(int k)
 31 {
 32     if(a[k*2].rm<(a[k*2].r-a[k*2].l+1))
 33         a[k].lm=a[k*2].lm;
 34     else
 35         a[k].lm=a[k*2].rm+a[k*2+1].lm;
 36     if(a[k*2+1].lm<a[k*2+1].r-a[k*2+1].l+1)
 37         a[k].rm=a[k*2+1].rm;
 38     else
 39         a[k].rm=a[k*2+1].lm+a[k*2].rm;
 40     a[k].m=max(max(a[k].lm,a[k].rm),a[k*2+1].lm+a[k*2].rm);
 41     if(a[k].m==a[k].r-a[k].l+1)
 42         a[k].lazy=1;
 43     else
 44         a[k].lazy=0;
 45     return 0;
 46 }
 47 int ins(int d,int n,int k)
 48 {
 49     if(d==a[k].l&&a[k].r==a[k].l)
 50     {
 51         a[k].rm=a[k].lm=a[k].m=n;
 52         a[k].lazy=n;
 53         return 0;
 54     }
 55     //pushdown(k)
 56     if(d<=a[k*2].r)
 57         ins(d,n,k*2);
 58     if(d>a[k*2].r)
 59         ins(d,n,k*2+1);
 60     pushup(k);
 61     return 0;
 62 }
 63 int sea(int n,int k)
 64 {
 65     if(a[k].lazy==1||a[k].r==a[k].l)
 66     {
 67         return a[k].m;
 68     }
 69     if(n<=a[k*2].r)
 70     {
 71         if(n>=a[k*2].r-a[k*2].rm+1)
 72             return sea(n,k*2)+sea(a[k*2+1].l,k*2+1);
 73         else
 74             return sea(n,k*2);
 75     }
 76     else
 77     {
 78         if(n<=a[k*2+1].lm+a[k*2+1].l-1)
 79             return sea(n,k*2+1)+sea(a[k*2].r,k*2);
 80         else
 81             return sea(n,k*2+1);
 82     }
 83     return 0;
 84 }
 85 int main()
 86 {
 87     int n,m,k,b;
 88     while(~scanf("%d%d",&n,&m))
 89     {
 90         build(1,n,1);
 91         k=0;
 92         while(m--)
 93         {
 94             char s[5];
 95             scanf("%s",s);
 96             if(s[0]==‘D‘)
 97             {
 98                 scanf("%d",&num[k]);
 99                 ins(num[k],0,1);
100                 k++;
101             }
102             else if(s[0]==‘Q‘)
103             {
104                 scanf("%d",&b);
105                 printf("%d\n",sea(b,1));
106             }
107             else
108             {
109                 k--;
110                 ins(num[k],1,1);
111             }
112         }
113     }
114 
115     return 0;
116 }