POJ 2992 Divisors 求组合数的约数个数

Description

Input

Output

Sample Input

5 1
6 3
10 4

Sample Output

2
6
16

Source

//3624K	610MS
#include<stdio.h>
#include<string.h>
#define N 100007
bool visit[1010];
long long c[1500][1500];
long long  prime[107];
void init_prim()//prime存的是下标，visit存的是数。visit[5]==true。
{
    long long num=0;
    memset(visit,true,sizeof(visit));
    for(long long i=2;i<1007;++i)
    {
        if(visit[i]==true)
        {
            num++;
            prime[num]=i;
        }
        for(long long j=1;((j<=num)&&(i*prime[j]<=10007));++j)
        {
            visit[i*prime[j]]=false;
            if(i%prime[j]==0) break;
        }
    }
}
void init()
{
    for(long long i=2;i<=437;i++)
        for(long long j=1;prime[j]<=i;j++)
        {
            long long n=i,res=0;
            while(n){n/=prime[j];res+=n;}
            c[i][prime[j]]=res;
        }
}
int main()
{
    long long n,k;
    init_prim();
    init();
    while(scanf("%I64d%I64d",&n,&k)!=EOF)
    {
        long long ans=1,a;
        for(long long i=1;prime[i]<=n;i++)
        {
            a=c[n][prime[i]]-c[k][prime[i]]-c[n-k][prime[i]];
            ans*=(1+a);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}