题意是:有n个数,q次查询。有一个集合初始状态为空,每次查询给出n个数中下标,若集合中没有这个数,则将这个数插入集合,否则将这个数从集合中删除,每次查询给出集合中gcd(x,y)==1的对数。
解法:令F(t)为gcd(x,y)%t==0的对数,f(t)为gcd(x,y)==t的对数 则
$$F(n)=\sum_{n|d}f(d)$$
根据莫比乌斯函数可得
$$f(n)=\sum_{n|d}u(\frac{d}{n})F(d)$$
则$$f(1)=\sum u(d)F(d)$$
F(d)代表gcd(x,y)%d==0的对数,则F(d)与集合中d的倍数相同,所以用一个数组cnt[x]记录集合中x的倍数。
首先预处理出n个数中,每个数的约数。然后每次查询更新cnt[x]的值
#include<bits/stdc++.h>
using namespace std;
int const maxn=5e5+5;
int p[maxn],mob[maxn],a[maxn],cnt[maxn];
bool noprime[maxn],vis[maxn],has[maxn];
vector<int> vt[maxn];
void Mobious()
{
int pnum=0;
mob[1]=1;
for(int i=2;i<maxn;i++)
{
if(!noprime[i])
{
p[pnum++]=i;
mob[i]=-1;
}
for(int j=0;j<pnum&&i*p[j]<maxn;j++)
{
noprime[i*p[j]]=true;
if(i%p[j]==0)
{
mob[i*p[j]]=0;
break;
}
mob[i*p[j]]=-mob[i];
}
}
}
int main()
{
Mobious();
int n,q,idx,ma=0;
scanf("%d%d",&n,&q);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
has[a[i]]=true;
ma=max(ma,a[i]);
}
for(int i=1;i<=ma;i++)
for(int j=i;j<=ma;j+=i)
{
if(has[j])
vt[j].push_back(i);
}
long long ans=0;
while(q--)
{
scanf("%d",&idx);
if(!vis[idx])
{
vis[idx]=true;
for(int i=0;i<(int)vt[a[idx]].size();i++)
{
int num=vt[a[idx]][i];
ans+=(long long)mob[num]*cnt[num];
cnt[num]++;
}
}
else
{
vis[idx]=false;
for(int i=0;i<(int)vt[a[idx]].size();i++)
{
int num=vt[a[idx]][i];
cnt[num]--;
ans-=(long long )mob[num]*cnt[num];
}
}
printf("%I64d\n",ans);
}
return 0;
}