POJ3904 Sky Code 容斥原理

这道题很容易想到用总的情况减去不互素的情况，需要统计不同素因子对应的数据个数，但这样会多算（比如2的因子减过一次，3的因子减过一次，这时6的情况(eg:6,12,18,24)就多减了一次），因此想到用容斥原理。问题的关键在于，怎么得到不同素因子组合对应的情况。附上大佬博客Orz：https://www.cnblogs.com/lmjer/p/9325754.html

这篇博客讲的很清楚，主要就是利用二进制。代码中有详细的解释，直接上AC代码：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
const int INF=0x3f3f3f3f;

const int MAX=10005;
int cnt[MAX];///含i的数据个数
int num[MAX];///i的素因子个数
///i指不同素因子组合相乘得到的数,eg:6=2*3,不是4=2*2(cnt=0)!!!

int pri[MAX];//每个数据的素因子
void solve(int n)
{
    int tot=0;
    for(int i=2;i*i<=n;i++)//素因子分解
    {
        if(n%i==0)
            pri[tot++]=i;
        while(n%i==0)
            n/=i;
    }
    if(n>1)///注意!i*i<=n可降低时间复杂度!
        pri[tot++]=n;
    for(int i=1;i<(1<<tot);i++)///二进制枚举不同素因子组合的各种情况
    {
        int k=1;///不同素因子相乘得到的数
        int sum=0;///该结果用了几个素因子
        for(int j=0;j<tot;j++)
            if(i&(1<<j))//i的第j位为1,表示用到了该素因子
            {
                k*=pri[j];
                sum++;
            }
        cnt[k]++;
        num[k]=sum;
    }
}

ll Cn4(ll n)///注意都是long long类型!
{
    ll ans=n*(n-1)*(n-2)*(n-3)/24;
    return ans;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(cnt,0,sizeof(cnt));
        memset(num,0,sizeof(num));
        int x;
        for(int i=0;i<n;i++)
        {
            scanf("%d",&x);
            solve(x);
        }
        ll tmp=0;//不互素的个数
        for(int i=2;i<MAX;i++)
            if(cnt[i]>=4)
            {   ///容斥定理
                if(num[i]&1)   //奇+
                    tmp+=Cn4(cnt[i]);
                else           //偶-
                    tmp-=Cn4(cnt[i]);
            }
        ll ans=Cn4(n)-tmp;
        printf("%lld\n",ans);
    }
	return 0;
}