[TJOI2017]异或和
对于这种异或类的题目,我们可以考虑从异或运算性质下手
我们记\(sum[i] = \sum_{j = 1} ^ {i}a[j]\)
不难发现,如果我们对每一位分开考虑,若我们在求第x为的答案,记所有区间的连续的和有K个该位为1,那么跟据异或的运算法则,这一位对答案有贡献当且仅当K为奇数,且对答案的贡献为\(K\ \%\ 2 * 2 ^ x\)
说得更具体点,我们要求的式子变成了:
我们单独看里面的柿子:
其实问题已经转化为:给定\(sum[i]\),求有多少个\(j(j\in[0,\ i - 1])\),满足\(sum[i] - sum[j]\)的第K位为1
这个问题看上去很好处理,貌似是只要求出有多少个\(sum[i]\)与\(sum[j]\)的第K位不同即可,我们只需要记录\(sum[i] >> k\)的一个前缀和即可。
但是这道题目十分弟弟,因为异或的减法是可以进位的,所以我们并不能直接向刚刚那么算。
为了方便,我们定义\(sum[i]_k\)表示\(sum[i]\)的第K位,\(sum[i]_{1\ -\ k}\)表示sum的1~k位 ~~(原谅我''打不出来)~ 的和
我们先进行分类讨论:
一、如果\(sum[i]_k = 1\),那么能对答案产生贡献只有两种情况:
\(1.sum[j]_k = 0\ \&\&\ sum[j]_{1\ - \ k - 1} <= sum[i]_{1 - k - 1}\)
\(2.sum[j]_k = 1\ \&\&\ sum[j]_{1\ -\ k - 1} > sum[i]_{1\ -\ k - 1}\)
二、如果\(sum[i]_k = 0\),那么能对答案产生贡献也只有两种情况:
\(1.sum[j]_k = 1\ \&\&\ sum[j]_{1\ - \ k - 1} <= sum[i]_{1 - k - 1}\)
\(2.sum[j]_k = 0\ \&\&\ sum[j]_{1\ -\ k - 1} > sum[i]_{1\ -\ k - 1}\)
那么现在问题来了,我们怎么判断进位呢?
注意到\(\sum a[i]\)不会太大
我们可以把\(sum[i]_{1\ -\ k - 1}\)扔到两颗权值树状数组内,一棵存0的,另一颗存1的
注意:树状数组的下标可能为0,所以我们要把所有下标+1
\(Code:\)
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdlib>
using namespace std;
#define il inline
#define re register
#define int long long
il int read() {
re int x = 0, f = 1; re char c = getchar();
while(c < '0' || c > '9') { if(c == '-') f = -1; c = getchar();}
while(c >= '0' && c <= '9') x = x * 10 + c - 48, c = getchar();
return x * f;
}
#define rep(i, s, t) for(re int i = s; i <= t; ++ i)
#define mem(k, p) memset(k, p, sizeof(k))
#define lb(x) (x)&(-(x))
#define maxn 1000005
int n, m, a[maxn], sum[maxn], ma, Ans, ans;
struct Tree {
int t[maxn];
il void add(int u, int v, int x) {
while(u <= x) t[u] += v, u += lb(u);
}
il void Add(int u, int v, int x) {
add(u + 1, v, x + 1);
}
il int query(int u) {
int ans = 0;
while(u) ans += t[u], u -= lb(u);
return ans;
}
il int query(int l, int r) {
return query(r + 1) - query(l);
}
}A, B;//A表示0的树状数组,B表示1的
signed main() {
n = read();
rep(i, 1, n) a[i] = read(), sum[i] = sum[i - 1] + a[i], ma = max(ma, sum[i]);
for(re int k = 0; (1 << k) <= ma; ++ k) {
memset(A.t, 0, sizeof(A.t)), memset(B.t, 0, sizeof(B.t)), Ans = 0;
int pax = (1 << k) - 1;//我们只考虑1~k位
rep(i, 0, n) {
int temp = sum[i] & pax;//求出sum[i]_{1-k}
//cout << k << ' ' << temp << endl;
if((sum[i] >> k) & 1) {
Ans += A.query(0, temp) + B.query(temp + 1, pax), B.Add(temp, 1, pax);
}
else {
Ans += B.query(0, temp) + A.query(temp + 1, pax), A.Add(temp, 1, pax);
}
}
//cout << Ans << ' ' << ans << endl;
ans += (Ans % 2) * (1 << k);
}
printf("%lld", ans);
return 0;
}