codeforces 547 C. Mike and Foam（反演）

题目链接

题目描述：
给出一列数，最开始集合为空，q次操作，每次给出一个x，如果第x个数存在则删去，不存在则添加，问操作完互质的数有多少对

分析：
设$f(d)$表示$gcd(a_i,a_j)=d$的数对$(i,j)$个数
设$F(d)$表示$d|gcd(a_i,a_j)$的数对$(i,j)$个数
显然$F(d)=\sum_{d|n}f(n),f(n)=\sum_{d|n}μ(d)*F(d)$
设含约数$d$的数字个数为$cnt(d)$
显然$F(d)={cnt(d)*(cnt(d)-1) \over 2}$
$f(1)=\sum_{d=1}^{max(a_i)}μ(d)*F(d)$

动态维护$cnt$

tip

感觉这个不像是正宗反演
实际上正宗反演是这样的：
$F(n)=\sum_{d|n}f(d),f(n)=\sum_{d|n}μ(d)F({n \over d})$

一开始我把数字都质因数分解了，这样就不符合$cnt$函数的定义了，所以直接根号枚举约数即可

#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long

using namespace std;

const int M=500005;
const int N=200005;
int n,m,a[N],sshu[M],tot;
int cnt[M],mu[M];
bool no[M],mark[N];
ll ans=0;

void prepare() {
    mu[1]=1;
    for (int i=2;i<M;i++) {
        if (!no[i]) {
            sshu[++tot]=i;
            mu[i]=-1;
        }
        for (int j=1;j<=tot&&sshu[j]*i<M;j++) {
            no[sshu[j]*i]=1;
            if (i%sshu[j]==0) {
                mu[sshu[j]*i]=0;
                break;
            }
            mu[sshu[j]*i]=-mu[i];
        }
    }
}

void change(int x,int z) {
    for (int i=1;i*i<=x;i++) 
        if (x%i==0) {
            ans-=(ll)mu[i]*cnt[i]*(cnt[i]-1)/2;
            cnt[i]+=z;
            ans+=(ll)mu[i]*cnt[i]*(cnt[i]-1)/2;
            if (x/i!=i) {
                int j=x/i;
                ans-=(ll)mu[j]*cnt[j]*(cnt[j]-1)/2;
                cnt[j]+=z;
                ans+=(ll)mu[j]*cnt[j]*(cnt[j]-1)/2;
            }
        }
}

int main()
{
    prepare();
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]);
    while (m--) {
        int x;
        scanf("%d",&x);
        if (mark[x]) {
            mark[x]=0;
            change(a[x],-1);
        }
        else {
            mark[x]=1;
            change(a[x],1);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}