题解ARC137D

Posted 2022-03-20 7KByte 的博客

有意思的找规律题。我们先简单写一个程序打表，打印一个01方阵 \\(u_i,j\\) 表示经过 \\(i\\) 次操作后，\\(a_n\\) 是否要异或 \\(a_j\\)。

#define N 100005
int n, a[N];
int main() 
	read(n);
	a[n] = 1;
	int m = 1; while(m < n) m <<= 1;
	rp(i, m)
		rp(j, n)if(a[j])rp(k, j - 1)a[k] ^= 1;
		rp(j, n)printf("%d ", a[j]); el;
	
	return 0;

当 \\(n=16\\) 时打的表如下。

1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
0 1 0 1 0 1 0 1 0 1 0 1 0 1 0 1
0 0 1 1 0 0 1 1 0 0 1 1 0 0 1 1
0 0 0 1 0 0 0 1 0 0 0 1 0 0 0 1
0 0 0 0 1 1 1 1 0 0 0 0 1 1 1 1
0 0 0 0 0 1 0 1 0 0 0 0 0 1 0 1
0 0 0 0 0 0 1 1 0 0 0 0 0 0 1 1
0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 1
0 0 0 0 0 0 0 0 1 1 1 1 1 1 1 1
0 0 0 0 0 0 0 0 0 1 0 1 0 1 0 1
0 0 0 0 0 0 0 0 0 0 1 1 0 0 1 1
0 0 0 0 0 0 0 0 0 0 0 1 0 0 0 1
0 0 0 0 0 0 0 0 0 0 0 0 1 1 1 1
0 0 0 0 0 0 0 0 0 0 0 0 0 1 0 1
0 0 0 0 0 0 0 0 0 0 0 0 0 0 1 1
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1

想必答案已经非常清楚了，\\(u_i,j = 1\\) 当且仅当 \\(i\\ \\&\\ j = i\\)。所以对于每个 \\(a_i\\)，当且仅当 \\(k\\ \\&\\ i = k\\) 时对 \\(ans_k\\) 产生贡献。所以这本质上是高维前缀和/子集卷积/FWT，直接做即可。

#define N 1000005
int n, m, w = 1, a[N], u[N];
void fwt()
	for(int l = 2, k = 1; l <= w; l <<= 1, k <<= 1)
		for(int i = 0; i < w; i += l)rep(j, 0, k - 1)
			u[i + j] ^= u[i + j + k];

int main() 
	read(n, m); rp(i, n)read(a[i]);
	while(w < n) w <<= 1;
	rp(i, n)u[w - n + i - 1] = a[i];
	fwt();
	rp(i, m)printf("%d ", u[(i - 1) % w]);
	return 0;