线段树-hdu3397

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3397

题目描述：

题目大意：给我们一串二进制串，需要我们对其进行以下操作：

1、输入0，a，b，将a，b范围内的所有二进制数字变为0；

2、输入1，a，b，将a，b范围内的所有二进制数字变为1；

3、输入2，a，b，将a，b范围内的所有二进制数字0变为1，1变为0；

4、输入3，a，b，统计a，b范围内的二进制数字1的个数并输出；

4、输入4，a，b，统计a，b范围内的最长连续的1个数并输出；

代码实现：

#include <cstdio>
#include <algorithm>
using namespace std;
const int MAXN =1e6;
int N,M;
struct node{
    /**val记录 0,1操作——赋值
    初始值为-1;
    val=0表示将当前节点（x）范围内的值全部赋值为0
    val=1表示将当前节点（x）范围内的值全部赋值为1

    flag记录2操作 ——转换
    默认值为0;
    flag=1表示将当前节点（x）范围内的0全部转换为1，1全部转换为0

    lmax_0记录x范围内从左开始连续的0的长度
    rmax_0记录x范围内从右开始连续的0的长度
    max_0记录x范围内最长连续的0的长度
    sum_0记录x范围内0的个数
    同理，lmax_1，rmax_1，max_1,sum_1是用来记录1的*/
    int l,r,val,flag;
    int lmax_0,rmax_0,max_0,sum_0;
    int lmax_1,rmax_1,max_1,sum_1;
    int len(){return r-l+;}//记录每个节点区间的总长度
    int mid(){return (r+l)/;}

    void change(int a,int rev){
        if(a==){//0操作
            lmax_0=rmax_0=max_0=sum_0=len();
            lmax_1=rmax_1=max_1=sum_1=;
            val=a;flag=;
        }else if(a==){//1操作
            lmax_0=rmax_0=max_0=sum_0=;
            lmax_1=rmax_1=max_1=sum_1=len();
            val=a;flag=;
        }
        if(rev){//如果rev=1，执行2操作
            swap(lmax_0,lmax_1);swap(rmax_0,rmax_1);
            swap(max_0,max_1);swap(sum_0,sum_1);
            flag^=;//即flag由0变为1
        }
    }
}tree[MAXN*];
int a[MAXN];
///push_up(tree[1],tree[2],tree[3]);
void push_up(node &a,node &b,node &c){
    //更新x节点的lmax_0、rmax_0、max_0、sum_0
    //以及x节点的lmax_1、rmax_1、max_1、sum_1
    a.lmax_0=b.lmax_0;a.rmax_0=c.rmax_0;//父亲的左边最长0串=左子树的左边最长0串，父亲的右边最长0串=右子树的右边最长0串
    a.lmax_1=b.lmax_1;a.rmax_1=c.rmax_1;
    a.sum_0=b.sum_0+c.sum_0; a.sum_1=b.sum_1+c.sum_1;//父亲总的0的个数=左子树0的个数+右子树0的个数，父亲总的1的个数=左子树1的个数+右子树1的个数
    a.max_0=max(max(b.max_0,c.max_0),(b.rmax_0+c.lmax_0));///如果出现像0000001011这种情况，a.max_0的值就应该为b.rmax_0+c.lmax_0
    a.max_1=max(max(b.max_1,c.max_1),(b.rmax_1+c.lmax_1));
    if(b.len()==b.lmax_0) a.lmax_0+=c.lmax_0;///比如像0000011011这种情况，注意a.lmax_0在刚开始就已经赋值为b.lmax_0了，这只是再次判断以防万一
    if(b.len()==b.lmax_1) a.lmax_1+=c.lmax_1;
    if(c.len()==c.rmax_0) a.rmax_0+=b.rmax_0;
    if(c.len()==c.rmax_1) a.rmax_1+=b.rmax_1;
}
///build(1,0,N-1);//第一个节点的区间的左右的值分别为0，N-1
void build(int x,int l,int r){
    tree[x].l=l;tree[x].r=r;
    tree[x].lmax_0=tree[x].rmax_0=;
    tree[x].lmax_1=tree[x].rmax_1=;
    tree[x].max_0=tree[x].max_1=;
    tree[x].sum_0=tree[x].sum_1=;
    tree[x].flag=;tree[x].val=-;
    if(l==r){
        tree[x].change(a[l],);
        return ;
    }
    int mid=(l+r)/;
    build(x<<,l,mid);
    build(x<<|,mid+,r);
    push_up(tree[x],tree[x<<],tree[x<<|]);
}
///if(a==0||a==1||a==2) update(1,b,c,a);
/*
0用于将范围内所有的数变为0,1用于将范围内所有的数变为1,2用于将范围内所有的0变为1，1变为0
*/
void update(int x,int l,int r,int val){
    int L=tree[x].l,R=tree[x].r;
    if(l<=L && R<=r){///[L,R]包含于[l,r]
        if(val==||val==) tree[x].change(val,);//这一步即能实现将值进行转换
        else tree[x].change(-,);//这一步实现2操作
        return ;
    }
    //push_down的作用，实现向下更新
    if(tree[x].val!=-||tree[x].flag)//仅针对于刚刚更新的节点来说，即上方代码实现将当前节点与该节点以上的节点更新后，
                                        //接下来的节点tree[x].val已经改变了，所以需要对其左右子区间进行更新
    {
        tree[x<<].change(tree[x].val,tree[x].flag);
        tree[x<<|].change(tree[x].val,tree[x].flag);
        tree[x].val=-;tree[x].flag=;
    }
    int mid=(L+R)/;
    ///如果mid直接处在l的右边就只更新左区间（条件1），处在r的左边就只更新右区间（条件2），如mid处在[l,r]之间，需要更新左右区间（条件1+2）
    if(mid>=l)
        update(x<<,l,r,val);
    if(mid<r)
        update(x<<|,l,r,val);
    push_up(tree[x],tree[x<<],tree[x<<|]);
}

int query(int x,int l,int r,int val)
{
    int L=tree[x].l,R=tree[x].r;
    if(l<=L && R<=r)
    {
        if(val==) return tree[x].sum_1;
        else return tree[x].max_1;
    }
    int mid=(L+R)/;
    //push_down的作用，实现向下更新
    if(tree[x].val!=-||tree[x].flag)
    {
        tree[x<<].change(tree[x].val,tree[x].flag);
        tree[x<<|].change(tree[x].val,tree[x].flag);
        tree[x].val=-;tree[x].flag=;
    }
    if(r<=mid)
        return query(x<<,l,r,val);
    else if(l>mid)
        return query(x<<|,l,r,val);
    else
    {
        int max1=query(x<<,l,r,val);
        int max2=query(x<<|,l,r,val);
        if(val==)
            return max1+max2;
        int max3=min(tree[x<<].r-l+,tree[x<<].rmax_1)+min(r-tree[x<<|].l+,tree[x<<|].lmax_1);
        return max(max(max1,max2),max3);
    }
}

int main(){
    int T;scanf("%d",&T);
    while(T--){
        scanf("%d%d",&N,&M);
        for(int i=;i<N;i++)
            scanf("%d",&a[i]);
        build(,,N-);
        while(M--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            if(a==||a==||a==) update(,b,c,a);
            else printf("%d\n",query(,b,c,a));//如果a等于3，输出[b,c]之间1的个数，a=4输出[b,c]之间最长连续1的个数
        }
    }
    return ;
}4