【Codeforces】547C Mike and Foam 容斥

题目传送门

题目要我们求出容器中与$a_x$互质的数的个数，也就是容器的总个数减去不与$a_x$互质的数的个数。这样一转化就变成了容斥问题。

题目给定每个数小于$5\times 10^5$，因为$2 \times 3 \times 5 \times 7 \times 11 \times 13 \times 17 \ge 5 \times 10^5$，也就是每个数最多可能有$7$个质因子，那么我们直接暴力$2^7$容斥即可。

附上AC代码：

#include <cstdio>
#include <cctype>
using namespace std;

typedef long long ll;
const int N=2e5+10,M=5e5+10;
int n,m,p[M],num,x,a[N][10],sum[M];
bool b[M],vis[N];
ll ans;

inline char nc(void){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
    return (void)(a*=f);
}

inline void print(ll x){
    int u[20],len=0;
    if (!x) return (void)(puts("0"));
    while (x) u[++len]=x%10,x/=10;
    for (int i=len; i; --i) putchar(u[i]+'0');
    return (void)(putchar('\n'));
}

inline void prime(int n){
    for (int i=2; i<=n; ++i){
        if (!b[i]) p[++num]=i;
        for (int j=1; j<=num&&p[j]*i<=n; ++j){
            b[p[j]*i]=1;
            if (i%p[j]==0) break;
        }
    }
    return;
}

inline void ask(int x,int y){
    for (int i=0; i<(1<<a[x][0]); ++i){
        int t=1,cnt=0;
        for (int j=1; j<=a[x][0]; ++j)
            if (i&(1<<j-1)) t*=a[x][j],++cnt;
        if (cnt&1) ans-=y*sum[t]; else ans+=y*sum[t];
    }
    return;
}

inline void updata(int x,int y){
    for (int i=0; i<(1<<a[x][0]); ++i){
        int t=1;
        for (int j=1; j<=a[x][0]; ++j)
            if (i&(1<<j-1)) t*=a[x][j];
        sum[t]+=y;
    }
    return;
}

int main(void){
    read(n),read(m),prime(5e5);
    for (int i=1; i<=n; ++i){
        read(x);
        for (int j=1; p[j]*p[j]<=x&&j<=num; ++j)
            if (x%p[j]==0){
                a[i][++a[i][0]]=p[j];
                while (x%p[j]==0) x/=p[j];
            }
        if (x>1) a[i][++a[i][0]]=x;
    }
    while (m--){
        read(x);
        if (!vis[x]) ask(x,1),updata(x,1);
        else updata(x,-1),ask(x,-1);
        vis[x]^=1,print(ans);
    }
    return 0;
}