洛谷p1019 单词接龙

题目描述

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

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1

5
at
touch
cheat
choose
tact
a

输出样例#1：

23

思路：先找出两个单词的最小公共部分（前一个单词的后缀和后一个单词的前缀最小的重叠部分），然后搜索，搜索的时候先找单词第一个字母是给出的字母的，搜索后面的时候要保证相邻的两部分不能存在包含关系（重叠部分长度小于要相接的两个单词的长度），还有就是一个单词最多使用两次

#include<iostream>
#include<cstring>
#include<string>
using namespace std;
string a[20];
int cl[20][20];//cl保存两个单词的最小重叠长度
int visit[20];
int n;
int ans=0;
int commonlenth(string a,string b)  //计算两个单词的最小重叠长度
{
	int i,j,k;
	int minlen=0x3f3f3f3f;
	for(k=1;k<=a.length();k++)
	{
		i=a.length()-1;
		j=k-1;
		int flag=0;
		while(i>=0&&j>=0)
		{
			if(a[i]==b[j])
			{
				i--;
				j--;
			}
			else
			{
				flag=1;
				break;
			}
		}
		if(!flag)
		minlen=min(minlen,k);
	}
	return minlen==0x3f3f3f3f?0:minlen;
	
}
void dfs(int lenth,int pre)
{
	ans=max(ans,lenth);
	for(int i=0;i<n;i++)
	{
		if(visit[i]<2) //每个单词最多使用两次
		{
			if(cl[pre][i]>=1&&cl[pre][i]<a[i].length()&&cl[pre][i]<a[pre].length())//相邻两个单词要有重叠部分，并且不能包含
			{
				visit[i]++;
				dfs(lenth+a[i].length()-cl[pre][i],i);
				visit[i]--;
			}
		}
	}
}
int main(){
	memset(visit,0,sizeof(visit));
	memset(cl,0,sizeof(cl));
	cin>>n;
	int i,j;
	for(i=0;i<n;i++)
	{
		cin>>a[i];
	}
	int x,y;
	for(x=0;x<n;x++)
	{
		for(y=0;y<n;y++)
		{
			cl[x][y]=commonlenth(a[x],a[y]); 
		}
	}
	char s;
	cin>>s;
	for(i=0;i<n;i++)
	{
		string temp=a[i];
		if(temp[0]==s)    //先确定排头的第一个单词
		{
			visit[i]++;
			dfs(temp.length(),i);
			visit[i]--;
		}
	}
	cout<<ans<<endl;
}