中国石油大学开放训练赛 K.数学问题（组合数预处理+二维前缀和）

数学-组合数学

数学问题（组合数预处理+二维前缀和）

• 题目

  给出整数n,m,g，求有多少对(i,j)满足g整除$C_i^j$，其中$0\le i\le n,0\le j\le \min (i,m)$

• 输入

  1	//T组数据
  4	//g
  5 4	//n,m
  

• 输出

  2
  

判断g是否整除$C_i^j$，就是判断$C_i^j\%g$是否为0。

对于组合数的计算，可以依据组合数的递推公式，$C_i^j=C_{i-1}^{j-1}+C_i^{j-1}$
然后对$C_i^j$进行取模，就转化为$C_i^j==0$时满足条件

但组数有1e4，n,m数据范围为2e3，所以可以利用二位前缀和$s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]$将范围内所有整数对的结果进行预处理，每次的查询就优化到了O(1)

#include <bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define endl '\n'
typedef long long ll;
const int maxn = 2e3 + 10;
int n, m, g;
ll c[maxn][maxn], s[maxn][maxn];
void solve()
{
    //预处理组合数
	c[1][1] = 1;
	for (int i = 0; i <= 2000; i++) c[i][0] = 1;
	for (int i = 2; i <= 2000; i++) {
		for (int j = 1; j <= i; j++) {
			c[i][j] = (c[i - 1][j] + c[i - 1][j - 1]) % g;
		}
	}
    //二位前缀和统计所有整数对的答案
	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 (c[i][j] == 0) s[i][j] += 1;
		}
        //考虑m大于n的情况
		for (int j = i + 1; j <= 2000; j++) {
			s[i][j] = s[i][i];
		}
	}
}
int main()
{
	ios::sync_with_stdio(0), cin.tie(0);
	int t;
	cin >> t >> g;
	solve();
	for (int i = 1; i <= t; i++) {
		cin >> n >> m;
		cout << s[n][m] << endl;
	}
	return 0;
}