P2167 [SDOI2009] Bill的挑戰

Fire_Raku發表於2024-05-31

P2167 [SDOI2009] Bill的挑戰

狀壓 dp/二項式反演

先說狀壓，考慮怎麼刻畫 \(S\) 和 \(T\) 匹配這個東西。實質上就是從前往後匹配每一位，直到哪一位不匹配了，那麼就不匹配，也就是每一位字元匹配的並集。同樣，對於多個串的匹配，設第 \(i\) 位字元為 \(j\) 時匹配的串集合為 \(g_{i,j}\)，對於一個串 \(s\) 匹配的串的個數，就是 \(g_{i,s_1}\) 到 \(g_{i,s_n}\) 的並集。這個是可以預處理的。

所以我們考慮從前往後考慮匹配，求方案數考慮 dp。設 \(f_{i,s}\) 表示考慮完前 \(i\) 位的匹配，匹配串的狀態為 \(s\) 的方案數。轉移列舉 \(i+1\) 的字元 \(c\)，那麼就轉移到 \(f_{i+1,s\cap g_{i+1,c}}\)。

複雜度 \(O(n2^n)\)。

法二：二項式反演

首先看到題面中的恰好兩字，可以想到組合計數。如果我們暴力列舉此時匹配的串的哪 \(i\) 個的話，我們可以容易求出此時的方案數，意義為至少匹配這 \(i\) 個串的方案數，匹配串個數相同的設為 \(f_{i}\)。

再套路設恰好匹配 \(i\) 個串的方案數為 \(g_i\)。有關係：

\[f_{k}=\sum_{i=k}^nC(i,k)g_i\Rightarrow g_k=\sum_{i=k}^n(-1)^{i-k}C(i,k)f_i \]

考慮如何求出 \(f_i\)。跑一遍 dfs 就可以了。

複雜度同樣是 \(O(n2^n)\)。

這裡附狀壓程式碼：

#include <bits/stdc++.h>
#define pii std::pair<int, int>
#define fi first
#define se second
#define pb push_back

using i64 = long long;
using ull = unsigned long long;
const i64 iinf = 0x3f3f3f3f, linf = 0x3f3f3f3f3f3f3f3f;
const int N = 20, M = 52, mod = 1e6 + 3;
i64 n, k, ans;
char a[N][M];
i64 f[M][1 << (N - 4)], g[M][27];
void solve() {
	std::cin >> n >> k;
	for(int i = 0; i < n; i++) {
		scanf("%s", a[i] + 1);
	}
	
	memset(g, 0, sizeof(g));
	int len = strlen(a[0] + 1);
	for(int i = 1; i <= len; i++) {
		for(int j = 0; j < 26; j++) {
			int sta = 0;
			for(int k = 0; k < n; k++) {
				if(a[k][i] == '?' || a[k][i] == j + 'a') sta |= (1 << k);
			}
			g[i][j] = sta;
		}
	}
	
	memset(f, 0, sizeof(f));
	f[0][(1 << n) - 1] = 1;
	for(int i = 0; i < len; i++) {
		for(int s = 0; s < (1 << n); s++) {
			if(!f[i][s]) continue;
			for(int j = 0; j < 26; j++) {
				f[i + 1][s & g[i + 1][j]] = (f[i + 1][s & g[i + 1][j]] + f[i][s]) % mod;
			}
		}	
	}
	for(int s = 0; s < (1 << n); s++) {
		if(__builtin_popcount(s) == k) ans = (ans + f[len][s]) % mod; 
	}

	std::cout << ans << "\n";
	ans = 0;
}

int main() {
    
    int t;
    std::cin >> t;

	while(t--) solve();

	return 0;
}

相關文章