题目描述
给出N个[0, 65535]的整数,编程支持以下操作:
修改操作:C d,所有数增加d。如果超过65535,把结果模65536。0 <= d <= 65535
查询操作:Q i,统计有多少整数的第i位非0,换句话说,有多少个整数与2i的“按位与”操作值为正。0 <= i <= 15
输出所有查询操作的统计值。
数据范围
样例输入
3 5
1 2 4
Q 1
Q 2
C 1
Q 1
Q 2
样例输出
1
1
2
1
解题思路
若x的第i位(从第0位开始)为1,则
如果x+d的第i位为1,则
所以,按位建16个线段树就行了无力吐槽左移右移的优先级
代码
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <cmath>
#define Maxn 100005
using namespace std;
inline int Getint(){int x=0,f=1;char ch=getchar();while('0'>ch||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while('0'<=ch&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}
inline char Getch(){char ch=getchar();while(ch!='Q'&&ch!='C')ch=getchar();return ch;}//开始我并没有写return ch;在本地对拍都是对的,交OJ就WA。。。
struct node{int L,r,sum=0;}Tree[131072][18];
void Build(int x,int v,int L,int r){
Tree[v][x].L=L;
Tree[v][x].r=r;
Tree[v][x].sum=0;
if(L==r)return;
Build(x,2*v,L,(L+r)/2);
Build(x,2*v+1,(L+r)/2+1,r);
}
void Insert(int x,int v,int vll){
if(vll<Tree[v][x].L||Tree[v][x].r<vll)return;
Tree[v][x].sum++;
if(Tree[v][x].L==Tree[v][x].r)return;
Insert(x,2*v,vll);
Insert(x,2*v+1,vll);
}
int Ask(int x,int v,int L,int r){
if(r<Tree[v][x].L||Tree[v][x].r<L)return 0;
if(L<=Tree[v][x].L&&Tree[v][x].r<=r)return Tree[v][x].sum;
return Ask(x,2*v,L,r)+Ask(x,2*v+1,L,r);
}
int main(){
int n=Getint(),q=Getint(),vl[Maxn];
for(int i=1;i<=n;i++)vl[i]=Getint();
for(int i=0;i<=16;i++)Build(i,1,0,(1<<(i+1)-1));
for(int i=1;i<=16;i++)
for(int j=1;j<=n;j++)
Insert(i,1,vl[j]%(1<<i));
int d=0;
while(q--){
char ch=Getch();
if(ch=='C')d=(d+Getint())%65536;
else{
int x=Getint(),MOD=1<<(x+1),L=(1<<x)-d,r=MOD-d-1;
L=(L%MOD+MOD)%MOD;
r=(r%MOD+MOD)%MOD;
if(r<L)cout<<Ask(x+1,1,L,MOD-1)+Ask(x+1,1,0,r)<<"\n";
else cout<<Ask(x+1,1,L,r)<<"\n";
}
}
return 0;
}