洛谷__P1278 单词游戏

题目链接：P1278 单词游戏 - 洛谷

---

题目大意：

给出 n 个元音字母的单词，把这些单词拼接，要求：后一个单词的首字母 = 前一个单词的尾字母

求：拼接后最大的长度；

---

思路：

状压DP

---

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<vector>
#include<queue>
#include<deque>
#include<stack>
#include<set>
#include<map>
#include<unordered_set>
#include<unordered_map>
#include<bitset>
#include<tuple>
#define inf 72340172838076673
#define int long long
#define endl '\n'
#define F first
#define S second
#define  mst(a,x) memset(a,x,sizeof (a))
using namespace std;
typedef pair<int, int> pii;

const int N = 17, mod = 998244353;

int n;
int f[N][1 << N];//使用 i 个单词 j的状态下拼接的长度
string s[N];
bool st[N];
int res = 0;

int dfs(int pos, int len, int us) {

    if (f[pos][us]) return f[pos][us];//这几个单词拼接过直接返回这几个的长度，剪枝
    f[pos][us] = len;
    
    for (int i = 1; i <= n; i++) {
        if (st[i] || pos && s[i][0] != s[pos].back()) continue;
        
        st[i] = true;
        dfs(i, len + s[i].size(), us | (1 << i));
        st[i] = false;//回溯
    }
    
    res = max(res, f[pos][us]);//最大值
    return res;
}

void solve() {
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> s[i];
    
    cout << dfs(0, 0, 0) << endl;
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr), cout.tie(nullptr);
    
    int T = 1;
// cin >> T;
    while (T--) solve();
    
    return 0;
}