题目:http://codeforces.com/contest/548/problem/E
题意:有n个位置,每个位置上对应一个数字a[i],开始所有位置都没有数字,有q次查询,每查询一次,输入位置,如果对应位置上这个数字存在,那么把该位置对应的数字删掉,如果对应位置上的数字不存在,那么把该位置上的数字添上,输出此时有多少对数互质。
分析:对一个数而言,找与它不互质的数更简单。找与一个数x不互质的数,就是找哪些数与x有公共的质因子。比如求40与哪些数不互质,就是找哪些数有质因子2或5。对于每次查询q[i],将a[q[i]]质因子分解,然后利用容斥原理算出哪些数与a[q[i]]不互质。还有一点,因为a[i]<=5 × 105,5 × 105<2*3*5*7*11*13*17,对于每次查询,用DFS二进制枚举的话时间复杂度是O(2^7)。
代码:
#include <iostream>
#include <sstream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <string>
#include <map>
#include <set>
//#include <unordered_map>
//#include <unordered_set>
#include <utility>
#include <stack>
#include <deque>
#include <queue>
#include <list>
#include <bitset>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
#define rep(i,x) for(int i=0;i<x;i++)
int prime[1001],nprime;
bool isprime[1001];
int fact[100],nfact;
bool visit[500006];
int use[500006],cnt[500006];
void doprime()
{
nprime=0;
int i,j;
for(i=1;i<1000;i+=2)
isprime[i]=true;
isprime[1]=false;
isprime[2]=true;
for(i=2;i<1000;i++)
{
if(isprime[i])
{
prime[nprime++]=i;
for(j=i*2;j<1000;j+=i)
isprime[j]=false;
}
}
}
void dofact(int x)
{
nfact=0;
for(int i=0;i<nprime;i++)
{
if(x%prime[i]==0)
{
fact[nfact++]=prime[i];
while(x%prime[i]==0)
x/=prime[i];
if(x==1)
return ;
}
}
if(x>1)
fact[nfact++]=x;
}
int main()
{
doprime();
int n,q,i,j,k,x,c,t,f,N=0,trans;
LL ans=0,ret;
scanf("%d%d",&n,&q);
for(i=1;i<=n;i++)
scanf("%d",&use[i]);
while(q--)
{
scanf("%d",&x);
trans=use[x];
if(trans==1)
{
if(visit[x])
{
visit[x]=false;
ans=ans-(N-1);
N--;
}
else
{
visit[x]=true;
ans=ans+N;
N++;
}
printf("%lld\n",ans);
continue;
}
dofact(trans);
if(!visit[x])
{
visit[x]=true;
ret=0;
for(i=1;i<(1<<nfact);i++)
{
f=1;
c=0;
for(j=0;j<7;j++)
{
if(i&(1<<j))
{
c++;
f*=fact[j];
}
}
if(c&1)
ret+=cnt[f];
else
ret-=cnt[f];
++cnt[f];
}
ans=ans+(N-ret);
N++;
}
else
{
visit[x]=false;
ret=0;
for(i=1;i<(1<<nfact);i++)
{
c=0;
f=1;
for(j=0;j<7;j++)
{
if(i&(1<<j))
{
c++;
f*=fact[j];
}
}
--cnt[f];
if(c&1)
ret+=cnt[f];
else
ret-=cnt[f];
}
N--;
ans=ans-(N-ret);
}
printf("%lld\n",ans);
}
return 0;
}