异或线段树模板

最新推荐文章于 2021-05-28 12:59:48 发布

咆哮_

最新推荐文章于 2021-05-28 12:59:48 发布

阅读量477

点赞数

本文链接：https://blog.csdn.net/Roar__/article/details/100710158

版权

https://ac.nowcoder.com/acm/contest/275/C

http://codeforces.com/problemset/problem/242/E

每个二进制位独占一棵线段树，很巧妙

题意：

给定一个长度为 n 的序列，有 m 次操作：

操作有2种：

1. 区间求和，即输入l,r，输出 $\sum_{i=l}^{r}a[i]$ .

2. 区间异或，即输入l,r,k，对于 $l\leq i\leq r$ ，将 a[i] 变为 $a[i]\oplus k$ .

数据范围：

$1\leq n,m\leq 10^{5}$ ， $0\leq a[i],k\leq 10^{5}$

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define MAXN 100005
#define M 20
ll num[MAXN];
int n,m;

struct node{
	int l,r;
	ll digit[M];//M棵线段树 每棵线段树管理一个二进制位 1e5不超过20位 
	int lazy[M];
}segtree[MAXN<<2];

void pushup(int root,int pos){
	segtree[root].digit[pos]=segtree[root<<1].digit[pos]+segtree[root<<1|1].digit[pos];
}

void pushdown(int root,int pos){
	if(segtree[root].lazy[pos]==1){
		segtree[root].lazy[pos]=0;
		segtree[root<<1].digit[pos]=(segtree[root<<1].r-segtree[root<<1].l+1)-segtree[root<<1].digit[pos];
		segtree[root<<1|1].digit[pos]=(segtree[root<<1|1].r-segtree[root<<1|1].l+1)-segtree[root<<1|1].digit[pos];
		segtree[root<<1].lazy[pos]^=1;//与1异或 数值翻转 0变1 1变0 
		segtree[root<<1|1].lazy[pos]^=1;//与1异或 数值翻转 0变1 1变0 
	}
}

void build(int root,int l,int r){
	segtree[root].l=l;segtree[root].r=r;

	if(l==r){
		for(int i=0;i<M;i++){
			if((num[l]&(1LL<<i)))++segtree[root].digit[i];//每棵线段树分别赋值 
		}
		return ;
	}
	
	int mid=l+r>>1;
	build(root<<1,l,mid);
	build(root<<1|1,mid+1,r);
	
	for(int i=0;i<M;i++)
		pushup(root,i);//更新父区间 
}

void update(int root,int l,int r,int pos){
	if(l>segtree[root].r||r<segtree[root].l)return ;
	if(l<=segtree[root].l&&segtree[root].r<=r){
		segtree[root].digit[pos]=(segtree[root].r-segtree[root].l+1)-segtree[root].digit[pos];
		segtree[root].lazy[pos]^=1;
		return ;
	}
	
	pushdown(root,pos);

	update(root<<1,l,r,pos);
	update(root<<1|1,l,r,pos);
	
	pushup(root,pos);
}

long long query(int root,int l,int r,int pos){
	if(l>segtree[root].r||r<segtree[root].l)return 0;
	if(l<=segtree[root].l&&segtree[root].r<=r)return segtree[root].digit[pos];
	
	pushdown(root,pos);
	
	ll ans=0;
	ans+=query(root<<1,l,r,pos);
	ans+=query(root<<1|1,l,r,pos);
	return ans;
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%lld",&num[i]);
	build(1,1,n);
	
	for(int i=1;i<=m;i++){
		int t;
		scanf("%d",&t);
		if(t==1){
			int l,r;
			scanf("%d%d",&l,&r);
			ll ans=0;
			for(int i=0;i<M;i++){//按位求和 
				ans+=query(1,l,r,i)<<i;
			}
			printf("%lld\n",ans);
		}
		else {
			int l,r,k;
			scanf("%d%d%d",&l,&r,&k);
			for(int i=0;i<M;i++){
				if(k&(1LL<<i)){//按位更新 
					update(1,l,r,i);
				}
			}
		}
	}
	return 0;
}

2019银川网络赛 A. Simple Data Structures（没法交题不知道对错）

三种位运算操作，只有下面几种情况值会发生变化

1.当k的当前位为1且进行异或操作

2.当k的当前位为1且进行或操作

3.当k的当前位为0且进行与操作

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define MAXN 100005
#define M 20
ll num[MAXN];

struct node{
	int l,r;
	ll digit[M];
	int lazyAO[M];
	int lazyXOR[M];
}segtree[MAXN<<2];


void pushup(int root,int pos){
	segtree[root].digit[pos]=segtree[root<<1].digit[pos]+segtree[root<<1|1].digit[pos];
}

void pushdown(int root,int pos){
	if(segtree[root].lazyAO[pos]!=-1){//lazyAO的值只可能为0 1 -1  
		segtree[root<<1].digit[pos]=(segtree[root<<1].r-segtree[root<<1].l+1)*segtree[root].lazyAO[pos];
		segtree[root<<1|1].digit[pos]=(segtree[root<<1|1].r-segtree[root<<1|1].l+1)*segtree[root].lazyAO[pos];
		
		segtree[root<<1].lazyAO[pos]=segtree[root<<1|1].lazyAO[pos]=segtree[root].lazyAO[pos];
		segtree[root<<1].lazyXOR[pos]=segtree[root<<1|1].lazyXOR[pos]=segtree[root].lazyXOR[pos]=0;//强制将区间全改为0或1 所以之前的异或标记可以忽略 
		segtree[root].lazyAO[pos]=-1;
	}
	if(segtree[root].lazyXOR[pos]){//上方的if语句已经将 lazyXOR清空 此分支为未进行与或操作只需要进行异或操作 
		segtree[root<<1].digit[pos]=(segtree[root<<1].r-segtree[root<<1].l+1)-segtree[root<<1].digit[pos];
		segtree[root<<1|1].digit[pos]=(segtree[root<<1|1].r-segtree[root<<1|1].l+1)-segtree[root<<1|1].digit[pos];
		
		if(segtree[root<<1].lazyAO[pos]!=-1)segtree[root<<1].lazyAO[pos]^=1;//如果被强制置1或0直接翻转  全0变全1 全1变全0 
		else segtree[root<<1].lazyXOR[pos]^=1;//否则异或 
		if(segtree[root<<1|1].lazyAO[pos]!=-1)segtree[root<<1].lazyAO[pos]^=1;//如果被强制置1或0直接翻转  全0变全1 全1变全0 
		else segtree[root<<1|1].lazyXOR[pos]^=1;//否则异或 
		
		segtree[root].lazyXOR[pos]=0;
	}
}	

void build(int root,int l,int r){
	segtree[root].l=l;segtree[root].r=r;

	memset(segtree[root].lazyXOR,0,sizeof(segtree[root].lazyXOR));
	memset(segtree[root].lazyAO,-1,sizeof(segtree[root].lazyAO));
	
	if(l==r){
		for(int i=0;i<M;i++){
			if(num[l]&(1LL<<i))++segtree[root].digit[i];
		}
		return ;
	}
	int mid=l+r>>1;
	build(root<<1,l,mid);
	build(root<<1|1,mid+1,r);
	
	for(int i=0;i<M;i++)
		pushup(root,i);
}

void update(int root,int l,int r,int pos,int val){
	if(l>segtree[root].r||r<segtree[root].l)return ;    //val只有三种值：0表示当前二进制位为0且进行与操作 
	if(l<=segtree[root].l&&segtree[root].r<=r){			//1表示当前二进制为为1且进行或操作 
		if(val==-1){//异或操作                           -1表示当前二进制位为1且惊醒异或操作  其他操作均不改变二进制位 
			segtree[root].digit[pos]=(segtree[root].r-segtree[root].l+1)-segtree[root].digit[pos];
			if(segtree[root].lazyAO[pos]!=-1)segtree[root].lazyAO[pos]^=1;
			else segtree[root].lazyXOR[pos]^=1;
		}
		else {
			segtree[root].digit[pos]=(segtree[root].r-segtree[root].l+1)*val;
			segtree[root].lazyAO[pos]=val;//强制置1或0 
			segtree[root].lazyXOR[pos]=0;//强制置1或0 
		}
		return ;
	}
	
	pushdown(root,pos);
 
	update(root<<1,l,r,pos,val);
	update(root<<1|1,l,r,pos,val);
	
	pushup(root,pos);
}

long long query(int root,int l,int r,int pos){
	if(l>segtree[root].r||r<segtree[root].l)return 0;
	if(l<=segtree[root].l&&segtree[root].r<=r)return segtree[root].digit[pos];
	
	pushdown(root,pos);
	
	ll ans=0;
	ans+=query(root<<1,l,r,pos);
	ans+=query(root<<1|1,l,r,pos);
	return ans;
}

int main(){
	int n,m;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%lld",&num[i]);
	
	build(1,1,n);
	
	scanf("%d",&m);
	for(int i=1;i<=m;i++){
		int t;
		scanf("%d",&t);
		if(t==1){//求和操作 
			int l,r;
			scanf("%d%d",&l,&r);
			ll ans=0;
			for(int i=0;i<M;i++)ans+=(query(1,l,r,i)*(1LL<<i));
			printf("%lld\n",ans);
		}
		else if(t==2){//异或操作 
			int l,r,k;
			scanf("%d%d%d",&l,&r,&k);
			for(int i=0;i<M;i++){
				if(k&(1LL<<i))update(1,l,r,i,-1);
			}
		}
		else if(t==3){//或操作 
			int l,r,k;
			scanf("%d%d%d",&l,&r,&k);
			for(int i=0;i<M;i++){
				if(k&(1LL<<i))update(1,l,r,i,1);
			}
		}
		else {//与操作 
			int l,r,k;
			scanf("%d%d%d",&l,&r,&k);
			for(int i=0;i<M;i++){
				if((k&(1LL<<i))==0)update(1,l,r,i,0);
			}
		}
	}
	return 0;
}

咆哮_

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
异或线段树模板

https://ac.nowcoder.com/acm/contest/275/Chttp://codeforces.com/problemset/problem/242/E每个二进制位独占一棵线段树，很巧妙题意：给定一个长度为 n 的序列，有 m 次操作：操作有2种：1. 区间求和，即输入l,r，输出.2. 区间异或，即输入l,r,k，对于，将 a[i] 变为....
复制链接

扫一扫