牛客16429-组合数问题

题目链接
题目描述


分析：
          这道题是一道组合数解决的题，数据范围较小，可以通过递推公式$C_a^b=C_{a-1}^b\ast C_{a-1}^{b-1}$来提前记录好所有组合数。跟这道题类似，因为这道题是求整除k的数，因此在存组合数时，只存对k取余后的余数即可，余数为0即为整除k的组合数。并且用前缀和数组保存好前缀和即可。
代码如下：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int N=2010;
int t,k;
int n,m;
int q[N][N];
int f[N][N];
void init()
{
    for(int i=1;i<=2000;i++)
        for(int j=0;j<=i;j++)
            if(j==0||i==j)
            {
                q[i][j]=1;
                q[i][j]%=k;
            }
            else
                q[i][j]=((ll)q[i-1][j]+q[i-1][j-1])%k;
    for(int i=1;i<=2000;i++)
            for(int j=0;j<=i;j++)
                if(!q[i][j])
                {
                    if(j) f[i][j]=f[i][j-1]+1;
                    else f[i][j]=1;
                }
                else
                {
                    if(j) f[i][j]=f[i][j-1];
                    else f[i][j]=0;
                }
}
int main()
{
    scanf("%d%d",&t,&k);    
    init();
    while(t--)
    {
        int n,m;
        int res=0;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
            res+=f[i][min(i,m)];
        printf("%d\n",res);
    }
    return 0;
}