单词接龙 字符串函数+dfs

单词接龙是一个与我们经常玩的成语接龙相类似的游戏。

现在我们已知一组单词，且给定一个开头的字母，要求出以这个字母开头的最长的“龙”，每个单词最多被使用两次。

在两个单词相连时，其重合部分合为一部分，例如 beast 和 astonish ，如果接成一条龙则变为 beastonish。

我们可以任意选择重合部分的长度，但其长度必须大于等于1，且严格小于两个串的长度，例如 at 和 atide 间不能相连。

输入格式

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

你可以假定以此字母开头的“龙”一定存在。

输出格式

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

数据范围

n≤20，
单词随机生成。

输入样例：

5
at
touch
cheat
choose
tact
a

输出样例：

23

提示

连成的“龙”为 atoucheatactactouchoose。

 

#include<iostream>
#include<cstring>
using namespace std;
const int N=1010;
int n;
string word[N];
int g[N][N];
int cnt[N];
int res;
void dfs(string ans,int id){
	res=max(res,(int)ans.size());
	
	cnt[id]++;
	for(int i=1;i<=n;i++)
	 if(g[id][i]&&cnt[i]<2)dfs(ans+word[i].substr(g[id][i]),i);
	
	cnt[id]--;
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)cin>>word[i];
	
	char str;
	cin>>str;
	
	
	for(int i=1;i<=n;i++)
	 for(int j=1;j<=n;j++)
	  {
	  	string a=word[i],b=word[j];
	  	for(int k=1;k<=min(a.size(),b.size());k++){
	  		if(a.substr(a.size()-k,k)==b.substr(0,k)){
	  			g[i][j]=k;
	  			break;
			  }
		  }
	  }
	  
	for(int i=1;i<=n;i++)
	 if(word[i][0]==str)dfs(word[i],i);
	 
	cout<<res;
}