BZOJ1273 [BeiJingWc2008]序列

由于是全局加，所以对于所有的加操作我们可以用一个变量X记录下来

对于x和X，如果(x+X)&2^i==1的话就说明x+X二进制展开之后的从第i-1位开始的后缀在2^i到2^(i+1)-1之间，也就是x在2^i-X到2^(i+1)-1-X之间（模2^(i+1)意义下）所以对于所有长度的后缀分别开一个权值线段树，然后查询就可以了

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<bitset>
using namespace std;
#define MAXN 100010
#define MAXM 1010
#define ll long long
#define INF 1000000000
#define MOD 1000000007
#define eps 1e-8
int cnt[20];
int a[MAXN];
int n,m;
int N=(1<<16)-1;
int X;
int v[16][MAXN<<2];
ll ans;
void change(int w,int x,int y,int z,int p){
	v[w][x]++;
	if(y==z){
		return ;
	}
	int mid=y+z>>1;
	if(p<=mid){
		change(w,x<<1,y,mid,p);
	}else{
		change(w,x<<1|1,mid+1,z,p);
	}
}
int ask(int w,int x,int y,int z,int l,int r){
	if(y==l&&z==r){
		return v[w][x];
	}
	int mid=y+z>>1;
	if(r<=mid){
		return ask(w,x<<1,y,mid,l,r);
	}else if(l>mid){
		return ask(w,x<<1|1,mid+1,z,l,r);
	}else{
		return ask(w,x<<1,y,mid,l,mid)+ask(w,x<<1|1,mid+1,z,mid+1,r);
	}
}
int main(){
	int i,x;
	char o[10];
	scanf("%d%d",&n,&m);
	while(n--){
		scanf("%d",&x);
		int t=0;
		for(i=0;i<16;i++){
			t+=x&(1<<i);
			change(i,1,0,N,t);
		}
	}
	while(m--){
		scanf("%s",o);
		if(o[0]=='A'){
			scanf("%d",&x);
			X+=x;
		}
		if(o[0]=='Q'){
			scanf("%d",&x);
			if(x>=16){
				printf("0\n");
				continue ;
			}
			int l=1<<x;
			int r=(1<<(x+1))-1;
			int mod=1<<(x+1);
			l=(l+mod-X%mod)%mod;
			r=(r+mod-X%mod)%mod;
			if(l<=r){
				ans+=ask(x,1,0,N,l,r);
				//printf("%d\n",ask(x,1,0,N,l,r));
			}else{
				ans+=ask(x,1,0,N,l,N)+ask(x,1,0,N,0,r);
				//printf("%d\n",ask(x,1,0,N,l,N)+ask(x,1,0,N,0,r));
			}
		}
	}
	printf("%lld\n",ans);
	return 0;
}

/*
3 5
1 2 4
Q 1
Q 2
A 1
Q 1
Q 2

*/