AT_diverta2019_e XOR Partitioning

先发现以下两个性质:

  • 对于所有段,它们的异或和就是所有数的异或和。
  • 由于每段的异或和都一样,设为 \(d\),并设 \(s\) 为所有数的异或和,总共分成了 \(x\) 段,则有 \(s=[x\bmod2=1]d\)

\(\left\{A\right\}\) 的前缀异或和为 \(\left\{pre\right\}\)

分类讨论。

如果 \(s\ne0\),这个比较清晰,说明 \(d=s\)\(x\) 是奇数,考虑所有 \(x\) 段的最后一个数的位置 \(p_1,\cdots,p_x\),那么说明 \(pre_{p_i}=[i\bmod2=1]d\),因此问题变成在 \(1,\cdots,n\) 选奇数个位置,且 \(n\) 必须选,满足奇数位置上的 \(pre=d\),偶数位置上的 \(pre=0\)

考虑 DP,设 \(f(i,0/1)\) 表示前 \(i\) 个数,选了偶数/奇数个位置的方案数。则有

\[f(i,0)=\left\{\begin{matrix} f(i-1,0)& a_i\ne0 \\f(i-1,0)+f(i-1,1)& a_i=0 \end{matrix}\right.\]

\[f(i,1)=\left\{\begin{matrix} f(i-1,1)& a_i\ne d \\ f(i-1,1)+f(i-1,0)& a_i=d \end{matrix}\right.\]

否则若 \(s=0\),继续分类讨论,若分成的段数是 \(x\) 是奇数,则说明 \(d=0\),问题同上,否则 \(d\) 未知,问题变成在 \(1,\cdots,n\) 中选偶数个位置,且 \(n\) 必须选,满足奇数位置上的 \(pre=d\),偶数位置上的 \(pre=0\)

如果对每个 \(d\)\(\mathcal O(n)\) 跑一遍 DP,将会爆炸,但是注意到转移只会在 \(pre=0\)\(pre=d\) 的位置更新 \(f\),于是预处理出所有 \(pre_i=d\) 的位置,然后对 \(pre_i=0\) 的位置进行前缀和预处理,就可以快速转移了。

时间复杂度 \(\mathcal O(n)\)

Code:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 500005, M = 1 << 20, mod = 1e9 + 7;
int n;
int pre[N], sum[N];
vector <int> pos[M];
bool vis[M];

void add(int &a, int b) {
	a += b;
	if (a >= mod) a -= mod;
}

int calc(int d) {
	if (!d) {
		int res = 1;
		for (int i = 0; i + 1 < pos[0].size(); ++i) res = res * 2 % mod;
		return res;
	}
	int odd = 0, even = 1, lst = 0;
	for (int i = 0; i + (pre[n] == d) < pos[d].size(); ++i) {
		add(even, 1ll * odd * (sum[pos[d][i]] - sum[lst]) % mod), add(odd, even);
		lst = pos[d][i];
	}
	add(even, 1ll * odd * (sum[n - 1] - sum[lst]) % mod);
	if (pre[n] == d) return even;
	else return odd;
}

int main() {
	scanf("%d", &n);
	for (int i = 1, x; i <= n; ++i) scanf("%d", &x), pos[pre[i] = pre[i - 1] ^ x].push_back(i), sum[i] = sum[i - 1] + (pre[i] == 0);
	if (pre[n]) return printf("%d", calc(pre[n])), 0;
	else {
		int ans = 0;
		for (int i = 1; i <= n; ++i) {
			if (vis[pre[i]]) continue;
			vis[pre[i]] = 1, add(ans, calc(pre[i]));
		}
		printf("%d", ans);
	}
	return 0;
}
```AT_diverta2019_e
posted @ 2022-11-18 20:00  Kobe303  阅读(37)  评论(0编辑  收藏  举报