GDUT_专题四 M - 组合数问题

最新推荐文章于 2024-07-25 17:57:45 发布

Hqwq_

最新推荐文章于 2024-07-25 17:57:45 发布

阅读量37

点赞数

文章标签： c++

本文链接：https://blog.csdn.net/Hqwq_/article/details/129129700

版权

题目：

M - 组合数问题

题意：

给定 $n,m,k$ ，有 $0\leq i\leq n,0\leq j\leq min(i,m)$ ，求有多少对 $i,j$ 满足 $k|C_i^j$ 。

分析：

定义 $f(i,j)=C_i^j$ ，可以利用杨辉三角求出 $f(i,j )$ 。定义 $s(i,j)$ 为当 $n=i,m=j$ 时的答案，易得状态转移方程 $s(i,j)=s(i-1,j)+s(i,j-1)-s(i-1,j-1)$ 。为了防止数据过大，每次求出一个 $f(i,j)$ 都令其mod $k$ ，当 $f(i,j)$ 为0时就说明其满足 $k|C_i^j$ 。同时为了防止越界，当 $m>n$ 时要令 $m=n$ ； $s(i,i+1)=s(i,i)$ 。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int f[2001][2001],s[2001][2001];

int main()
{
	int t,k;
	scanf("%d %d",&t,&k);
	for(int i=0;i<=2000;i++)
	{
		f[i][0]=f[i][i]=1;
		for(int j=1;j<i;j++)
		{
			f[i][j]=(f[i-1][j]+f[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 (f[i][j]==0) s[i][j]++;
		}
		s[i][i+1]=s[i][i];	
	}
	while(t--){
		int n,m;
		scanf("%d %d",&n,&m);
		if (m>n) m=n;
		printf("%d\n",s[n][m]);
	}
}