题解 P2253 【好一个一中腰鼓!】

线段树好题 虽然由于数据水 一大堆暴力过的

做完之后对线段树的理解加深了很多

本题的本质是求满足条件的最长子序列长度

那么我们利用线段树维护三个数据

int lm,rm,mm;//分别记录最左边开始的最长段 最右边开始的最长段 中间最长段 

在每次进行修改的时候判断左右区间是否可以合并

代码如下:

#include<bits/stdc++.h>
using namespace std;//len=r-l+1
int n,m,x,a[20005];
struct SegmentTree{
    int l,r;
    int lm,rm,mm;//分别记录最左边开始的最长段 最右边开始的最长段 中间最长段 
}t[100005];
void push(int p){
    t[p].lm=t[p<<1].lm;
    //判断左区间能否与右区间合并
    if (a[t[p<<1].r]!=a[t[p<<1|1].l]&&t[p<<1].lm==(t[p<<1].r-t[p<<1].l+1)) t[p].lm+=t[p<<1|1].lm;
    t[p].rm=t[p<<1|1].rm;
    if (a[t[p<<1|1].l]!=a[t[p<<1].r]&&t[p<<1|1].rm==(t[p<<1|1].r-t[p<<1|1].l+1)) t[p].rm+=t[p<<1].rm;
    t[p].mm=max(t[p<<1].mm,t[p<<1|1].mm);
    if (a[t[p<<1].r]!=a[t[p<<1|1].l]) t[p].mm=max(t[p].mm,t[p<<1].rm+t[p<<1|1].lm);
}
void change(int p,int d){
    if (t[p].l==d&&t[p].r==d){
        a[d]^=1;
        return;
    }
    int mid=(t[p].l+t[p].r)>>1;
    if (d<=mid) change(p<<1,d);
    else change(p<<1|1,d);
    push(p);
}
void build(int p,int l,int r){
    t[p].l=l;t[p].r=r; 
    if (l==r){
        t[p].lm=1;
        t[p].mm=1;
        t[p].rm=1;
        return;
    }
    int mid=(l+r)>>1;
    build (p<<1,l,mid);
    build (p<<1|1,mid+1,r);
    push (p);
}
int main(){
    scanf("%d%d",&n,&m);
    build (1,1,n);
    for (int i=1;i<=m;i++){
        scanf("%d",&x);
        change(1,x);
        printf("%d\n",max(t[1].mm,max(t[1].lm,t[1].rm)));
    }
    return 0;
} 

猜你喜欢

转载自www.cnblogs.com/Hiraeth-dh/p/10961822.html