hdu5072 Coprime | 2014鞍山赛区C题 | 容斥原理

版权声明:如果此条博客对你有一丝的帮助,请留言支持一下博主。 https://blog.csdn.net/csuhoward/article/details/44978087

感谢 http://dtyfc.com/acmblog/acm/980 的解题思路,“如果三个数a, b, c不符合条件,那么一定有一对是互质的,有一对是不互质的。不妨令a, b互质,b, c不互质。于是我们可以枚举b来统计答案。在除了b自己的所有数中,要么与b互质,要么与b不互质。假设n个数中有x个与b不互质的数,那么b对答案的贡献就是(x - 1) * (n - x)。注意这里的求出答案之后要除以2,这是因为如果a, c互质,那么可以交换b, c的位置;如果a, c不互质,那么可以交换a, b的位置,每个答案都被计算了两遍”是我认为最精彩的地方,把题目的脑洞推向了高潮~~


这道题是要求互不相同的n个数中有多少abc这样的组合,满足abc两两互质或者两两不互质。三元关系很伤脑筋,转化为二元关系也许能简化问题(这里脑洞要慢慢打开了)。


由于数据范围不大10^5以内,总组合数C(n,3) longlong不会爆。abc两两互质和两两不互质,就对应着两个互质另两个不互质,这两个集合构成了全集U。不妨把前者称为集合A,后者称为集合B,那么A并B等于U,且A交B为空。U的大小为C(n,3)。


如果a,b,c不符合条件,必然有一对互质,一对不互质,不妨设a,b互质,b,c不互质,于是我们可以枚举b来统计所有的三元组:如果a,c互质那么这样的三元组中b,c可以互换位置;如果a,c不互质,那么a,b可以互换位置。每个答案被算了两遍。


所以只要枚举每个b,统计出k个和它不互质的,那么剩下n-1-k个就是和它互质的,那么三元组就有k*(n-1-k)/2种。


如果会快速统计1~n个数中有多少和b不互质的数,那么题目就差不多就做完了...

===================================================================

怎么快速统计不互质的数呢?


对于b不超过10^5,质因子的个数不超过6个(2*3*5*7*11*13 *17>10^5)。用状压搜索质因子组成的每个因数,如果某数是该因数的倍数,那么就说明该数和b是不互质的。利用容斥原理统计出与b不互质的数的综述。


由于数据范围不超过10^5,预处理筛除出每个质数和每个质因子,复杂度为nlogn。对于具体的n个数,再筛出在n个数中以他们为倍数的数的个数也是nlogn。(代码中用cntExtend[]记录)


==================================================


#include <cstdio>
#include <cstring>
#include <iostream>
#include <cstring>
using namespace std;
typedef long long LL;
const int maxn = 111111;

//filter prim
int prim[maxn],isprim[maxn];
int primnum=0;
void initprim(){
    memset(isprim,-1,sizeof isprim);
    isprim[0]=isprim[1]=0;
    prim[primnum++] = 2;
    for(int i=4;i<maxn;i+=2){
        isprim[i]=0;
    }
    for(int i=3;i<maxn;i+=2){
        if(isprim[i]){
            prim[primnum++]=i;
            for(int j=i+i;j<maxn;j+=i){
                isprim[j]=0;
            }
        }
    }
}

//get factor
int has[maxn];
int factor[maxn][6];
int factornum[maxn];
void getfactor(){
    for(int num=2;num<maxn;num++){
        int n=num,cnt = 0;
        for(int i=0;i<primnum;i++){
            if(isprim[n]) {
                factor[num][cnt++]=n;
                break;
            }
            if(n%prim[i]==0){
                factor[num][cnt++]=prim[i];
                while(n%prim[i]==0){
                    n/=prim[i];
                }
            }
        }
        factornum[num]=cnt;
    }
}

// count the number of factors of every number
int num[maxn],cntExtend[maxn];
void factorExtend(int len){
    memset(cntExtend,0,sizeof cntExtend);
    for(int i=1; i<maxn; i++){
        for(int j=i; j<maxn; j+=i){
            if(has[j])
                cntExtend[i]++;
        }
    }
}

//for every single factor in each number, find out how many numbers is not co-prime with it
LL solve(int len){
    LL re = 0;
    for(int i=0; i<len; i++){
        int n = num[i];
        if(n==1) continue;
        int facnum = factornum[n];
        LL sum=0;
        for(int k=(1<<facnum)-1; k>0; k--){
            int mul=1,b=0;
            for(int j=0; j<facnum; j++){
                if((1<<j) & k) {
                    mul*=factor[n][j];
                    b^=1;
                }
            }
            if(b){
                sum+=cntExtend[mul]-1;
            }else{
                sum-=cntExtend[mul]-1;
            }
        }
        re+=(sum)*(len-1-sum);
    }
    return re;
}

int main(){
//    freopen("data.in","r",stdin);
    int T,n,x;
    initprim();
    getfactor();
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        memset(has,0,sizeof has);
        for(int i=0;i<n;i++){
            scanf("%d",&x);
            has[x]++;
            num[i]=x;
        }
        factorExtend(n);
        LL ans = (LL)n*(n-1)*(n-2)/6 - solve(n)/2;
        printf("%I64d\n",ans);
    }

}



展开阅读全文

没有更多推荐了,返回首页