ARC137D Prefix XORs

ARC137D Prefix XORs

洛谷ARC137D Prefix XORs

题目大意

给你一个长度为$n$的序列$A$和一个整数$m$。

对$k=1,2,\dots .m$，求经过如下$k$次操作后$A_n$的值

• 对每个$i(1\le i\le n)$，同时让$A_i=A_1\oplus A_2\oplus \cdots \oplus A_i$

$1\le n\le 10^6,1\le m\le 10^6,1\le A_i\le 2^{30}$

题解

如果将异或改为加法，则对数组做$k$次前缀和，最终$a_n$的值为

$$\sum _{i=1}^n{C}_{k-1+i}^i\cdot a_{n-i}$$

因为本题是异或，所以就按$C_{k-1+i}^i$的奇偶性来判断贡献。若$C_{k-1+i}^i\mathrm{mod}2=1$，则有贡献；否则没有贡献。

那怎么知道$C_n^m$的奇偶性呢？用lucas定理，对于$n$和$m$的每一个二进制位，如果$n$和$m$都为$0$或$1$，或者$n$的这一位为$1$、$m$的这一位为$0$，则$C_n^m$为奇数。只要有一位不满足上述条件，则$C_n^m$为偶数。

我们发现，当在二进制中$m$为$n$的子集时，$C_n^m$为奇数。也就是说，$a_{n-i}$有贡献，要满足$i\subset k-1+i$时有贡献。那么，$k-1\subset {\complement }_{U}i$，即$i\subset {\complement }_{U}k-1$。

综上所述，当$i\subset {\complement }_{U}k-1$时，$a_{n-i}$对$k$次操作后的$a_n$有贡献。

我们可以先倒着输入，即$a_{n-i}=a_i$。再对$a$数组求子集和。令$v_i$表示对$a$求$i$的子集和，则对于每一个$k$，令$j={\complement }_{U}k-1$，则答案为$v_j$。

时间复杂度为$O(n\log n)$。

code

#include<bits/stdc++.h>
using namespace std;
int n,m,a[(1<<20)+5];
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&a[n-i]);
	for(int j=0;j<=19;j++){
		for(int i=0;i<(1<<20);i++){
			if(i&(1<<j)) a[i]^=a[i^(1<<j)];
		}
	}
	for(int i=0;i<m;i++){
		printf("%d ",a[i^((1<<20)-1)]);
	}
	return 0;
}