HDU 1540 Tunnel Warfare 线段树区间合并

Posted 2020-08-02 blues

Tunnel Warfare

题意：D代表破坏村庄，R代表修复最后被破坏的那个村庄，Q代表询问包括x在内的最大连续区间是多少

思路：一个节点的最大连续区间由（左儿子的最大的连续区间，右儿子的最大连续区间，左儿子的最大连续右区间+右儿子的最大连续左区间）决定

所以线段树的节点应该维护当前节点的最大连续左区间，最大连续右区间，和最大连续区间。

注意更新的时候，如果左儿子全满，父亲节点的左连续区间还要加上右儿子的左区间。反之同理。

查询的时候，可以剪枝，如果是叶子，或为空，或满，则不用往下查询。

查询的时候还要注意，当前查询点在左儿子的最大右连续区间内时，最大连续区间还要加上右儿子的最大连续区间。反之同理；

  1 // #pragma comment(linker, "/STACK:1024000000,1024000000")
  2 #include <iostream>
  3 #include <cstdio>
  4 #include <cstring>
  5 #include <sstream>
  6 #include <string>
  7 #include <algorithm>
  8 #include <list>
  9 #include <map>
 10 #include <vector>
 11 #include <queue>
 12 #include <stack>
 13 #include <cmath>
 14 #include <cstdlib>
 15 // #include <conio.h>
 16 using namespace std;
 17 #define clc(a,b) memset(a,b,sizeof(a))
 18 #define inf 0x3f3f3f3f
 19 #define lson l,mid,rt<<1
 20 #define rson mid+1,r,rt<<1|1
 21 const int N = 50010;
 22 const int MOD = 1e9+7;
 23 #define LL long long
 24 #define mi() (l+r)>>1
 25 double const pi = acos(-1);
 26 void fre() {
 27     freopen("in.txt","r",stdin);
 28 }
 29 // inline int r() {
 30 //     int x=0,f=1;char ch=getchar();
 31 //     while(ch>‘9‘||ch<‘0‘) {if(ch==‘-‘) f=-1;ch=getchar();}
 32 //     while(ch>=‘0‘&&ch<=‘9‘) { x=x*10+ch-‘0‘;ch=getchar();}return x*f;
 33 // }
 34 int s[N];
 35 struct node {
 36     int l,r;
 37     int ls,rs,ms;
 38 } e[N<<2];
 39 
 40 void pushup(int rt) {
 41 
 42 }
 43 void build(int l,int r,int rt) {
 44     e[rt].l=l;
 45     e[rt].r=r;
 46     e[rt].ls=e[rt].rs=e[rt].ms=r-l+1;
 47     if(l!=r) {
 48         int mid=mi();
 49         build(lson);
 50         build(rson);
 51     }
 52 }
 53 
 54 void update(int rt,int c,int x) {
 55     if(e[rt].l==e[rt].r) {
 56         if(x>0) {
 57             e[rt].ls=e[rt].rs=e[rt].ms=1;
 58         } else
 59             e[rt].ls=e[rt].rs=e[rt].ms=0;
 60         return;
 61     }
 62     int mid=(e[rt].l+e[rt].r)>>1;
 63     if(c<=mid) {
 64         update(rt<<1,c,x);
 65     } else
 66         update(rt<<1|1,c,x);
 67     e[rt].ls=e[rt<<1].ls;
 68     e[rt].rs=e[rt<<1|1].rs;
 69     e[rt].ms=max(max(e[rt<<1].ms,e[rt<<1|1].ms),e[rt<<1].rs+e[rt<<1|1].ls);
 70     if(e[rt<<1].ls==e[rt<<1].r-e[rt<<1].l+1) {
 71         e[rt].ls+=e[rt<<1|1].ls;
 72     }
 73     if(e[rt<<1|1].rs==e[rt<<1|1].r-e[rt<<1|1].l+1) {
 74         e[rt].rs+=e[rt<<1].rs;
 75     }
 76 }
 77 
 78 int query(int rt,int c) {
 79     if(e[rt].l==e[rt].r||e[rt].ms==0||e[rt].ms==e[rt].r-e[rt].l+1) {
 80         return e[rt].ms;
 81     }
 82     int mid=(e[rt].l+e[rt].r)>>1;
 83     if(c<=mid) {
 84         if(c>=e[rt<<1].r-e[rt<<1].rs+1) {
 85             return query(rt<<1,c)+query(rt<<1|1,mid+1);
 86         } else
 87             return query(rt<<1,c);
 88     } else {
 89         if(c<=e[rt<<1|1].l+e[rt<<1|1].ls-1) {
 90             return query(rt<<1|1,c)+query(rt<<1,mid);
 91         } else
 92             return query(rt<<1|1,c);
 93     }
 94 }
 95 int main() {
 96 //    fre();
 97     int n,q,top;
 98     while(~scanf("%d%d",&n,&q)) {
 99         build(1,n,1);
100         top=0;
101         while(q--) {
102             char c;
103             int x;
104             cin>>c;
105             if(c==‘D‘) {
106                 cin>>x;
107                 s[top++]=x;
108                 update(1,x,0);
109             } else if(c==‘Q‘) {
110                 cin>>x;
111                 printf("%d\n",query(1,x));
112             } else {
113                 x=s[--top];
114                 update(1,x,1);
115             }
116         }
117     }
118     return 0;
119 }