首页 > 代码库 > Hdu1540Tunnel Warfare线段树

Hdu1540Tunnel Warfare线段树

  

#include <cstdio>#include <cstring>#include <algorithm>#include <climits>#include <string>#include <iostream>#include <map>#include <cstdlib>#include <list>#include <set>#include <queue>#include <stack>#include<math.h>using namespace std;#define lson l,mid,rt<<1#define rson mid+1,r,rt<<1|1const int maxn = 55555;int sum[maxn << 2];int lsum[maxn << 2];int rsum[maxn << 2];void up(int rt, int m){    lsum[rt] = lsum[rt << 1];    rsum[rt] = rsum[rt << 1 | 1];    if (lsum[rt] == (m - (m >> 1))) lsum[rt] += lsum[rt << 1 | 1];    if (rsum[rt] == (m >> 1)) rsum[rt] += rsum[rt << 1];    sum[rt] = max(max(sum[rt << 1], sum[rt << 1 | 1]), rsum[rt << 1] + lsum[rt << 1 | 1]);}void build(int l, int r, int rt){    if (l == r){        sum[rt] = lsum[rt] = rsum[rt] = 1;        return;    }    int mid = (l + r) >> 1;    build(lson);    build(rson);    up(rt, r - l + 1);}void update(int key, int add, int l, int r, int rt){    if (l == r){        if (add) sum[rt] = lsum[rt] = rsum[rt] = 0;        if (!add) sum[rt] = lsum[rt] = rsum[rt] = 1;        return;    }    int mid = (l + r) >> 1;    if (key <= mid) update(key, add, lson);    else update(key, add, rson);    up(rt, r - l + 1);}int ask(int key, int l, int r, int rt){    if (l == r){        if (sum[rt] == 1) return 1; return 0;    }    int mid = (l + r) >> 1;    int a = l + lsum[rt] - 1; int b = r - rsum[rt] + 1;    int ans = 0;    if (key >= l&&key <= a) ans = lsum[rt];    if (key >= b&&key <= r) ans = max(ans, rsum[rt]);    int c = mid - rsum[rt << 1] + 1; int d = lsum[rt << 1 | 1] + mid;    if (key >= c&&key <= d) ans = max(ans, rsum[rt << 1] + lsum[rt << 1 | 1]);    if (ans) return ans;    if (key <= mid) return ask(key, lson);    else return ask(key, rson);}int q[maxn<<1];int main(){    int n, m;    int a; char str[100];    while (scanf("%d%d",&n,&m)!=EOF){        build(1, n, 1);        int ans = 0;        for (int i = 0; i < m; i++){            scanf("%s%d", str, &a);            if (str[0] == D){                update(a, 1, 1, n, 1);                q[ans++] = a;            }            if (str[0] == Q){                printf("%d\n", ask(a, 1, n, 1));            }            if (str[0] == R){                if(ans<=0) continue;                int t = q[--ans];                update(t, 0, 1, n, 1);            }        }    }    return 0;}