zufeoj 单词接龙 DFS

问题 E: 单词接龙

题目链接

http://acm.ocrosoft.com/problem.php?cid=2876&pid=4

题目描述

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

输入格式

输入的第一行为一个单独的整数n(n≤20)表示单词数，以下n行每行有一个单词（只含有大写或小写字母，长度不超过20），输入的最后一行为一个单个字符，表示“龙”开头的字母。你可以假定以此字母开头的“龙”一定存在。

输出格式 

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

输入样例  

5
at
touch
cheat
choose
tact
a

输出样例  

23

 思路

用两个二维数组来分别存放两个单词是否能接龙，前后两个单词重合部分长度。

再遍历一遍单词，找出能作为头的单词，并开始搜索。

代码  

#include <iostream>
#include <cstring>

using namespace std;

string word[25];
bool st[25][25];//s[i][j]表示 i j 是否能接龙
int g[25][25];//记录重合部分长度
int f[25];//记录单词用了几次
char head;
int n,ans;

bool check(string a,string b)//判断 a b 能否接龙
{
    int lena=a.size(),lenb=b.size();
    int l=min(lena,lenb);

    for(int len=l-1;len>=1;len--)
        if(a.substr(lena-len,lena-1)==b.substr(0,len))return 1;

    return 0;
}

int samepart(string a,string b)//计算 a b 重合部分长度
{
    int lena=a.size(),lenb=b.size();
    int l=min(lena,lenb);

    for(int len=l-1;len>=1;len--)
        if(a.substr(lena-len,lena-1)==b.substr(0,len))return len;

    return 0;
}

void dfs(int last,string dragon)
{
    ans=max(ans,(int)dragon.size());
    for(int i=1;i<=n;i++)
    {
        if(g[last][i]&&f[i]<2)
        {
            f[i]++;
            dfs(i,dragon+word[i].substr(g[last][i]));
            f[i]--;
        }
    }
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)cin>>word[i];
    cin>>head;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            st[i][j]=check(word[i],word[j]);
            g[i][j]=samepart(word[i],word[j]);
        }
    }

    for(int i=1;i<=n;i++)
    {
        if(word[i][0]==head)
        {
            f[i]++;
            dfs(i,word[i]);
            f[i]--;
        }
    }

    cout<<ans<<endl;
    return 0;
}