HDU 3911 Black And White

题目

线段树。两种操作，一种是异或，一种是查询最长的连续的1；

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 100010

inline int input()
{
    bool IsN=0;
    int ret=0;
    char c;
    c=getchar();
    while(c<'0'||c>'9')
    {
        if(c=='-') IsN=1;
        c=getchar();
    }
    while(c>='0'&&c<='9')
    {
        ret=ret*10+c-'0';
        c=getchar();
    }
    return IsN?-ret:ret;
}

struct node
{
    int l,r;
    int lsum1,lsum0,rsum1,rsum0,msum1,msum0;
    int lazy;
}root[N*6];
int n,m,a[N],op,x,y;

inline void change(int t)
{
    swap(root[t].lsum1,root[t].lsum0);
    swap(root[t].rsum1,root[t].rsum0);
    swap(root[t].msum1,root[t].msum0);
}
inline void Push_up(int t)
{
    int ll=root[t*2].r-root[t*2].l+1;
    int lr=root[t*2+1].r-root[t*2+1].l+1;

    root[t].lsum1=root[t*2].lsum1+(root[t*2].lsum1==ll?root[t*2+1].lsum1:0);
    root[t].rsum1=root[t*2+1].rsum1+(root[t*2+1].rsum1==lr?root[t*2].rsum1:0);
    root[t].lsum0=root[t*2].lsum0+(root[t*2].lsum0==ll?root[t*2+1].lsum0:0);
    root[t].rsum0=root[t*2+1].rsum0+(root[t*2+1].rsum0==lr?root[t*2].rsum0:0);

    root[t].msum0=max(root[t*2].rsum0+root[t*2+1].lsum0,max(root[t*2].msum0,root[t*2+1].msum0));
    root[t].msum1=max(root[t*2].rsum1+root[t*2+1].lsum1,max(root[t*2].msum1,root[t*2+1].msum1));
}
inline void Push_down(int t)
{
    if(root[t].lazy!=0)
    {
        root[t].lazy^=1;
        root[t*2].lazy^=1;
        root[t*2+1].lazy^=1;

        change(t*2),change(t*2+1);
    }
}
inline void build(int t,int x,int y)
{
    root[t].l=x;
    root[t].r=y;
    root[t].lazy=0;
    if(x==y)
    {
        root[t].lsum0=root[t].rsum0=root[t].msum0=(a[x]==0);
        root[t].lsum1=root[t].rsum1=root[t].msum1=(a[x]==1);
        return;
    }
    int m=(x+y)>>1;
    build(t*2,x,m);
    build(t*2+1,m+1,y);
    Push_up(t);
}
inline void Modefiy(int t,int x,int y)
{
    int l=root[t].l,r=root[t].r;
    int m=(l+r)>>1;
    if(l==x&&r==y)
    {
        root[t].lazy^=1;
        change(t);
        return;
    }
    Push_down(t);
    if(x<=m) Modefiy(t*2,x,min(m,y));
    if(y>m) Modefiy(t*2+1,max(x,m+1),y);
    Push_up(t);
}
inline int query(int t,int x,int y)
{
    int l=root[t].l,r=root[t].r;
    int m=(l+r)>>1;
    if(l==x&&r==y)
    {
        return root[t].msum1;
    }
    Push_down(t);
    if(y<=m) return query(t*2,x,y);
    else if(x>m) return query(t*2+1,x,y);
    else
    {
        int p=max(query(t*2,x,m),query(t*2+1,m+1,y));
        int p1=min(root[t*2].rsum1,root[t*2].r-x+1);
        int p2=min(root[t*2+1].lsum1,y-root[t*2+1].l+1);
        return max(p1+p2,p);
    }
}

int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
        a[i]=input();
        build(1,1,n);
        m=input();
        for(int i=1;i<=m;i++)
        {
            op=input(),x=input(),y=input();
            if(op) Modefiy(1,x,y);
            else printf("%d\n",query(1,x,y));
        }
    }
    return 0;
}