Sum

链接:https://ac.nowcoder.com/acm/problem/14269
来源:牛客网

时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 131072K,其他语言262144K
64bit IO Format: %lld
题目描述
考虑维护一个这样的问题:
(1) 给出一个数组A,标号为1~n
(2) 修改数组中的一个位置。
(3) 询问区间[l,r]中所有子集的位运算and之和mod(109+7)。
位运算and即为“pascal中的and”和“C/C++中的&”
我们定义集合S={ l , l+1 , … , r-1 , r}
若集合T,T ∩ S = T,则称T为S的子集
设f(T)=AT1 and AT2 and … and ATk (设k为T集大小,若k=0则f(T)=0)
所有子集的位运算and之和即为∑f(T)
那么,现在问题来了。
输入描述:
第一行,一个正整数N
第二行,N个非负整数,为数组A
第三行,一个正整数M,为操作次数
接下来M行格式如下
修改操作: 1 x y,将Ax修改为y
询问操作: 2 l r,区间[l,r]中所有子集的位运算and之和 mod(109+7)
输出描述:
对于每次询问输出一行,为该次询问的答案mod(109+7)。
long long 请使用lld
示例1
输入
复制
3
1 2 3
6
2 1 3
1 1 2
2 1 3
2 2 3
1 2 5
2 1 3
输出
复制
9
15
7
13
说明
第一次询问:
Answer =1+2+3+(1 and 2)+(1 and 3)+(2 and 3)+(1 and 2 and 3)
=1+2+3+0+1+2+0
=9
第二次询问:
Answer =2+2+3+(2 and 2)+(2 and 3)+(2 and 3)+(2 and 2 and 3)
=2+2+3+2+2+2+2
=15
第三次询问:
Answer =2+3+(2 and 3)
=2+3+2
=7
第四次询问:
Answer =2+5+3+(2 and 5)+(2 and 3)+(3 and 5)+(2 and 5 and 3)
=2+5+3+0+2+1+0
=13
备注:
M,N≤105,Ai≤109

题解:

这种位运算多半都与二进制有关系
我们把每个数进行拆分,然后统计每一位上每个数贡献了多少1
很多题解都提到式子:
(2x-1) * 2p
x是这一位为1的数的个数
p:是指第p位
怎么来的呢?
2x-1
其实就是比如说有x个数字某一位都有1
答案=x+ (x中任取两个取对)+(x中任取三个)+(x中任取x-1个)=2x-1
(高中数学都学过)
线段树和树状数组都可以做

代码:

该为本人写的代码。。为错误代码,但我没找到原因

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+9;
typedef long long ll;
ll mod=1e9+7;
struct node{
	int a[40];
}p[maxn<<2];
ll b[maxn];
void init()
{
	b[0]=1;
	for(int i=1;i<=32;i++)b[i]=b[0]<<i%mod;
}
void build(int l,int r,int rt)
{
	if(l==r)
	{
		ll x;
		cin>>x;
		for(int i=0;i<=32;i++)
		{
			p[rt].a[i]=x%2;
			x/=2;
		}
		return ;
	}
	int mid=(l+r)>>1;
	build(l,mid,rt<<1);
	build(mid+1,r,rt<<1+1);
	for(int i=0;i<=32;i++)
	{
		p[rt].a[i]=p[rt<<1].a[i]+p[rt<<1+1].a[i];
		//左右子树的第i位相加 
	}
	return ;
}
void update(int pos,int v,int l,int r,int rt)//将第pos位置的数改成v 
{
	if(l==r)
	{
		for(int i=0;i<=32;i++)
		{
			p[rt].a[i]=v%2;
			v/=2;
		}
		return ;

	}
	int mid=(l+r)>>1;
	if(pos<=mid) update(pos,v,l,mid,rt*2);
	else update(pos,v,mid+1,r,rt*2+1);
	for(int i=0;i<32;i++)
	{
		p[rt].a[i]=p[rt*2].a[i]+p[rt*2+1].a[i];
	}
	return ;
}
node query(int L,int R,int l,int r,int rt)
{
	if(L>=l&&R<=r) return p[rt];
	int mid=(l+r)>>1;
	node p1,p2,p3;
//	memset(p1.a,0,sizeof(p1.a));
//    memset(p2.a,0,sizeof(p2.a));
	if(L<=mid)p1=query(L,R,l,mid,rt<<1);
	if(R>mid)p2=query(L,R,mid+1,r,rt<<1+1);
	for(int i=0;i<=32;i++)p3.a[i]=p1.a[i]+p2.a[i];
	return p3;
	
}
int main()
{
	ios::sync_with_stdio(false);
	init();
	int n;
	cin>>n;
	build(1,n,1);//root 
	int q;
	cin>>q;
	while(q--)
	{
		int x,l,r;
		cin>>x>>l>>r;
		if(x==1)update(l,r,1,n,1);
		else 
		{
			node res=query(l,r,1,n,1);
			ll sum=0;
			for(int i=0;i<=32;i++)
			{
				sum=(sum+b[i]*(b[res.a[i]-1])%mod)%mod;
			}
			cout<<sum<<endl;
		}
	}
	return 0;
}

树状数组做法:

#include<stdio.h>
#include<algorithm>
using namespace std;
#define maxn 100005
#define mod 1000000007
#define lowbit(x) x&(-x)
typedef long long ll;
int n,m,a[maxn],p[maxn],c[35][maxn];
void add(int k,int x,int val)
{
	while(x<=n)
	{
		c[k][x]+=val;
		x+=lowbit(x);
	}
}
int sum(int k,int x)
{
	int res=0;
	while(x)
	{
		res+=c[k][x];
		x-=lowbit(x);
	}
	return res;
}
int main(void)
{
	p[0]=1;
	int i,j,t,x,y;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
		p[i]=1ll*2*p[i-1]%mod;
	for(i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		for(j=0;j<=30;j++)
			if(a[i]&(1<<j))//如果第j位是1的话 
				add(j,i,1);//在该位上添加1 
	}
	scanf("%d",&m);
	while(m--)
	{
		scanf("%d%d%d",&t,&x,&y);
		if(t==1)
		{
			for(i=0;i<=30;i++)
				if(a[x]&(1<<i))
					add(i,x,-1);
			a[x]=y;
			for(i=0;i<=30;i++)
				if(a[x]&(1<<i))
					add(i,x,1);
		}
		else
		{
			ll ans=0;
			for(i=0;i<=30;i++)
			{
				int tmp=sum(i,y)-sum(i,x-1);//查询x到y区间内在第i位上1的个数 
				ans=(ans+1ll*(p[tmp]-1)*p[i]%mod)%mod;
			}
			printf("%d\n",ans);
		}
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值