单词接龙~

蓝桥杯算法训练VIP-单词接龙

时间限制: 1s 内存限制: 128MB

题目描述

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

样例说明
连成的“龙”为atoucheatactactouchoose

输入格式

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

输出格式

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

样例输入

5
at
touch
cheat
choose
tact
a

样例输出

23

代码

#include <bits/stdc++.h>
using namespace std;
int n;
string wrd[40];
int flag[40];
char start;
int sum,m;
void dfs(int x){
	sum=max(sum,m);
	
	for(int i=1;i<wrd[x].length();i++){//从该字符串的第二个字符开始
		for(int j=0;j<2*n;j++){//找未标记的、可以接龙的单词
		
		    if(flag[j]==0&&wrd[j][0]==wrd[x][i]){
		    	
		    	int u=i;
				int v=0;
				int l1=wrd[x].length();
				int l2=wrd[j].length();
		    	while(u<l1&&v<l2-1){//查看后续字母，注意下标范围
		    		if(wrd[x][u]!=wrd[j][v]){
		    			break;
					}
					u++;v++;
				}
				if(u==l1&&v<=l2-1){//注意范围
					m+=(wrd[j].length()-v);//v其实就是相同字母的个数
					flag[j]=1;
					dfs(j);
					flag[j]=0;
					m-=(wrd[j].length()-v);
				}
				
			}
		}
	}
}
int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>wrd[i];
		wrd[n+i]=wrd[i];
	}
	cin>>start;
	
	for(int i=0;i<n;i++){
		if(wrd[i][0]==start){
			memset(flag,0,sizeof(flag));//注意初始化
			
			flag[i]=1;//标记
			m=wrd[i].length();
			dfs(i);
			m=0;
			flag[i]=0;
			
		}
	}
	cout<<sum<<endl;
	return 0;
}

总结

本题是有关字符串的dfs，做了很久…，因为字符串掌握的不太好，要注意的地方写在注释里了^^