CF 305 div2 E. Mike and Foam (容斥原理)

题目：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 × 10⁵，5 × 10⁵<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;
}