[NOIp2016] 组合数问题

洛谷 P2822 传送门

这么水的题我居然还调了好几次......

对于所有询问，k都是一样的。

所以先用杨辉三角求出2000以内所有组合数mod k的值c[i][j]。

如果c[i][j]%k==0，那么c[i][j]对答案的贡献f[i][j]就是1。

最后对f[i][j]求个前缀和就好了。

#include<cstdio>

int t,k,n,m;
int c[2005][2005];
int f[2005][2005];

int main()
{
    scanf("%d%d",&t,&k);
    c[0][0]=1%k;
    for(int i=1;i<=2000;i++)
    {
        c[i][0]=1%k;
        for(int j=1;j<=i;j++)
        {
            c[i][j]=(c[i-1][j]+c[i-1][j-1])%k;
        }
    }
    for(int i=0;i<=2000;i++)
    {
        for(int j=0;j<=i;j++)
        {
            if(!c[i][j])f[i][j]=1;
        }
    }
    for(int i=1;i<=2000;i++)
    {
        f[i][0]+=f[i-1][0];
        for(int j=1;j<=2000;j++)
        {
            f[i][j]+=f[i-1][j]+f[i][j-1]-f[i-1][j-1];
        }
    }
    for(int i=1;i<=t;i++)
    {
        scanf("%d%d",&n,&m);
        if(m>n)m=n;
        printf("%d
",f[n][m]);
    }
    return 0;
}