Ural-1091 Tmutarakan Exams

題解： 容斥原理經典題目，mark一下~

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;

int k,s;
int pri[55],ans=0;
bool vis[55];
int c[55][55];

void getpri(int n){ // 歐拉篩法
    memset(vis,0,sizeof(vis));
    for(int i=2;i<=n;i++){
        if(!vis[i])pri[ans++]=i;
        for(int j=0;j<ans;j++){
            if(i*pri[j]>n)break;
            vis[i*pri[j]]=1;
            if(i%pri[j]==0)break;
        }
    }
}

bool jud(int n){ // 判斷n是兩個素數的乘積
    for(int i=0;i<ans&&pri[i]<n;i++){
        if(n%pri[i]==0&&!vis[n/pri[i]]&&pri[i]!=n/pri[i])return true;
    }
    return false;
}

void init(int n){ // 初始化組合數C(n,m)
    memset(c,0,sizeof(c));
    for(int i=0;i<=n;i++)
        c[i][0]=1;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=i;j++)
            c[i][j]=c[i-1][j-1]+c[i-1][j];
    }
}
int main()
{
    getpri(50);
    init(50);
    scanf("%d%d",&k,&s);
    ll sum=0;
    for(int i=2;i<=s;i++){
        if(!vis[i])sum+=c[s/i][k];
        else if(jud(i))sum-=c[s/i][k];

    }
    printf("%lld\n",sum>10000?10000:sum);
    return 0;
}