洛谷 P6492 [COCI2010-2011#6] STEP（端点线段树）

 

 其实这个题直接模拟即可，也算是要记住的一个模板了

对于线段树维护的每一个区间，都有一个区间最值 maxx，左端点前缀，以及右端点后缀

当更新一个点时，每次 push_up 都要做到从子节点转移到本节点，还要为其父节点的更新做铺垫

所以对于任意一个节点 x 来说

x 的左前缀是其左儿子的左前缀，x 的右后缀为其右儿子的右后缀，区间最值为其左右儿子的最大值

但是当端点处更新时，我们要判断连接处是否更大，以及为父节点考虑

如果 x 是父节点的左区间，那么判断 x 的左前缀是否要更新

如果 x 是父节点的右区间，那么判断 x 的右后缀是否要更新

#include <iostream>
#include <cstdio>
using namespace std;
const int N=2e5+5;

    int n,m;
    int a[N];
    struct Node
    {
        int sum,lsum,rsum;
    }t[N<<2];

void push_up(int id,int l,int r)
{
    t[id].lsum=t[id<<1].lsum,t[id].rsum=t[id<<1|1].rsum;
    t[id].sum=max(t[id<<1].sum,t[id<<1|1].sum);
    int mid=l+r>>1;
    if(a[mid]!=a[mid+1]){
        t[id].sum=max(t[id].sum,t[id<<1].rsum+t[id<<1|1].lsum);
        if(t[id].lsum==mid-l+1) t[id].lsum+=t[id<<1|1].lsum; //判断 x 的左前缀是否要更新
        if(t[id].rsum==r-mid) t[id].rsum+=t[id<<1].rsum; //判断 x 的右后缀是否要更新
    }
}

void build(int l,int r,int id)
{
    t[id].sum=t[id].lsum=t[id].rsum=1;
    if(l==r) return ;
    else{
        int mid=l+r>>1;
        build(l,mid,id<<1);
        build(mid+1,r,id<<1|1);
        push_up(id,l,r);
    }
}

void update(int pos,int id,int L,int R)
{
    if(L>=pos && pos>=R) a[pos]^=1;
    else{
        int mid=L+R>>1;
        if(mid>=pos) update(pos,id<<1,L,mid);
        if(pos>=mid+1) update(pos,id<<1|1,mid+1,R);
        push_up(id,L,R);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    build(1,n,1);
    while(m--){
        int x;
        scanf("%d",&x);
        update(x,1,1,n);
        printf("%d\n",t[1].sum);
    }
    return 0;
}