LuoguP2822 组合数问题(组合数，二维前缀和)

P2822 组合数问题

输入输出样例

输入样例#1： 复制

1 2
3 3

输出样例#1： 复制

1

输入样例#2： 复制

2 5
4 5
6 7

输出样例#2： 复制

0
7

说明

【样例1说明】

在所有可能的情况中，只有C_2^1 = 2C21​=2是2的倍数。

【子任务】

#include<bits/stdc++.h>

#define N 2001
#define ll long long

using namespace std;
ll n,m,t,k,ans,cnt;
ll C[N+][N+],sum[N+][N+],a[N+][N+];

inline ll read()
{
    int x=,f=;char c=getchar();
    while(c>''||c<''){if(c=='-')f=-;c=getchar();}
    while(c>=''&&c<=''){x=x*+c-'';c=getchar();}
    return x*f;
}

void init()
{
    for(int i=;i<=N;i++) C[i][i]=C[i][]=;
    for(int i=;i<=N;i++) for(int j=;j<i;j++)
    {
        C[i][j]=(C[i-][j]+C[i-][j-])%k;
        if(C[i][j]==) a[i][j]++;
    }
    for(int i=;i<=N;i++) for(int j=;j<=N;j++)
        sum[i][j]=sum[i-][j]+sum[i][j-]-sum[i-][j-]+a[i][j];

}

int main()
{
    t=read();k=read();
    init();
    while(t--)
    {
        n=read();m=read();
        printf("%lld\n",sum[n][m]);
    }
    return ;
}