天天看点

hdu1796(容斥)

链接:点击打开链接

题意:给定n和一个大小为m的集合,求[1,n-1]内能被集合里任意一个数整除的数字个数。

代码:

#include <map>
#include <set>
#include <queue>
#include <string>
#include <math.h>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
int main(){
    long long a[30];
    long long i,j,k,t,n,m,op,ans,tmp;
    while(scanf("%I64d%I64d",&n,&m)!=EOF){
        op=0,n--;
        for(i=0;i<m;i++){                       //直接用lcm容斥,注意一下有0
            scanf("%I64d",&j);
            if(j!=0)
            a[op++]=j;
        }
        m=op,ans=0;
        for(i=1;i<(1<<m);i++){
            j=i;
            k=op=0,tmp=1;
            while(j){
                if(j&1){
                    op++;
                    tmp=tmp/__gcd(tmp,a[k])*a[k];
                }
                k++;
                j>>=1;
            }
            if(op%2!=0)
            ans+=(n/tmp);
            else
            ans-=(n/tmp);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}