NOIP2016——组合数问题

组合数表示的是从n个物品中选出m个物品的方案数。举个例子,从(1,2,3) 三个物品中选择两个物品可以有(1,2),(1,3),(2,3)这三种选择方法。根据组合数的定 义,我们可以给出计算组合数的一般公式:
20170607153135_30638
其中n! = 1 × 2 × · · · × n
小葱想知道如果给定n,m和k,对于所有的0 <= i <= n,0 <= j <= min(i,m)有多少对 (i,j)满足 20170607153151_90755 是k的倍数。
输入
第一行有两个整数t,k,其中t代表该测试点总共有多少组测试数据,k的意义见 【问题描述】。
接下来t行每行两个整数n,m,其中n,m的意义见【问题描述】。
输出
t行,每行一个整数代表答案。
样例输入
1 2
3 3
输入样例#2:
2 5
4 5
6 7
样例输出
1
输出样例#2:
0
7
提示
【样例1说明】
在所有可能的情况中C(1,2),只有是2的倍数。
【子任务】
在这里插入图片描述

开始还以为是道数学题,推了半天式子看了一下数据范围。woc

直接预处理出2000以内的组合数

但是需要先读入k再处理,在处理的过程中取模

然后再处理一下每一行或者每一列前几个的个数

这样可以做到O(n)查询

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll c[2005][2005],f[2005][2005];
int m,n,k,t;
inline int read(){
	char ch=getchar();
	int res=0;
	while(!isdigit(ch)) ch=getchar();
	while(isdigit(ch)) res=(res<<3)+(res<<1)+(ch^48),ch=getchar();
	return res;
}
inline void init(){
	for(int i=0;i<=2003;i++)c[i][1]=i,c[i][0]=1;
	for(int i=1;i<=2003;i++){
		for(int j=1;j<=i;j++){
			c[i][j]=c[i-1][j-1]+c[i-1][j];
			c[i][j]%=k;
		}
	}
	for(int i=1;i<=2003;i++){
		for(int j=1;j<=i;j++){
			f[i][j]=f[i][j-1]+(c[i][j]%k==0);
		}
	}
}
int main(){
	t=read(),k=read();
	init();	
	while(t--){
		n=read(),m=read();
		m=min(m,n);
		ll ans=0;
		for(int i=1;i<=n;i++){
			int h=min(m,i);
			ans+=f[i][h];
		}
		cout<<ans<<'\n';
	}
}

猜你喜欢

转载自blog.csdn.net/qq_42555009/article/details/83175813
今日推荐