算法训练 单词接龙
时间限制:1.0s 内存限制:256.0MB
问题描述
单词接龙是一个与我们经常玩的成语接龙相类似的游戏,现在我们已知一组单词,且给定一个开头的字母,要求出以这个字母开头的最长的“龙”(每个单词都最多在“龙”中出现两次),在两个单词相连时,其重合部分合为一部分,例如 beast和astonish,如果接成一条龙则变为beastonish,另外相邻的两部分不能存在包含关系,例如at 和 atide 间不能相连。
输入格式
输入的第一行为一个单独的整数n (n<=20)表示单词数,以下n 行每行有一个单词,输入的最后一行为一个单个字符,表示“龙”开头的字母。你可以假定以此字母开头的“龙”一定存在.
输出格式
只需输出以此字母开头的最长的“龙”的长度
样例输入
5
at
touch
cheat
choose
tact
a
样例输出
23
样例说明
连成的“龙”为atoucheatactactouchoose
分析:从首字母为开头字母的字符串开始,进行深搜。对于每一个字符串,判断是否可以接龙的标准是:从前一个字符串的最后一个字符开始向前遍历,若存在字符与后面的字符串的首字母相同,则从此处开始,与后一个字符串的字符两两比较。若直到前一个字符串遍历完成仍未出现不同字符,则两字符串可对接,对接点为第二个字符串此时的索引。
对接方法:取第二个字符串的索引的子串。这样可完成对接,也可滤去有包含关系的字符串(这样对接完成相当于是取较大的那个字符串)。
代码:
#include
#include
#include
#include
using namespace std;
int n;
int ans = 0;
vector visit;
vector 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(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;
}