【2020.10.29 洛谷团队赛 普及组】T3 U138032 组合数

题目描述

---

输入格式
第一行有两个整数$t，k，$其中t表示该测试点总共有多少组测试数据，k的意义见「题目描述」。
接下来t行每行两个整数$n，m$，其中$n，m$的意义见「题目描述」。

输出格式
输出共t行，每行一个整数代表对于所有的 $0\le i\le n，0\le j\le min(i,m)$有多少对 $(i,j)$ 满足 $C(i,j)$是k 的倍数。

---

输入输出样例
输入 #1
1 2
3 3
输出 #1
1

输入 #2
2 5
4 5
6 7
输出 #2
0
7

输入 #3
3 6
2 8
4 6
6 9
输出 #3
0
1
3

---

说明/提示
数据范围
对于15%的数据，$3\le n,m\le 10；$
对于50%的数据，$t=1；$
对于100%的数据，$3\le n,m\le 2000,2\le k\le 21,1\le t\le 10000；$

---

样例解释
对于样例1，在所有可能的情况中，只有$C（2,1）=2$是 2 的倍数。

---

解题思路

---

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m,kk,t,ans,a[2003][2003],b[2003][2003];
long long x,w;
int main() {
	scanf("%d%d",&t,&kk);
	for(int i=1;i<=2000;i++)
		a[i][1]=i%kk,a[i][i]=1;
	for(int i=2;i<=2000;i++)
	{
		for(int j=2;j<=i-1;j++)
		{
			a[i][j]=(a[i-1][j]%kk+a[i-1][j-1]%kk)%kk;
		}
	}
	for(int i=1;i<=2000;i++)
	{
		for(int j=1;j<=i;j++)
		{
			if(a[i][j]) b[i][j]=b[i][j-1];
			else b[i][j]=b[i][j-1]+1;
		}
	}
	while(t--) {
		scanf("%d%d",&n,&m);
		ans=0;
		for(int i=1;i<=n;i++)
			if(m<i)ans+=b[i][m];
			else ans+=b[i][i];
		printf("%d\n",ans);
	}
}