【CF1097F Alex and a TV Show】

洛谷

题目描述

维护n个初始为空的可重集，支持以下操作：

1 x v:令集合x等于{v}

2 x y z:令集合x等于集合y与z的并

3 x y z:令集合x等于集合y与z的积， A ∗ B = { g c d ( a , b ) ∣ a ∈ A , b ∈ B } A∗B=\{gcd(a,b)∣a∈A,b∈B\} A∗B={gcd(a,b)∣a∈A,b∈B}

4 x v:询问v在集合x中出现次数模2的结果

Sol

个人感觉是一道非常好的题目。

题目中说了值域只有 7000 ,而且答案居然是对 2 取模??
$bitset$ 跑不了了

顿时感觉是道简单题 :
1操作直接A[x].set(v);
2操作直接A[x]|=A[y];
4操作直接cout<<A[x][v];

3操作 …???什么鬼,这个是个啥玩意,好像做不了了,这个要怎么变换啊
显然先把这个玩意写成一个多项式 , 不然式子都推不下去了,设$A(x)$表示集合的多项式,其系数$A_i$表示元素$i$的出现次数模2的结果
那么推一波式子(下面求和的上界都懒得写了,多了也没有贡献):
$$A(x)\ast B(x)=\sum _{i=1}\sum _{j=1}{A}_i\ast B_j\ast x^{gcd(i,j)}$$
常规操作枚举$gcd$:
$$A(x)\ast B(x)=\sum _{d=1}\sum _{i=1}\sum _{j=1}{A}_{id}\ast B_{jd}\ast x^d[gcd(i,j)=1]$$
这怕不是要莫比乌斯反演的节奏:
$$A(x)\ast B(x)=\sum _{d=1}\sum _{i=1}\sum _{j=1}{A}_{id}\ast B_{jd}\ast x^d\sum _{k|gcd(i,j)}\mu (k)$$
交换求和顺序:
$$A(x)\ast B(x)=\sum _{d=1}{x}^d\sum _{i=1}\sum _{j=1}\sum _{k|gcd(i,j)}\mu (k)\ast A_{id}\ast B_{jd}$$
好像我写的比较繁琐:
$$A(x)\ast B(x)=\sum _{d=1}{x}^d\sum _{k=1}\mu (k)\sum _{i=1}\sum _{j=1}{A}_{idk}\ast B_{jdk}$$
常规$T=dk$:

$$A(x)\ast B(x)=\sum _{d=1}{x}^d\sum _{d|T}\mu (\frac{T}{d})\sum _{i=1}{A}_{Ti}\sum _{j=1}{B}_{Tj}$$

后面这个东西好像很妙 , 就是莫比乌斯反演的式子,那么结果是啥咧?
首先容易看出后面是一个多项式莫比乌斯变换后的结果,即:
$$A_i=\sum _{i|d}{A}_d$$
也就是倍数前缀和 , 记莫比乌斯变换后的结果为:$FMT(A)$ , 其第$i$次方项系数为$FMT(A{)}_i$,后面就是多项式点值相乘的结果! , 然后$IFMT$,也就是莫比乌斯反演还原多项式:

$$A(x)\ast B(x)=\sum _{d=1}{x}^d\sum _{T=1}\mu (\frac{T}{d})FMT(A{)}_{T}FMT(B{)}_T$$

两边同时做$FMT$:

$$FMT(A(x)\ast B(x))=FMT(A)\ast FMT(B)$$
那么$FMT$后就可以多项式按位相乘了

显然 $FMT$ 后的加法还是能直接加,乘法的话稍微想想就知道是按位与的操作。完美解决所有操作…
慢着 , 发现直接这样是没有办法询问的 , 必须要进行逆莫比乌斯变换,也就是莫比乌斯反演还原多项式,但是暴力还原是要枚举倍数的,没有办法保证复杂度。

问题就只有求解:

$$A(x)=IFMT(A)=\sum _{i=1}{x}^i\sum _{i|T}\mu (\frac{T}{i})FMT(A{)}_T$$

其实我们没有必要真的还原,因为它只是询问一个数的出现次数又不是所有数 , 说明白点就是我们只需要还原后多项式第 $i$ 次方项的系数 , $i$ 是询问的数

那么就是要管后面那一坨东西了,对于一个 $i$ 而言,怎么快速求出后面的东西?
发现这玩意也很像多项式系数对应相乘 , 但是要求和 , 那不就完了么 , 我们直接预处理出每一个 $i$ 对应的 $\mu$ 的 $bitset$ , 就是把 $i$的所有倍数的第 $i$ 项置为$\mu (\frac{T}{i})$ , 然后查询的时候按位乘上就是与一下 , 最后$count()$一下模个2就是答案了。

代码很短的:

#include<bits/stdc++.h>
using namespace std;
template<class T>inline void init(T&x){
	x=0;char ch=getchar();bool t=0;
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') t=1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=(x<<1)+(x<<3)+(ch-48);
	if(t) x=-x;
}
const int MAXN=7001;
const int N=1e5+10;
const int MAXQ=1e6+10;
typedef bitset<MAXN> Bit;
Bit A[N];
Bit val[MAXN];
int mu[MAXN];
Bit Mu[MAXN];
inline void Sieve(){
	for(int i=mu[1]=1,j;i<=7000;++i) for(val[i].set(i),j=i+i;j<=7000;j+=i) mu[j]^=mu[i],val[j].set(i);
	for(int i=1;i<=7000;++i)for(int j=i,k=1;j<=7000;j+=i,++k)if(mu[k]) Mu[i].set(j);
}
int n,m;

int main()
{
	init(n),init(m);Sieve();
	for(int i=1;i<=m;++i) {
		int opt,x,y,z;
		init(opt);
		if(opt==1) {init(x),init(y);A[x]=val[y];}
		else if(opt==2) {init(x),init(y),init(z);A[x]=A[y]^A[z];}
		else if(opt==3) {init(x),init(y),init(z);A[x]=A[y]&A[z];}
		else if(opt==4) {
			init(x),init(y);
			Bit ans=A[x];ans&=Mu[y];
			printf("%d",ans.count()&1);
		}
	}
	return 0;
}