状态压缩动规--洛谷P1278单词游戏

单词游戏

洛谷链接

题目描述

Io 和 Ao 在玩一个单词游戏。

他们轮流说出一个仅包含元音字母的单词，并且后一个单词的第一个字母必须与前一个单词的最后一个字母一致。

游戏可以从任何一个单词开始。

任何单词禁止说两遍，游戏中只能使用给定词典中含有的单词。

游戏的复杂度定义为游戏中所使用的单词长度总和。

编写程序，求出使用一本给定的词典来玩这个游戏所能达到的游戏最大可能复杂度。

输入格式

输入文件的第一行，表示一个自然数 $N(1\le N\le 16)$，$N$ 表示一本字典中包含的单词数量以下的每一行包含字典中的一个单词，每一个单词是由字母 A、E、I、O 和 U 组成的一个字符串，每个单词的长度将小于等于 $100$，所有的单词是不一样的。

输出格式

输出文件仅有一行，表示该游戏的最大可能复杂度。

样例 #1

样例输入 #1

5
IOO
IUUO
AI
OIOOI
AOOI

样例输出 #1

16

---

思路：状态压缩，二进制整数位置i表示第i个字符串是否使用，为1即为使用

定义状态 f[k,j]表示状态为k且最后一个字符串为时所能表示的字符串最大长度

一次枚举每一层状态，当前字符串i以及 可拼接字符串 j

使用按位与计算当前字符串是否能被选择

(k>>i-1)&1,(k>>j-1)&1

需满足题意i的最后一个字母与j 开头相同

即s[i,s[i].size()-1]==s[j,0]

最终状态转移方程为 f[k+1<<j-1，j]=max(f[k+1<<j-1，j]，f[k,i]+s[i].size());

表示当前字符串是否拼接，拼接后最大长度更新

注意本题需要初始化,即对任意一个字符串i初始化为f[1<<i-1,i]=s[i].size();

表示此仅字符串使用下初始长度即为自身长度

code

#include<iostream>
#include<algorithm>
#include<unordered_map>
#include<cstring>
using namespace std;
const int N=1<<16;
int f[N][20];
string s[20];
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>s[i];
        f[1<<i-1][i]=s[i].size();//状态压缩数组初始化
    }
    for(int k=0;k<1<<n;k++){
        for(int j=1;j<=n;j++){
            for(int i=1;i<=n;i++){
                if(i!=j&&s[j][s[j].size()-1]==s[i][0]&&(k>>j-1&1)&&!(k>>i-1&1)){//注意条件，我们枚举的是使用了j结尾，准备拼接i
                    f[k+(1<<i-1)][i]=max(f[k+(1<<i-1)][i],f[k][j]+(int)s[i].size());
                    //枚举状态转移方程
                }
            }
        }
    }
    int ans=0;
    for(int i=0;i<1<<n;i++){
        for(int j=1;j<=n;j++){
            ans=max(ans,f[i][j]);//枚举求得最大值
        }
    }
    cout<<ans<<endl;
    return 0;
}