题目链接: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)范围内的值全部赋值为0val=1表示将当前节点(x)范围内的值全部赋值为1flag记录2操作 ——转换默认值为0;flag=1表示将当前节点(x)范围内的0全部转换为1,1全部转换为0lmax_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_1a.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_0a.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