洛谷P2822 组合数问题(组合数打表预处理+二维前缀和)

最新推荐文章于 2024-03-07 15:30:01 发布

Sqwlly

最新推荐文章于 2024-03-07 15:30:01 发布

阅读量269

点赞数

分类专栏：算法 ACM 洛谷组合数学预处理打表文章标签：洛谷P2822 组合数打表二维前缀和

本文链接：https://blog.csdn.net/Eternally831143/article/details/82765363

版权

算法同时被 3 个专栏收录

203 篇文章 1 订阅

订阅专栏

ACM

127 篇文章 0 订阅

订阅专栏

洛谷

32 篇文章 0 订阅

订阅专栏

P2822 组合数问题

题解

$\Theta(N^2)$ 预处理打出范围在 $2000$ 以内的组合数，做个二维前缀和 $\Theta(1)$ 查询。

代码

#include<bits/stdc++.h>
typedef long long LL;
using namespace std;

//组合数打表模板,适用于N<=3000
//c[i][j]表示从i个中选j个的选法。
int C[2018][2018], sum[2018][2018];

void get_C(int maxn,int k)
{
    C[0][0] = 1;
    for(register int i=1;i<=maxn;i++)
    {
        C[i][0] = 1;
        for(register int j=1;j<=i;j++)
            C[i][j] = (C[i-1][j]%k+C[i-1][j-1]%k)%k;
        //C[i][j] = (C[i-1][j]+C[i-1][j-1])%MOD;
    }
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("input.in","r",stdin);
#endif
	int n,m,t,k;
	scanf("%d%d",&t,&k);
	get_C(2017,k);
	for(int i = 1; i < 2018; ++i){
		for(int j = 1; j < 2018; ++j){
			sum[i][j] += sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1];
			if(C[i][j]==0)
				sum[i][j]++;
		}
		sum[i][i+1] = sum[i][i];
	}
	while(t--){
		scanf("%d%d",&n,&m);
		if(n < m) m = n;
		printf("%d\n",sum[n][m]);
	}
    return 0;
}