【BZOJ4888】[TJOI2017]异或和（树状数组）

【BZOJ4888】[TJOI2017]异或和（树状数组）

题面

BZOJ
洛谷

题解

考虑每个位置上的答案，分类讨论这一位是否存在一，值域树状数组维护即可。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
#define ll long long
#define MAX 100100
inline int read()
{
	int x=0;bool t=false;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=true,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return t?-x:x;
}
int n,m,a[MAX],c[MAX*10];ll ans;
int lb(int x){return x&(-x);}
void add(int x,int w){x+=1;while(x<=m)c[x]+=w,x+=lb(x);}
int getsum(int x){int ret=0;while(x)ret+=c[x],x-=lb(x);return ret;}
int Query(int l,int r)
{
	if(l>r)return 0;
	return getsum(min(r+1,m))-getsum(min(l,m));
}
int main()
{
	n=read();
	for(int i=1;i<=n;++i)a[i]=read()+a[i-1];
	m=a[n];add(0,1);
	for(int j=0;j<=23;++j)
	{
		int P=1<<j,sum=0;
		for(int i=1;i<=n;++i)
		{
			if(a[i]&P)
				sum+=Query(0,a[i]%P)+Query(P+a[i]%P+1,P+P-1);
			else
				sum+=Query(a[i]%P+1,P-1)+Query(P,P+a[i]%P);
			add(a[i]%(P+P),1);
		}
		for(int i=1;i<=n;++i)add(a[i]%(P+P),-1);
		if(sum&1)ans^=P;
	}
	printf("%lld\n",ans);
	return 0;
}