POJ 2992

1609 ワード

この問題は私に午後1時間詰まった.
ずっとTLEなので、テストデータはきっと神多組だと推測しています.
データを読むたびに連続数乗の因数分解を繰り返すことに気づきました
だからn!の分解方式はすべて前処理して出力を計算すればいいです.
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define N 432
typedef long long LL;
const int cnt = 83;
int p[400] = {2,3,5,7,11,13,17,19,23,29,//10
              31,37,41,43,47,53,59,61,67,//9
              71,73,79,83,89,97,101,103,107,//9
              109,113,127,131,137,139,149,151,//8
              157,163,167,173,179,181,191,193,
              197,199,211,223,227,229,233,239,
              241,251,257,263,269,271,277,281,
              283,293,307,311,313,317,331,337,
              347,349,353,359,367,373,379,383,
              389,397,401,409,419,421,431};
int num[N+1][400];
void fac()
{
    memset(num,0,sizeof(num));
    for(int k=2;k<N;k++)
    for(int i=2;i<=k;i++)
    {
        int t = i;
        for(int j=0;j<cnt;j++)
        {
            while(t%p[j] == 0)
            {
                t /= p[j];
                num[k][j]++;
            }
            if(t == 1)break;
        }
    }
}
int main()
{
    fac();
    LL ret[N+1][N+1];
    for(int i=0;i<N;i++)
        for(int j=0;j<N;j++)
            ret[i][j] = -1;
    int n,k;
    while(~scanf("%d%d",&n,&k))
    {
        if(ret[n][k]!=-1){printf("%I64d
",ret[n][k]);continue;} ret[n][k] = 1LL; for(int i=0;i<cnt;i++) ret[n][k] *= (LL)(num[n][i]-num[n-k][i]-num[k][i]+1); printf("%I64d
",ret[n][k]); } return 0; }