线段树-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+1;}//记录每个节点区间的总长度
    int mid(){return (r+l)/2;}

    void change(int a,int rev){
        if(a==0){//0操作
            lmax_0=rmax_0=max_0=sum_0=len();
            lmax_1=rmax_1=max_1=sum_1=0;
            val=a;flag=0;
        }else if(a==1){//1操作
            lmax_0=rmax_0=max_0=sum_0=0;
            lmax_1=rmax_1=max_1=sum_1=len();
            val=a;flag=0;
        }
        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^=1;//即flag由0变为1
        }
    }
}tree[MAXN*4];
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=0;
    tree[x].lmax_1=tree[x].rmax_1=0;
    tree[x].max_0=tree[x].max_1=0;
    tree[x].sum_0=tree[x].sum_1=0;
    tree[x].flag=0;tree[x].val=-1;
    if(l==r){
        tree[x].change(a[l],0);
        return ;
    }
    int mid=(l+r)/2;
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
    push_up(tree[x],tree[x<<1],tree[x<<1|1]);
}
///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==0||val==1) tree[x].change(val,0);//这一步即能实现将值进行转换
        else tree[x].change(-1,1);//这一步实现2操作
        return ;
    }
    //push_down的作用,实现向下更新
    if(tree[x].val!=-1||tree[x].flag)//仅针对于刚刚更新的节点来说,即上方代码实现将当前节点与该节点以上的节点更新后,
                                        //接下来的节点tree[x].val已经改变了,所以需要对其左右子区间进行更新
    {
        tree[x<<1].change(tree[x].val,tree[x].flag);
        tree[x<<1|1].change(tree[x].val,tree[x].flag);
        tree[x].val=-1;tree[x].flag=0;
    }
    int mid=(L+R)/2;
    ///如果mid直接处在l的右边就只更新左区间(条件1),处在r的左边就只更新右区间(条件2),如mid处在[l,r]之间,需要更新左右区间(条件1+2)
    if(mid>=l)
        update(x<<1,l,r,val);
    if(mid<r)
        update(x<<1|1,l,r,val);
    push_up(tree[x],tree[x<<1],tree[x<<1|1]);
}

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==3) return tree[x].sum_1;
        else return tree[x].max_1;
    }
    int mid=(L+R)/2;
    //push_down的作用,实现向下更新
    if(tree[x].val!=-1||tree[x].flag)
    {
        tree[x<<1].change(tree[x].val,tree[x].flag);
        tree[x<<1|1].change(tree[x].val,tree[x].flag);
        tree[x].val=-1;tree[x].flag=0;
    }
    if(r<=mid)
        return query(x<<1,l,r,val);
    else if(l>mid)
        return query(x<<1|1,l,r,val);
    else
    {
        int max1=query(x<<1,l,r,val);
        int max2=query(x<<1|1,l,r,val);
        if(val==3)
            return max1+max2;
        int max3=min(tree[x<<1].r-l+1,tree[x<<1].rmax_1)+min(r-tree[x<<1|1].l+1,tree[x<<1|1].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=0;i<N;i++)
            scanf("%d",&a[i]);
        build(1,0,N-1);
        while(M--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            if(a==0||a==1||a==2) update(1,b,c,a);
            else printf("%d\n",query(1,b,c,a));//如果a等于3,输出[b,c]之间1的个数,a=4输出[b,c]之间最长连续1的个数
        }
    }
    return 0;
}4

猜你喜欢

转载自www.cnblogs.com/LJHAHA/p/10018697.html