蓝桥杯 ALGO-18 单词接龙

  算法训练 单词接龙  

时间限制：1.0s   内存限制：256.0MB

      

问题描述 

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

输入格式 

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

输出格式 

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

样例输入
　　5
　　at
　　touch
　　cheat
　　choose
　　tact
　　a

样例输出

23

样例说明

　　连成的“龙”为atoucheatactactouchoose

分析：从首字母为开头字母的字符串开始，进行深搜。对于每一个字符串，判断是否可以接龙的标准是：从前一个字符串的最后一个字符开始向前遍历，若存在字符与后面的字符串的首字母相同，则从此处开始，与后一个字符串的字符两两比较。若直到前一个字符串遍历完成仍未出现不同字符，则两字符串可对接，对接点为第二个字符串此时的索引。

对接方法：取第二个字符串的索引的子串。这样可完成对接，也可滤去有包含关系的字符串（这样对接完成相当于是取较大的那个字符串）。

代码：

#include<iostream>
#include<algorithm>
#include<vector>
#include<string>
using namespace std;
int n;
int ans = 0;
vector<int> visit;
vector<string> vs;
int check(string s, int index) {
	string sub = vs[index];
	for (int i = s.size() - 1; i >= 0; i--) {
		if (sub[0] == s[i]) {
			int flag = 1;
			int j = i + 1, k = 1;
			for (; j < s.size(); j++, k++) {
				if (s[j] != sub[k]) {
					flag = 0;
					break;
				}
			}
			if (flag == 1) {
				return k;
			}
		}
	}
	return 0;
}
void dfs(string s) {
	int size = s.size();
	ans = max(ans, size);
	for (int i = 0; i < n; i++) {
		if (visit[i] != 2) {
			int len = check(s, i);
			if (len) {
				visit[i]++;
				string sub = string(vs[i], len);
				dfs(s + sub);
				visit[i]--;
			}
		}
	}
}
int main() {
	cin >> n;
	vs.resize(n);
	visit = vector<int>(n);
	for (int i = 0; i < n; i++) {
		cin >> vs[i];
	}
	string head;	
	cin >> head;
	for (int i = 0; i < n; i++) {
		if (head[0] == vs[i][0]) {
			visit[i]++;
			dfs(vs[i]);
			visit[i]--;
		}
	}
	cout << ans << endl;
	return 0;
}