UESTC 1061 秋实大哥与战争 线段树区间合并

秋实大哥与战争

Time Limit: 3000/1000MS (Java/Others)     Memory Limit: 65535/65535KB (Java/Others)

男儿何不带吴钩，收取关山五十州。

征战天下是秋实大哥一生的梦想，所以今天他又在练习一个对战游戏。

秋实大哥命令所有士兵从左到右排成了一行来抵挡敌人的攻击。

敌方每一次会攻击一个士兵，这个士兵就会阵亡，整个阵列就会从这个位置断开；同时有的时候已阵亡的士兵会受人赢气息感染而复活。

秋实大哥想知道某一时刻某一个士兵所在的阵列的长度是多少。

Input

第一行包含两个整数nn，mm，表示秋实大哥的士兵数目和接下来发生的事件数目。

接下来mm行，每一行是以下三种事件之一：

0 x : 表示x位置的士兵受到攻击阵亡
1 x : 表示x位置的士兵受人赢气息感染复活
2 x : 秋实大哥想知道第x个士兵所在阵列的长度

1≤n，m≤1000001≤n，m≤100000，1≤x≤n。1≤x≤n。

Output

对于每一个22 xx事件，输出对应的答案占一行。

Sample input and output

Sample Input	Sample Output
5 3 2 2 0 3 2 2	5 2

Source

2015 UESTC Training for Data Structures

板子题；

#include<bits/stdc++.h>
using namespace std;
#define ll __int64
#define mod 1000000007
#define pi (4*atan(1.0))
const int N=1e5+,M=4e6+,inf=1e9+;
struct is
{
    int l,r;
    int lm,mm,rm;
}tree[N<<];
void buildtree(int l,int r,int pos)
{
    tree[pos].l=l;
    tree[pos].r=r;
    tree[pos].lm=tree[pos].rm=tree[pos].mm=(r-l+);
    if(l==r)
    return;
    int mid=(l+r)>>;
    buildtree(l,mid,pos<<);
    buildtree(mid+,r,pos<<|);
}
void update(int point,int change,int pos)
{
    if(tree[pos].l==point&&tree[pos].r==point)
    {
        tree[pos].lm=tree[pos].mm=tree[pos].rm=change;
        return;
    }
    int mid=(tree[pos].l+tree[pos].r)>>;
    if(point<=mid)
    update(point,change,pos<<);
    else
    update(point,change,pos<<|);
    tree[pos].mm=max(tree[pos<<].mm,max(tree[pos<<|].mm,tree[pos<<].rm+tree[pos<<|].lm));
    tree[pos].rm=tree[pos<<|].rm;
    tree[pos].lm=tree[pos<<].lm;
    if(tree[pos<<|].rm==tree[pos<<|].r-tree[pos<<|].l+)
    tree[pos].rm+=tree[pos<<].rm;
    if(tree[pos<<].lm==tree[pos<<].r-tree[pos<<].l+)
    tree[pos].lm+=tree[pos<<|].lm;
}
int query(int x,int pos)
{
    if(tree[pos].l==tree[pos].r||tree[pos].mm==||tree[pos].mm==tree[pos].r-tree[pos].l+)
        return tree[pos].mm;
    int mid=(tree[pos].r+tree[pos].l)>>;
    if(x<=mid)
    {
        if(x>=tree[pos<<].r-tree[pos<<].rm+)
        return query(x,pos<<)+query(mid+,pos<<|);
        else
        return query(x,pos<<);
    }
    else
    {
        if(x<=tree[pos<<|].lm+tree[pos<<|].l-)
            return query(x,pos<<|)+query(mid,pos<<);
        else
        return query(x,pos<<|);
    }
}
int main()
{
    int x,y,z,i,t;
    while(~scanf("%d%d",&x,&y))
    {
        buildtree(,x,);
        while(y--)
        {
            int flag,x;
            scanf("%d%d",&flag,&x);
            if(flag==||flag==)
            update(x,flag,);
            else
            printf("%d\n",query(x,));
        }
    }
    return ;
}