洛谷P1019（单词接龙）

题目描述

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

输入格式

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

输出格式

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

解题思路

  这一题要看清楚题目哦，每个单词可能出现2次，然后还有个隐藏条件没说的，就是龙头不一定只有一个啊，比如龙头的字母是a，但是有单词able和apple，那么这两个单词都能作为龙头。
  深度遍历尽量别用递归，虽然这一题的单词量小，但是我学ac的初衷是提高自己的解决问题的思路水平，万一以后需要处理大数据量的相同题目，过度依赖递归就显然违背了我的初衷，所以果断堆栈寄存器，别说，用堆栈的实现方法运行时间也足够优秀。可是有个问题，这一题的dfs和普通dfs有点不一样的就是每个点允许经过两次，那么我定义了一个begin，初始值为-1，每一次取数时从begin开始取数，如果那个单词可取，即使用次数少于2，且与上一次当前遍历的堆栈寄存器的栈顶单词有交叉部分，那么就将其入栈，且更新begin为-1，如果取不到数，就跟新begin为堆栈的栈顶单词的序号，堆栈做一次退栈处理。直到栈为空。
  Talking is cheap,show you my code.

代码

#include<iostream>
#include<string>
#include<algorithm>
#include<stack>
using namespace std;
string word[30];
int jc[30][30];
int use[30];
stack<int> sw;
int n;
char c;
int lianjie(int i, int j) {
	string a = word[i];
	string b = word[j];
	int size = min(a.size(), b.size());
	for (int i = 1;i < size;i++) {
		bool flag = true;
		for (int j = 0;j < i;j++) {
			if (a[a.size() - i + j] != b[j]) {
				flag = false;
			}
		}
		if (flag) {
			return i;
		}
	}
	return 0;
}
void yuchuli(int n) {
	for (int i = 0;i < n;i++) {
		for (int j = 0;j < n;j++) {
			jc[i][j] = lianjie(i, j);
		}
	}
}
int dfs(int i) {
	int len = 0;
	int l = 0;
	stack<int> mystack;
	use[i]++;
	mystack.push(i);
	len += word[i].size();
	l = len;
	int begin = -1;
	while (true) {
		int j = begin + 1;
		while (j < n) {
			if ((use[j]<2) && (jc[mystack.top()][j])) {
				break;
			}
			j++;
		}
		if (j >= n) {
			len = max(len, l);
			begin = mystack.top();
			use[begin]--;
			l -= word[begin].size();
			mystack.pop();
			if (mystack.empty()) {
				break;
			}
			l += jc[mystack.top()][begin];
			if ((mystack.top() == i)&&(mystack.size()==1)) {
				break;
			}
		}
		else {
			begin = -1;
			use[j]++;
			l -= jc[mystack.top()][j];
			mystack.push(j);
			l += word[j].size();
		}
	}
	use[i] = 0;
	return len;
}
int main() {
	cin >> n;
	for (int i = 0;i < n;i++) {
		cin >> word[i];
	}
	cin >> c;
	yuchuli(n);
	for (int i = 0;i < n;i++) {
		if (word[i][0] == c) {
			sw.push(i);
		}
	}
	int len = 0;
	while (!sw.empty()) {
		len = max(len, dfs(sw.top()));
		sw.pop();
	}
	cout << len << endl;
	return 0;
}