洛谷P2822 组合数问题

题目描述

组合数 \binom{n}{m}(mn​) 表示的是从 nn 个物品中选出 mm 个物品的方案数。举个例子，从 (1,2,3)(1,2,3) 三个物品中选择两个物品可以有 (1,2),(1,3),(2,3)(1,2),(1,3),(2,3) 这三种选择方法。根据组合数的定义，我们可以给出计算组合数 \binom{n}{m}(mn​) 的一般公式：

\binom{n}{m}=\frac{n!}{m!(n-m)!}(mn​)=m!(n−m)!n!​

其中 n!=1\times2\times\cdots\times nn!=1×2×⋯×n；特别地，定义 0!=10!=1。

小葱想知道如果给定 n,mn,m 和 kk，对于所有的 0\leq i\leq n,0\leq j\leq \min \left ( i, m \right )0≤i≤n,0≤j≤min(i,m) 有多少对 (i,j)(i,j) 满足 k|\binom{i}{j}k∣(ji​)。

输入格式

第一行有两个整数 t,kt,k，其中 tt 代表该测试点总共有多少组测试数据，kk 的意义见问题描述。

接下来 tt 行每行两个整数 n,mn,m，其中 n,mn,m 的意义见问题描述。

输出格式

共 tt 行，每行一个整数代表所有的 0\leq i\leq n,0\leq j\leq \min \left ( i, m \right )0≤i≤n,0≤j≤min(i,m) 中有多少对 (i,j)(i,j) 满足 k|\binom{i}{j}k∣(ji​)。

输入输出样例

输入 #1复制

1 2
3 3

输出 #1复制

1

输入 #2复制

2 5
4 5
6 7

输出 #2复制

0
7

说明/提示

【样例1说明】

在所有可能的情况中，只有 \binom{2}{1} = 2(12​)=2 一种情况是 22 的倍数。

【子任务】

• 对于全部的测试点，保证 0 \leq n, m \leq 2 \times 10^30≤n,m≤2×103，1 \leq t \leq 10^41≤t≤104。

上代码：

 

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int t,k,n,m;
int c[2005][2005],s[2005][2005];
void prepare();
int main(){
//    freopen("problem.in","r",stdin);
//    freopen("problem.out","w",stdout);
    memset(c,0,sizeof(c));
    memset(s,0,sizeof(s));
    cin>>t>>k;
    prepare();
    while(t--){
        cin>>n>>m;
        if(m>n) m=n;
        cout<<s[n][m]<<endl;
    }    
    return 0;
} 
void prepare(){
    c[1][1]=1;
    for(int i=0;i<=2000;i++) c[i][0]=1;
    for(int i=2;i<=2000;i++){
        for(int j=1;j<=i;j++){
            c[i][j]=(c[i-1][j]+c[i-1][j-1])%k;
        }
    }
    for(int i=2;i<=2000;i++){
        for(int j=1;j<=i;j++){
            s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1];
            if(c[i][j]==0) s[i][j]+=1;
        }
        s[i][i+1]=s[i][i];
    }
}