洛谷 P1019 单词接龙 深搜

题目描述

单词接龙是一个与我们经常玩的成语接龙相类似的游戏，现在我们已知一组单词，且给定一个开头的字母，要求出以这个字母开头的最长的“龙”（每个单词都最多在“龙”中出现两次），在两个单词相连时，其重合部分合为一部分，例如 beast和astonish，如果接成一条龙则变为beastonish，另外相邻的两部分不能存在包含关系，例如at 和 atide 间不能相连。

输入输出格式

输入格式：

输入的第一行为一个单独的整数n (n<=20)表示单词数，以下n 行每行有一个单词，输入的最后一行为一个单个字符，表示“龙”开头的字母。你可以假定以此字母开头的“龙”一定存在.

输出格式：

只需输出以此字母开头的最长的“龙”的长度

输入输出样例

输入样例#1：

5
at
touch
cheat
choose
tact
a

输出样例#1：

23           （连成的“龙”为atoucheatactactouchoose）   

思路：

将单词看成点，如果 a 单词后面能接上 b 单词，则 a 到 b 有一条边，边的长度为 a 后面和 b 前面最短的公共部分，这样能使最后的长度最长。

#include <iostream>
#include <cstring>
using namespace std;

const int MAX = 25;

char s[MAX][2005];
int G[MAX][MAX];        //可以将问题转化成图 
int vis[MAX];
int n;
char start;
int ans;

int getEdge(char *s1, char *s2);    // s1 是否可以到 s2，若可以，返回重叠部分的字符数，若不可以，返回 0

void dfs(int i, int len);

int main(){
//    freopen("input.txt", "r", stdin);
    
    cin >> n;
    for(int i=1; i<=n; i++){
        cin >> s[i];
    }
    //初始化图
    for(int i=1; i<=n; i++){
        for(int j=1; j<=n; j++){
            G[i][j] = getEdge(s[i], s[j]);
        }
    }
    
    //深搜 
    cin >> start;
    ans = 0;
    for(int i=1; i<=n; i++){
        if(s[i][0] == start){
            memset(vis, 0, sizeof(vis));
            dfs(i, strlen(s[i]));
        }
    }
    
    cout << ans;
    
    return 0; 
} 

int getEdge(char *s1, char *s2){
    //从 s1 第二个字母开始，s2 不能包含在 s1 中
    int len1 = strlen(s1), len2 = strlen(s2);
    for(int i=len1-1; i>=1; i--){        //公共部分越少越好 
        int n1, n2;    
        for(n1=i, n2=0; n1<len1 && n2<len2; n1++, n2++){
            if(s1[n1] != s2[n2])
                break;
        }
        if(n1 == len1 && n2 < len2)
            return strlen(s2) - n2;
    } 
    return 0;
}

void dfs(int i, int len){
    vis[i]++;
    bool isEnd = true;
    for(int j=1; j<=n; j++){
        if(G[i][j] != 0 && vis[j] < 2){
            dfs(j, len + G[i][j]);
            vis[j]--;
            isEnd = false;
        }
    }
    
    if(isEnd)
        ans = max(ans, len);
    
}