AT2000 [AGC002F] Leftmost Ball 排列组合+dp

最新推荐文章于 2022-05-15 01:42:41 发布

nKessi

最新推荐文章于 2022-05-15 01:42:41 发布

阅读量115

点赞数

分类专栏： DP 数学

本文链接：https://blog.csdn.net/Clever_Hard/article/details/115334536

版权

DP 同时被 2 个专栏收录

18 篇文章 0 订阅

订阅专栏

数学

8 篇文章 0 订阅

订阅专栏

题意

在这里插入图片描述

分析

令 $d p [i] [j]$ 为选 $i$ 个白色球，选 $j$ 个其他颜色的方案总数。

则 $dp[i][j]=dp[i-1][j]+dp[i][j-1]\times (n-j+1)\times C_{n\times k-i-(k-1)(j-1)-1}^{k-2}(i>=j)$

这里有一个很重要的思维，即限定避免重复思维：在这道题中，选其他颜色时，必须规定第一个必须选非白色球，不然会重复。
注意要特判 $k = 1$ 的情况： $a n s = 1$ 。

Code

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <climits>
#define LL long long
using namespace std;
const int MAXN = 2005, Mod = 1e9 + 7;
// 问颜色序列在疯狂暗示将白色单独提出来 
// 这里有一个很重要的思维，即限定避免重复思维：
// 在这道题中，选其他颜色时，必须规定第一个必须选非白色球，不然会重复。
int n, k, dp[MAXN][MAXN], jc[MAXN * MAXN], Inv[MAXN * MAXN];
int Quick_Pow(int x, int y) {
	int ans = 1;
	for(; y; y >>= 1) {
		if(y & 1) ans = (LL)ans * x % Mod;
		x = (LL)x * x % Mod;
	}
	return ans;
}
int main() {
	scanf("%d%d", &n, &k); jc[0] = 1; dp[0][0] = 1; Inv[0] = 1;
	for(int i = 1; i <= n * k; i ++) jc[i] = (LL)jc[i - 1] * i % Mod, Inv[i] = Quick_Pow(jc[i], Mod - 2);
	if(k == 1) { printf("%d", 1); return 0; }
	for(int i = 1; i <= n; i ++) {
		for(int j = 0; j <= i; j ++) {
			dp[i][j] = dp[i - 1][j];
			int u = k - 2, v = n * k - i - (k - 1) * (j - 1) - 1;
			if(u > v || !j) continue;
			dp[i][j] = (dp[i][j] + (LL)dp[i][j - 1] * (n - j + 1) % Mod * jc[v] % Mod * Inv[u] % Mod * Inv[v - u] % Mod) % Mod;
		}
	}
	printf("%d", dp[n][n]);
	return 0;
}