hdu 1540 Tunnel Warfare

线段树

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

D x 表示将单元x毁掉

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

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

 

要记录单元被损坏的顺序，用一个栈就好了，毁坏就入栈，修复就出栈

说说思路，最难的是查询一个点附近有那些的连接着的区间

这需要在线段树记录三个信息，tlen，llen，rlen，这个记录和 poj 3667 Hotel 记录的意义是相同的 ， tlen表示该节点内最长的可用区间的长度，llen表示最左端数起的区间长度，rlen表示从最右端数起的区间长度

对于一个点，看它是在当前区间的左半还是右半

在左半的话，看看是不是在右端的连续区间内，是的话，还要加上右半区间的左端连续区间。否则的话，只要计算它在左半区间的连接情况即可

在右半的话同理，看看是不是在左端的连续区间内，是的话，还要加上左半区间的右端连续区间。否则的话，只要计算它在右半区间的连接情况即可

 

所以需要时刻维护好每个节点的tlen，llen，rlen，在updata函数中，和 poj 3667 Hotel 的维护是一样的

 

 

#include <cstdio>
#include <cstring>
#include <stack>
using namespace std;
#define lch(i) ((i)<<1)
#define rch(i) ((i)<<1|1)
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define N 50010

struct node
{
    int l,r;
    int mark;
    int tlen,llen,rlen;
    int mid(){
        return (l+r)>>1;
    }
    int cal_len(){
        return r-l+1;
    }
    void updata_len(){
        tlen = llen = rlen = (mark ? 0 : cal_len());
    }
}t[4*N];

void build(int l , int r , int rt)
{
    t[rt].l = l; t[rt].r = r; 
    t[rt].mark = 0;
    t[rt].tlen = t[rt].llen = t[rt].rlen = t[rt].cal_len();
    if(l == r) return ;
    int mid = t[rt].mid();
    build(l , mid , lch(rt));
    build(mid+1 , r , rch(rt));
    return ;
}

void updata(int pos ,int val ,int rt)
{
    if(t[rt].l == t[rt].r)
    {
        t[rt].mark = val;
        t[rt].updata_len();
        return ;
    } 
    if(t[rt].mark != -1)
    {
        t[lch(rt)].mark = t[rch(rt)].mark = t[rt].mark;
        t[rt].mark = -1;
        t[lch(rt)].updata_len();
        t[rch(rt)].updata_len();
    }
    int mid = t[rt].mid();
    if(pos <= mid) //在左半
        updata(pos , val , lch(rt));
    else //在右半
        updata(pos , val , rch(rt));

    int temp = max(t[lch(rt)].tlen , t[rch(rt)].tlen);
    t[rt].tlen = max(temp , t[lch(rt)].rlen + t[rch(rt)].llen);
    t[rt].llen = t[lch(rt)].llen;
    t[rt].rlen = t[rch(rt)].rlen;
    if(t[lch(rt)].tlen == t[lch(rt)].cal_len())
        t[rt].llen += t[rch(rt)].llen;
    if(t[rch(rt)].tlen == t[rch(rt)].cal_len())
        t[rt].rlen += t[lch(rt)].rlen;
    return ;
}

int query(int pos , int rt)
{
    if(t[rt].l == t[rt].r || t[rt].tlen == 0 || t[rt].tlen == t[rt].cal_len())
        return t[rt].tlen;
    //上面的部分可以改变一下写法，看看时间会不会有明显的变化
    
    if(t[rt].mark != -1)
    {
        t[lch(rt)].mark = t[rch(rt)].mark = t[rt].mark;
        t[rt].mark = -1;
        t[lch(rt)].updata_len();
        t[rch(rt)].updata_len();
    }
    int mid = t[rt].mid();
    if(pos <= mid) //查询的点在左边 
    {
        int index = mid-t[lch(rt)].rlen+1;
        if(index <= pos) //包含在内
            return query(pos , lch(rt)) + query(mid+1 , rch(rt));
        else
            return query(pos , lch(rt));
    }
    else
    {
        int index = mid+t[rch(rt)].llen;
        if(pos <= index) //包含在内
            return query(mid , lch(rt)) + query(pos , rch(rt));
        else
            return query(pos , rch(rt));
    }
}

int main()
{
    int n,m;
    stack<int>sta;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(1,n,1);
        while(!sta.empty()) sta.pop();
        while(m--)
        {
            char op[5];
            int pos , len;
            scanf("%s",op);
            if(op[0] == 'R')
            {
                if(sta.empty()) continue;
                int pos = sta.top();
                sta.pop();
                updata(pos , 0 , 1);
            }
            else if(op[0] == 'D')
            {
                scanf("%d",&pos);
                sta.push(pos);
                updata(pos , 1 , 1);
            }
            else
            {
                scanf("%d",&pos);
                len = query(pos,1);
                printf("%d\n",len);
            }
        }
    }
    return 0;

}