NOIP 2016 Day2 T1 组合数问题

题目大意

T组数据，每组给出n、m，问有多少个C（i，j）（0<=i<=n，0<=j<=min（i，m））是k的倍数。

T<=10000，n,m<=2000，k<=21
时间限制 1s
空间限制 512M

解题思路

首先，我们可以对最大的n、m求答案，其余的T-1个询问就可以顺便求出来了。
那么如何求有多少个C（i，j）（0<=i<=n，0<=j<=min（i，m））是k的倍数呢？

1、预处理pr[i]、sum[i]、c[i][j]分别表示k的第i个因子，k的第i个因子的个数，以及i！含有多少个pr[j]；
2、对于每个c（i，j），我们可以求出它含有的第x个因子的个数为c[i][x]-c[j][x]-c[i-j][x]，如果这个数大于sum[x]，则c（i，j）可以被第x个因子除尽。然后如果k的所有因子都满足上述条件，c（i，j）就是k的倍数了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 2006
#define maxt 10006
#define fr(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;

int T,i,j,k,nx,mx,tot,s,n[maxt],m[maxt];
int c[maxn][10],sum[maxn],pr[maxn],ans[maxn][maxn];
int calc(int n,int x)
{
    int i,s=0,t=x;
    while (x<=n)
    {
        s+=n/x;
        x*=t;
    }
    return s;
}
void pre(int n)
{
    int i,j,t=max(nx,mx),nn=n;
    fr(i,2,nn)
    {
        if (n%i==0)
        {
            pr[++tot]=i;
            while (n%i==0) ++sum[tot],n/=i;
        }
        if (n==1) break;
    }
    fr(i,1,tot)
        fr(j,1,t) c[j][i]=calc(j,pr[i]);
    return;
}
bool check(int n,int m)
{
    int i;
    fr(i,1,tot)
        if (c[n][i]-c[m][i]-c[n-m][i]<sum[i]) return 0;
    return 1;
}
int main()
{
    freopen("problem.in","r",stdin);
    freopen("problem.out","w",stdout);
    scanf("%d%d",&T,&k);
    fr(i,1,T)
    {
        scanf("%d%d",&n[i],&m[i]);
        nx=max(nx,n[i]),mx=max(mx,m[i]);
    }
    pre(k);
    fr(i,0,nx)
    {
        int g=min(i,mx);
        fr(j,0,g) if (check(i,j)) ans[i][j]++;
    }
    fr(i,1,nx)
        fr(j,1,mx) ans[i][j]+=ans[i-1][j]+ans[i][j-1]-ans[i-1][j-1];
    fr(i,1,T) 
        if (m[i]<n[i]) printf("%d\n",ans[n[i]][m[i]]);
        else printf("%d\n",ans[n[i]][n[i]]);
    return 0;
}