P2822 组合数问题

最新推荐文章于 2021-05-21 02:40:08 发布

ssl_xxy

最新推荐文章于 2021-05-21 02:40:08 发布

阅读量172

点赞数

文章标签： P282 组合数

本文链接：https://blog.csdn.net/xuxiayang/article/details/108781587

版权

组合计数同时被 2 个专栏收录

13 篇文章 0 订阅

订阅专栏

二维前缀和

3 篇文章 0 订阅

订阅专栏

本文介绍了一种利用组合数公式预处理模k的结果，并通过二维前缀和求解特定条件下的组合数求和问题。该方法适用于解决给定n, m和k值，计算Cij mod k等于0的组合数之和的问题，时间复杂度为O(n^2+T)，适用于数据范围不超过3000的题目。

摘要由CSDN通过智能技术生成

$H y p e r l i n k$

https://www.luogu.com.cn/problem/P2822

$D e s c r i p t i o n$

给定 $T$ 组数据和一个数 $k$ ，每组数据给定一个 $n, m$ ，要求回答
$\sum _{i=1}^n \sum_{j=1}^{min(n,m)} [C_i^j\mod k==0]$

数据范围： $n,m\leq 3\times 10^3,T\leq 10^4,k\leq 21$

$S o l u t i o n$

利用组合数公式 $C_i^j=C_{i-1}^{j-1}+C_{i-1}^j$ 预处理 $C$ 对 $k$ 取模的结果
用二维前缀和保存其为0的情况即可

时间复杂度： $O(n^2+T)$

$C o d e$

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;int T,n,m,k,C[2010][2010],s[2010][2010];
inline LL read()
{
	char c;LL d=1,f=0;
	while(c=getchar(),!isdigit(c)) if(c=='-') d=-1;f=(f<<3)+(f<<1)+c-48;
	while(c=getchar(),isdigit(c)) f=(f<<3)+(f<<1)+c-48;
	return d*f;
}
signed main()
{
	T=read();k=read();
	C[0][0]=1;C[1][0]=1;C[1][1]=1;
	for(register int i=2;i<=2000;i++)
	{
		C[i][0]=1;
		for(register int j=1;j<=i;j++)
		{
			C[i][j]=(C[i-1][j-1]+C[i-1][j])%k;
			s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+(C[i][j]==0);
		}
		s[i][i+1]=s[i][i];
	}
	while(T--)
	{
		n=read();m=read();
		if(m<=n) printf("%d\n",s[n][m]);
		else printf("%d\n",s[n][n]);
	}
}