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+10,M=4e6+10,inf=1e9+10;
struct is
{
    int l,r;
    int lm,mm,rm;
}tree[N<<2];
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+1);
    if(l==r)
    return;
    int mid=(l+r)>>1;
    buildtree(l,mid,pos<<1);
    buildtree(mid+1,r,pos<<1|1);
}
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)>>1;
    if(point<=mid)
    update(point,change,pos<<1);
    else
    update(point,change,pos<<1|1);
    tree[pos].mm=max(tree[pos<<1].mm,max(tree[pos<<1|1].mm,tree[pos<<1].rm+tree[pos<<1|1].lm));
    tree[pos].rm=tree[pos<<1|1].rm;
    tree[pos].lm=tree[pos<<1].lm;
    if(tree[pos<<1|1].rm==tree[pos<<1|1].r-tree[pos<<1|1].l+1)
    tree[pos].rm+=tree[pos<<1].rm;
    if(tree[pos<<1].lm==tree[pos<<1].r-tree[pos<<1].l+1)
    tree[pos].lm+=tree[pos<<1|1].lm;
}
int query(int x,int pos)
{
    if(tree[pos].l==tree[pos].r||tree[pos].mm==0||tree[pos].mm==tree[pos].r-tree[pos].l+1)
        return tree[pos].mm;
    int mid=(tree[pos].r+tree[pos].l)>>1;
    if(x<=mid)
    {
        if(x>=tree[pos<<1].r-tree[pos<<1].rm+1)
        return query(x,pos<<1)+query(mid+1,pos<<1|1);
        else
        return query(x,pos<<1);
    }
    else
    {
        if(x<=tree[pos<<1|1].lm+tree[pos<<1|1].l-1)
            return query(x,pos<<1|1)+query(mid,pos<<1);
        else
        return query(x,pos<<1|1);
    }
}
int main()
{
    int x,y,z,i,t;
    while(~scanf("%d%d",&x,&y))
    {
        buildtree(1,x,1);
        while(y--)
        {
            int flag,x;
            scanf("%d%d",&flag,&x);
            if(flag==0||flag==1)
            update(x,flag,1);
            else
            printf("%d\n",query(x,1));
        }
    }
    return 0;
}