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

<题目链接>

题目大意：

一个长度为n的线段，下面m个操作

D x 表示将单元x毁掉

R  表示修复最后毁坏的那个单元

Q x  询问这个单元以及它周围有多少个连续的单元，如果它本身已经被毁坏了就是0。

解题分析：

用线段树求指定点所在的最长连续区间，属于线段树区间合并类型的题，线段树的每个节点需要维护三个值，分别是对应区间的最长连续区间长度，对应区间最长连续区间前缀，对应区间最长连续后缀，然后就是在每次update之后都维护一下这三个值就行。并且注意一下query 时的操作。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define Lson rt<<1,l,mid
#define Rson rt<<1|1,mid+1,r
const int N=50000+4;
int len[N<<2],llen[N<<2],rlen[N<<2];
int n,m;

void Pushup(int rt,int length){    //更新维护每个节点的三个值
    llen[rt]=llen[rt<<1];   //首先当前节点的最长连续前缀==左子节点的最长连续前缀     
    rlen[rt]=rlen[rt<<1|1];
    /*-- 进一步更新 --*/   
    if(llen[rt<<1]==(length-(length>>1)))llen[rt]+=llen[rt<<1|1]; //如果左子节点的最长连续前缀为整个左子区间，那么本节点的前缀还要加上右子区间的最长前缀
    if(rlen[rt<<1|1]==(length>>1))rlen[rt]+=rlen[rt<<1];          //同理
    /*-- 最终更新本节点的最长连续长度  --*/
    len[rt]=max(max(len[rt<<1],len[rt<<1|1]),rlen[rt<<1]+llen[rt<<1|1]); 
}

void build(int rt,int l,int r){
    if(l==r){
        len[rt]=llen[rt]=rlen[rt]=1;
        return;
    }
    int mid=(l+r)>>1;
    build(Lson);
    build(Rson);
    Pushup(rt,r-l+1);
}

void update(int rt,int l,int r,int loc,int c){   //单点更新
    if(l==r){
        len[rt]=llen[rt]=rlen[rt]=c; 
        return;
    }
    int mid=(l+r)>>1;
    if(loc<=mid)update(Lson,loc,c);
    if(loc>mid)update(Rson,loc,c);
    Pushup(rt,r-l+1);
}

int query(int rt,int l,int r,int loc){
    if(l==r)return len[rt];
    int mid=(l+r)>>1;
    if(loc<=mid){
        if(loc+rlen[rt<<1]>mid)return rlen[rt<<1]+llen[rt<<1|1]; //如果loc在左子区间的最长后缀和右子区间的最长前缀中，直接输出这两个前后缀之和即可
        else return query(Lson,loc);  //否则的话，继续向左子节点查询
    }
    else{
        if(mid+llen[rt<<1|1]>=loc)return rlen[rt<<1]+llen[rt<<1|1];//因为右子区间是(mid,r]左开区间，所以这里判断loc是否在右子区间的前缀的范围内用的是 ">="
        else return query(Rson,loc);
    }
}

int main(){
    while(scanf("%d %d",&n,&m)!=EOF){
        build(1,1,n);
        int tot=0,stk[N+10]; //stk模拟栈
        while(m--){
            char str[10];
            scanf("%s",str);
            if(str[0]=='D'){
                int cal;scanf("%d",&cal);
                stk[++tot]=cal;
                update(1,1,n,cal,0);
            }
            else if(str[0]=='R'){
                int cal=stk[tot--];
                update(1,1,n,cal,1);
            }
            else{
                int cal;scanf("%d",&cal);
                printf("%d\n",query(1,1,n,cal));
            }
        }
    }
    return 0;
}

 

 

2018-09-23