[COCI2020-2021#6] Anagramistica

maniubi發表於2024-09-11

[COCI2020-2021#6] Anagramistica

題意

給定 \(n\) 個字串和正整數 \(k\)

定義兩個字串相似當且僅當兩個字串排序後相等。

可以從中選出一些字串,求有多少種方案,使得其中恰好有 \(k\) 對字串相似。

思路

先將所有字串排序,相同的歸為一類,求出 \(cnt_i\) 表示第 \(i\) 種字串的個數。

定義 \(dp_{i,j}\) 表示前 \(i\) 種字串,有 \(j\) 對相似的方案數。

轉移方程:\(dp_{i,j} = \sum dp_{i-1,j-C_k^{2}}\times C_{cnt_i}^{k}\)

程式碼

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 2e3 + 5;
const int mod = 1e9 + 7;
int n, k, dp[N][N], fac[N], facinv[N];
map <string, int> M;
int qpow(int a, int b) {
	int res = 1;
	for (; b; b >>= 1, a = a * a % mod) 
		if (b & 1) res = res * a % mod;
	return res;
}
int C(int x, int y) {
	return fac[x] * facinv[y] % mod * facinv[x - y] % mod;
}
void solve() {
	cin >> n >> k;
	for (int i = 1; i <= n; i ++) {
		string s; cin >> s;
		sort(s.begin(), s.end());
		M[s] ++;
	}
	fac[0] = facinv[0] = 1;
	for (int i = 1; i <= 2000; i ++) {
		fac[i] = fac[i - 1] * i % mod;
		facinv[i] = qpow(fac[i], mod - 2);
	}
	dp[0][0] = 1; int l = 0;
	for (auto p : M) {
		l ++;
		for (int i = 0; i <= k; i ++)
			for (int j = 0; j <= p.second; j ++) {
				if (i < j * (j - 1) / 2) continue;
				dp[l][i] += dp[l - 1][i - j * (j - 1) / 2] * C(p.second, j);
				dp[l][i] %= mod;
			}
	}
	cout << dp[l][k] << "\n"; 
}
signed main() {
	int Case = 1;
//	cin >> Case;
	while (Case --)
		solve();
	return 0;
}