HDU 1540 Tunnel Warfare

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1540

题意：有n个相邻的村子，D操作将该村子与左右的连接切断，R操作是将上次切断的道路恢复，Q操作是询问包括x在内的连续的村子

思路：线段树区间合并，和hotel不一样的是这题是单点更新，更新大同小异，因为是单点更新所以没用lazy标志，询问的时候比较不一样，判断改点是否在横跨左右子数的区间内，是则直接输出答案，否则继续向子树查询

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <stack>
#define maxn 50030
using namespace std;

struct Tree
{
    int l,r;
    int lsum,rsum;
}tree[maxn*3];

stack <int> s;

void Pushup(int root)
{
    int ll=tree[root<<1].r-tree[root<<1].l+1;
    tree[root].lsum=tree[root<<1].lsum;
    if (tree[root].lsum==ll) tree[root].lsum+=tree[root<<1|1].lsum;
//cout<<":"<<root<<endl;
    int rr=tree[root<<1|1].r-tree[root<<1|1].l+1;
    tree[root].rsum=tree[root<<1|1].rsum;
    if (tree[root].rsum==rr) tree[root].rsum+=tree[root<<1].rsum;
}


void build(int root,int l,int r)
{
    tree[root].l=l;
    tree[root].r=r;
    tree[root].lsum=r-l+1;
    tree[root].rsum=r-l+1;
    if (l==r) return;
    int mid=(l+r)>>1;
    build(root<<1,l,mid);
    build(root<<1|1,mid+1,r);
    Pushup(root);
}

void update(int root,int gold,int val)
{
//cout<<":"<<root<<endl;
    if (tree[root].l==tree[root].r)
    {
        if (val==0) tree[root].lsum=tree[root].rsum=0;
        else tree[root].lsum=tree[root].rsum=1;
        return;
    }

    int mid=(tree[root].l+tree[root].r)>>1;
    if (gold<=mid) update(root<<1,gold,val);
    else if (gold>mid) update(root<<1|1,gold,val);
    Pushup(root);
   // cout<<":"<<root<<endl;
}

int que(int root,int gold)
{

    if (tree[root].l==tree[root].r ) return tree[root].lsum;
    int mid=(tree[root].l+tree[root].r)>>1;
    int ll=mid-tree[root<<1].rsum+1;
    int rr=mid+tree[root<<1|1].lsum;//cout<<ll<<":"<<rr<<endl;
    if (gold>=ll && gold <=rr) return tree[root<<1].rsum+tree[root<<1|1].lsum;
    else if (gold<ll) return que(root<<1,gold);
    else if (gold>rr) return que(root<<1|1,gold);

}

int main()
{
    int n,m;
    while (scanf("%d%d",&n,&m)!=EOF)
    {
        //while (s.size()) s.pop();
        build(1,1,n);
        for (int i=0;i<m;i++)
        {
            char tem;
            cin>>tem;
            if (tem=='D')
            {
                int a;
                scanf("%d",&a);
                update(1,a,0);
                s.push(a);
            }
            else if (tem=='R')
            {
                int a=s.top();
                s.pop();
                update(1,a,1);
            }
            else if (tem=='Q')
            {
                int a;
                scanf("%d",&a);
                printf("%d\n",que(1,a));
            }
        }
    }
}