BZOJ 1858 序列操作 (线段树)

题目链接：http://61.187.179.132/JudgeOnline/problem.php?id=1858

题意：给一个01序列，有5种操作：

(1)0 L R 将[L,R]之间的数字都变成0；

(2)1 L R 将[L,R]之间的数字都变成1；

(3)2 L R 将[L R]之间的数字都取反；

(4)3 L R 询问[L R]之间1的个数；

(5)4 L R 询问[L,R]之间连续1的个数最大是多少

思路：

（1）每个节点信息：

l0,l1:左端开始连续01的个数；

r0,r1:右端开始连续01的个数；

max0,max1:区间内最长连续01的长度；

cnt0,cnt1:区间内01的个数

flag:-1表示无标记，0表示全为0,1表示全为1,2表示取反

（2）不管查询还是更新，最主要的是当前区间的flag不是-1时，向下传递该状态。

 

struct node
{
    int l0,l1,r0,r1,max0,max1,flag,len,cnt0,cnt1;
    int L,R,mid;
};

const int MAX=100005;
node a[MAX<<2];
int n,m,p[MAX];

void update(int t)
{
    a[t].l0=a[t*2].l0;
    if(a[t*2].l0==a[t*2].len) a[t].l0+=a[t*2+1].l0;
    a[t].r0=a[t*2+1].r0;
    if(a[t*2+1].r0==a[t*2+1].len) a[t].r0+=a[t*2].r0;
    a[t].max0=max(a[t*2].max0,a[t*2+1].max0);
    a[t].max0=max(a[t].max0,a[t*2].r0+a[t*2+1].l0);
    a[t].cnt0=a[t*2].cnt0+a[t*2+1].cnt0;

    a[t].l1=a[t*2].l1;
    if(a[t*2].l1==a[t*2].len) a[t].l1+=a[t*2+1].l1;
    a[t].r1=a[t*2+1].r1;
    if(a[t*2+1].r1==a[t*2+1].len) a[t].r1+=a[t*2].r1;
    a[t].max1=max(a[t*2].max1,a[t*2+1].max1);
    a[t].max1=max(a[t].max1,a[t*2].r1+a[t*2+1].l1);
    a[t].cnt1=a[t*2].cnt1+a[t*2+1].cnt1;
}


void build(int t,int L,int R)
{
    a[t].L=L;
    a[t].R=R;
    a[t].mid=(L+R)>>1;
    a[t].len=R-L+1;
    a[t].flag=-1;
    if(L==R)
    {
        a[t].l0=a[t].r0=a[t].max0=a[t].cnt0=!p[L];
        a[t].l1=a[t].r1=a[t].max1=a[t].cnt1=p[L];
        return;
    }
    build(t*2,L,a[t].mid);
    build(t*2+1,a[t].mid+1,R);
    update(t);
}

void swap(int &x,int &y)
{
    int temp;
    temp=x;
    x=y;
    y=temp;
}

void change(int t,int L,int R,int cmd)
{
    if(a[t].L==L&&a[t].R==R)
    {
        if(cmd==0)
        {
            a[t].l0=a[t].r0=a[t].max0=a[t].cnt0=a[t].len;
            a[t].l1=a[t].r1=a[t].max1=a[t].cnt1=0;
            a[t].flag=0;
        }
        else if(cmd==1)
        {
            a[t].l0=a[t].r0=a[t].max0=a[t].cnt0=0;
            a[t].l1=a[t].r1=a[t].max1=a[t].cnt1=a[t].len;
            a[t].flag=1;
        }
        else if(cmd==2)
        {
            swap(a[t].l0,a[t].l1);
            swap(a[t].r0,a[t].r1);
            swap(a[t].cnt0,a[t].cnt1);
            swap(a[t].max0,a[t].max1);
            a[t].flag=1-a[t].flag;
        }
        return;
    }
    if(a[t].flag>=0)
    {
        change(t*2,a[t*2].L,a[t*2].R,a[t].flag);
        change(t*2+1,a[t*2+1].L,a[t*2+1].R,a[t].flag);
        a[t].flag=-1;
    }
    if(R<=a[t].mid) change(t*2,L,R,cmd);
    else if(L>a[t].mid) change(t*2+1,L,R,cmd);
    else
    {
        change(t*2,L,a[t].mid,cmd);
        change(t*2+1,a[t].mid+1,R,cmd);
    }
    update(t);
}

node get(int t,int L,int R)
{
    if(a[t].L==L&&a[t].R==R) return a[t];
    if(a[t].flag>=0)
    {
        change(t*2,a[t*2].L,a[t*2].R,a[t].flag);
        change(t*2+1,a[t*2+1].L,a[t*2+1].R,a[t].flag);
        a[t].flag=-1;
    }
    node t1,t2,t3;
    if(R<=a[t].mid) return get(t*2,L,R);
    else if(L>a[t].mid) return get(t*2+1,L,R);
    else
    {
        t1=get(t*2,L,a[t].mid);
        t2=get(t*2+1,a[t].mid+1,R);
        t3.cnt1=t1.cnt1+t2.cnt1;
        t3.l1=t1.l1;
        if(t1.l1==t1.len) t3.l1+=t2.l1;
        t3.r1=t2.r1;
        if(t2.r1==t2.len) t3.r1+=t1.r1;
        t3.max1=max(t1.max1,t2.max1);
        t3.max1=max(t3.max1,t1.r1+t2.l1);
        t3.len=t1.len+t2.len;
        return t3;
    }
}



int main()
{
    while(scanf("%d%d",&n,&m)!=-1)
    {
        int i;
        for(i=1;i<=n;i++) scanf("%d",p+i);
        build(1,1,n);
        int cmd,L,R;
        node t;
        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&cmd,&L,&R);
            L++;
            R++;
            if(cmd<=2) change(1,L,R,cmd);
            else
            {
                t=get(1,L,R);
                if(cmd==3) printf("%d\n",t.cnt1);
                else printf("%d\n",t.max1);
            }
        }
    }
    return 0;
}

　　

                     

转载于:https://www.cnblogs.com/jianglangcaijin/archive/2012/12/04/2802011.html