P1019 单词接龙

点击查看代码

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>

using namespace std;

int n;
string words[25];
int g[25][25];  //记录每两个单词重叠的长度
int vis[25];    //记录每一个单词使用的个数
int ans = 0;    //答案

//记录单词a，b的最小重叠长度，如果不存在返回0
int get_overlap(string a, string b) {
    // k要小于两个单词中最小的哪一个，这样就避免了完全重合
    for (int k = 1; k < min(a.length(), b.length()); k++) {
        
        if (a.substr(a.length() - k) == b.substr(0, k)) {
            return k; 
        }
    }
    return 0; // 没找到或者存在包含关系
}

//1.状态定义，now当前处理的单词，length，当前累加的长度
void dfs(int now, int length) {
    
    ans = max(ans, length);

    // 2.枚举选择，与其他所有单词的重叠长度
    for (int i = 0; i < n; i++) {
        //选择的条件①有重叠长度②使用不超过两次
        if (g[now][i] > 0 && vis[i] < 2) {
            vis[i]++; // 标记使用
            
            
            dfs(i, length + words[i].length() - g[now][i]);
            //3.回溯
            vis[i]--; // 回溯
        }
    }
}
//没有截止条件，我们不知道龙最后有多长，也不知道要用到多少单词，这题的截止条件就是举目四望，无路可走，在for循环的if语句中一次也进不去
//也没有剪枝，因为求最小值好减，求最大值难，并且这道题实际已经做了不重合剪枝了（已经做过了，只是没有在对ans做处理，显得没做

int main() {
    cin >> n;
    for (int i = 0; i < n; i++) {
        cin >> words[i];
    }
    
    char start_char;
    cin >> start_char; // 读取龙头字母

    //步骤1：预处理所有单词的重叠长度，这样就不用在dfs中反复比较了，节约时间
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            g[i][j] = get_overlap(words[i], words[j]);
        }
    }

   //步骤2：寻找龙头并启动dfs，看看哪些单词可以做龙头
    for (int i = 0; i < n; i++) {
        if (words[i][0] == start_char) {
            vis[i]++; 
            
            // 初始长度就是龙头单词的长度
            dfs(i, words[i].length()); 
            
            vis[i]--; // 回溯 (这一步其实在 main 里不写也不影响，但保持习惯很好)
        }
    }

    cout << ans << endl;

    return 0;
}