hdu 1540 Tunnel Warfare（线段树单点更新+区间合并）

Tunnel Warfare

题目链接：

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

解题思路：

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

用线段树来记录。然后用三个变量分别代表左边连续区间，右边连续区间和最大连续区间。

AC代码：

#include <iostream>
#include <cstdio>
#include <stack>
#include <algorithm>
using namespace std;

const int maxn = 50005;
int n,m;
stack<int> sta;

struct node{
    int l,r;
    int ls,rs,ms;
    //ls,左端最大连续区间，rs右端最大连续区间，ms区间内最大连续区间
}tree[maxn<<2];

void build(int id,int l,int r){
    tree[id].l = l;
    tree[id].r = r;
    tree[id].ls = tree[id].rs = tree[id].ms = r-l+1;
    if(l == r)
        return ;
    int mid = (l+r)>>1;
    build(id<<1,l,mid);
    build(id<<1|1,mid+1,r);
}

void update(int id,int x,int val){
    if(tree[id].l == tree[id].r){
        if(val == 1)
            tree[id].ls = tree[id].rs = tree[id].ms = 1;//修复
        else
            tree[id].ls = tree[id].rs = tree[id].ms = 0;//破坏
        return ;
    }
    int mid = (tree[id].l+tree[id].r)>>1;
    if(x <= mid)
        update(id<<1,x,val);
    else
        update(id<<1|1,x,val);
    tree[id].ls = tree[id<<1].ls;//左区间
    tree[id].rs = tree[id<<1|1].rs;//右区间
    tree[id].ms = max(max(tree[id<<1].ms,tree[id<<1|1].ms),tree[id<<1].rs+tree[id<<1|1].ls);
    if(tree[id<<1].ls == tree[id<<1].r-tree[id<<1].l+1)//左子树区间如果满了，则父亲左区间要加上右孩子的左区间
        tree[id].ls += tree[id<<1|1].ls;
    if(tree[id<<1|1].rs == tree[id<<1|1].r-tree[id<<1|1].l+1)//同理
        tree[id].rs += tree[id<<1].rs;
}

int query(int id,int x){
    if(tree[id].l == tree[id].r ||tree[id].ms == tree[id].r-tree[id].l+1)//到了叶子节点或者该访问区间为空或者已满都不必要往下走了
        return tree[id].ms;
    int mid = (tree[id].l+tree[id].r)>>1;
    if(x <= mid){
        if(x >= tree[id<<1].r-tree[id<<1].rs+1)//因为t<=mid，看左子树，a[2*i].r-a[2*i].rs+1代表左子树右边连续区间的左边界值，如果t在左子树的右区间内，则要看右子树的左区间有多长并返回
            return query(id<<1,x)+query(id<<1|1,mid+1);
        else
            return query(id<<1,x);//如果不在左子树的右边界区间内，则只需要看左子树
    }
    else{
        if(x <= tree[id<<1|1].l+tree[id<<1|1].ls-1)//同理
            return query(id<<1|1,x)+query(id<<1,mid);
        else
            return query(id<<1|1,x);
    }
}

int main(){
    while(~scanf("%d%d",&n,&m)){
        while(!sta.empty())
            sta.pop();
        char op[5];
        int x;
        build(1,1,n);
        while(m--){
            scanf("%s",op);
            if(op[0] == 'D'){
                scanf("%d",&x);
                sta.push(x);
                update(1,x,0);
            }
            else if(op[0] == 'Q'){
                scanf("%d",&x);
                printf("%d\n",query(1,x));
            }
            else{
                if(!sta.empty()){
                    x = sta.top();
                    sta.pop();
                    update(1,x,1);
                }
            }
        }
    }
    return 0;
}