单词接龙 DFS

原题链接

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

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

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

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

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

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

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

数据范围
n≤20
输入样例：
5
at
touch
cheat
choose
tact
a
输出样例：
23
提示
连成的“龙”为 atoucheatactactouchoose。

一道DFS的题目，要注意的是状态是一整个整体，所以要回溯！

首先计算出，每个单词前后缀重叠的最小单词数，然后dfs一遍开头的单词

#include
#include
using namespace std;

const int N = 25;
int n,ans,used[N],con[N][N];
string a[N];
char head;

void set() //计算每个单词前后缀重叠最小单词数，储存在con数组中
{
	for(int i = 0; i < n; i++)
	{
		for(int j = 0; j < n; j++)
		{
			int len =  min(a[i].size(),a[j].size());
			for(int k = 1; k < len; k++)
			{
				if(a[i].substr(a[i].size() - k) == a[j].substr(0,k))
				{
					con[i][j] = k;
					break;
				}	
			}
		}
	}
}

int dfs(int cur,int len)//cur 当前单词下标，len 当前拼接的长度
{
	ans = max(ans,len);//更新答案
	used[cur]++; //使用次数
	
	for(int i = 0; i < n; i++)
	{
		if(used[i] >= 2)	continue;
		//超过两次
		if(con[cur][i])//能拼接上
		{
			dfs(i,len + a[i].size() - con[cur][i]);//递归
			used[i]--;//回溯
		}
	}
}

int main()
{
	cin >> n;
	for(int i = 0; i < n; i++)	cin >> a[i];
	cin >> head;
	
	set();	
	
	for(int i = 0; i < n; i++)
		if(a[i][0] == head)
			dfs(i,a[i].size());
	
	cout << ans << endl;
	
	return 0;
}