P1019 [NOIP2000 提高组] 单词接龙

题目

P1019 [NOIP2000 提高组] 单词接龙

分析

首先解决重合长度问题。

我们可以枚举重合长度从1到max(lena,lenb)-1（不能包含），截取子串来判断。

（纠正一下：题目中的包含是指串a和串b接龙后长度没有改，比如babyb和byb可以接龙。）

// 重合长度
int catch_len(string a, string b){
    for(int len=1;len<max(a.size(),b.size());len++)
        if(b.substr(0,len)==a.substr(a.size()-len,len)) return len;
    return 0;
}

设数组f[i][j]表示第i个单词和第j个单词的重合长度，我们预处理出f数组，找出符合开头的单词，dfs即可。

代码

#include<iostream>
using namespace std;
const int N=50;
string a[N];
int vis[N],n,f[N][N],ans=0;
int catch_len(string a, string b){
    for(int len=1;len<min(a.size(),b.size());len++){
        if(b.substr(0,len)==a.substr(a.size()-len,len)) return len;
    }
    return 0;
}
void dfs(int size, int last){
    ans=max(ans,size);
    for(int i=0;i<n;i++){
        if(f[last][i]!=0&&vis[i]<2){
            vis[i]++;
            dfs(size+a[i].size()-f[last][i],i);
            vis[i]--;
        }
    }
}
int main(){
    char s;
    cin>>n;
    for(int i=0;i<n;i++) cin>>a[i];
    cin>>s;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++) f[i][j]=catch_len(a[i],a[j]);
        
    for(int i=0;i<n;i++){
        if(a[i][0]==s){
            vis[i]=1;
            dfs(a[i].size(),i);
            vis[i]=0;
        }
    }
    cout<<ans<<endl;
    return 0;
}