cf#302 E. Remembering Strings 状压dp
题意:给定n个长度均为m的字符串,下面n行给出字符串,下面n*m的矩阵表示把对应的字母修改成其他字母的花费。
对于一个字符串,若它是easy to remembering 当 它存在一个字母,使得这个字母在这一列是独一无二的。
要使得n个字符串都是easy to remembering 的最小花费。
第一个样例是把第一列的4个a中3个a修改成别的字母,所以花费为3.
思路:
因为最多只有20个字符串,也就是修改后的字母种类至多只有20种。
然后状压已经easy to remembering的字符串的最小花费。
dp[i] 表示已经easy to remembering 的字符串状态为i时的最小花费。
两个转移:
1、直接修改字母
2、把这一列中所有与这个字母相同的字母都修改成别的字母。
当然可以剩下一个,剩下花费最大的那个即可。
cost[i][j] 就表示除了花费最大的那个 同列中与str[i][j]字母相同的花费和。
bit[i][j] 表示哪些字符串 在第j列 与 a[i][j] 字母相同。
代码:
1 #include <bits/stdc++.h> 2 using namespace std; 3 4 char s[30][30]; 5 int a[30][30]; 6 int bit[30][30],cost[30][30],dp[(1<<21)+1]; 7 8 int main(){ 9 int n,m; cin>>n>>m; 10 for(int i=0; i<n; i++) 11 cin >> s[i]; 12 for(int i=0; i<n; i++) 13 for(int j=0; j<m; j++) 14 cin >> a[i][j]; 15 for(int i=0; i<n; i++) 16 for(int j=0; j<m; j++){ 17 int ans=0, mx=-1; 18 for(int k=0; k<n; k++){ 19 if(s[i][j] == s[k][j]){ 20 ans += a[k][j]; 21 mx = max(mx,a[k][j]); 22 bit[i][j] |= (1<<k); 23 } 24 } 25 26 cost[i][j] = ans - mx; 27 } 28 29 memset(dp,0x3f,sizeof(dp)); 30 dp[0] = 0; 31 for(int i=0; i<(1<<n); i++){ 32 for(int j=0; j<n; j++){ 33 if((i&(1<<j)) == 0){ 34 for(int k=0; k<m; k++){ 35 dp[i|(1<<j)] = min(dp[i|(1<<j)],dp[i]+a[j][k]); 36 dp[i|bit[j][k]] = min(dp[i|bit[j][k]],dp[i]+cost[j][k]); 37 } 38 } 39 } 40 } 41 42 cout << dp[(1<<n)-1] << endl; 43 }