接龙游戏 [Codevs 1051]

AC通道：http://codevs.cn/problem/1051/

[分析]

法一：

看到这道题目，我的第一感觉就是字典树。将输入的每个字符串构建一棵字典树，对于每一个叶子结点，统计以它到根的路径上的节点为末尾的单词的个数，就是它能够接出的最长的龙。结果我就把自己坑了。因为题目所给的空间只有128M，而字典树所需要的空间比较大，所以这种解法是不可行的（至少我打不出来）。可怜我debug一个上午来省空间啊！

80分代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <string>

using namespace std;

int n,sssss=0,maxn=0,cnt;
string s[100001];

struct ss{
	char c;
	int nxt;
	ss(char a=0,int b=-1):c(a),nxt(b){}
};

struct Node{
	vector<ss>child;
	bool yes;
	Node(){yes=false;}
};

int havecheck(vector<ss>a,char b){
	if(a.empty())return false;
	for(vector<ss>::iterator it=a.begin();it!=a.end();it++){
		if((*it).c==b)return (*it).nxt;
	}
	return false;
}

vector<Node>trie;

void make_tree(int x){
//	printf("%s\n",s[x]);
	int l=(s[x]).size(),now=0,tmp;
	if(tmp=havecheck(trie[now].child,s[x][0])){
		now=tmp;
	}
	else{
		Node a;
		trie.push_back(a);
		sssss++;
//		trie[now].child[s[x][0]-'a']=++sssss;
		(trie[now].child).push_back(ss(s[x][0],sssss));
		now=sssss;
	}
	
	for(int i=1;i<l;i++){
		
		if(tmp=havecheck(trie[now].child,s[x][i])){
			now=tmp;
		}
		else{
			Node a;
			trie.push_back(a);
			sssss++;
			(trie[now].child).push_back(ss(s[x][i],sssss));
			now=sssss;
		}
	}
	trie[now].yes=true;
}

void dfs(int now,int tot){
	if(trie[now].yes)tot++;
	if((trie[now].child).empty()){
		if(tot>maxn)maxn=tot;
		return;
	}
	for(vector<ss>::iterator it=(trie[now].child).begin();it!=(trie[now].child).end();it++)
		dfs((*it).nxt,tot);
	if(tot>maxn)maxn=tot;
}

void dfs2(int now){
	if((trie[now].child).empty())return;
	for(vector<ss>::iterator it=(trie[now].child).begin();it!=(trie[now].child).end();it++){
		dfs2((*it).nxt);
	}
}

int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>s[++cnt];
		if(s[cnt]==s[cnt-1])cnt--;
	}
	Node a;
	trie.push_back(a);
	for(int i=1;i<=cnt;i++){
		make_tree(i);
	}
	dfs(0,0);
	dfs2(0);
	cout<<maxn<<endl;
	return 0;
}

法二：

这个方法是我看题解才知道的。这个题目原来是一道栈的简单题……因为可以发现这个单词接龙与我们以往的单词接龙不同，它要求接在后面的单词一定要是前面的单词的前缀，所以我们可以先将单词按照字典序排一个序，这样，能够接上的几个单词就都在一块了。先将第一个单词入栈，再判断第二个单词能否接上。若能接上，则继续入栈。依此类推。直到有一个单词接不上前面的单词，就更新maxn为当前栈的大小，然后退栈，看这个接不上的单词与新的栈顶元素能否接上，如果不能接上，则继续退栈，直到能够接上为止，把这个单词压入栈（如果都接不上它，就让栈中只留它一个）。接下来就都一样了。因为这个方法比较简单，所以我就不附代码了。看来，学东西不能学死板了。有时候，高级算法解决不了的问题，普通算法可以轻松解决呢。