一中OJ #1505 最长词链 | Trie树 / 栈 | 解题报告

一中OJ | #1505 最长词链 | Trie树 / 栈

时限 1000MS/Case 内存 64MB/Case

题目描述

给出一系列单词，每个单词都是一个给定的仅包含小写字母的英语单词表，每个单词至少包含一个字母，至多75个字母。如果在一个由一个单词或多个单词组成表中，除最后一个以外，每个单词都被后一个单词所包含（相等也是包含，例如：单词int 包含 int ），即前一个单词是后一个单词的前缀，则称该词表为一个单词链。例如下面单词组成了一个词链：

i
int
integer

但下面的单词不组成词链：

integer
intern

现在你要做的就是在一个给定的单词表中取出一些词，组成最长的词链，就是包含单词数最多的词链。将它的单词数统计出来。

输入格式

第一行一个整数N，为单词的数量。
接下来N行，每行给出一个单词。

输出格式

一个整数，为最长单词链包含单词数。

样例输入

5

internet
int
integer
i
intern

样例输出

4

数据范围

30%的数据：N<100

50%的数据： N<=1000
100%的数据： N<=20000

样例解释

最长单词词链为： i -> int -> intern -> internet

----------------------------------------------------------

题目分析

比较容易想到的是Trie树模拟链条，dfs获取最长词链的词数

还有一种办法，就是把所有的词进行排序，然后可以知道能够组成词链的词一定在一段连续的区间内

把排序好的词压入栈，如果栈顶词是等待压入的词的前缀，那么直接压入；否则弹出栈顶继续比较

答案就是栈中元素数量最大的时候的元素数（其实跟Trie树的原理差不多啦）

----------------------------------------------------------

代码

#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <vector>
#include <cctype>
#include <iomanip>
#define inf 0x7f7f7f7f
using namespace std;
int np=1,root=0,n;
char s[80],dict[20005][77];
int trie[1500005][27],val[1500005];
void trie_ins(char s[])
{
	int ptr=root;
	for(int i=0;i<strlen(s);i++)
	{
		if(trie[ptr][s[i]-'a']==0)
			trie[ptr][s[i]-'a']=np++;
		ptr=trie[ptr][s[i]-'a'];
	}
	val[ptr]++;
	return;
}
int dfs(int ptr,int ans)
{
	int t=ans;
	ans+=val[ptr];
	for(int i=0;i<26;i++)
	{
		if(trie[ptr][i]!=0)
		{
			t=max(t,dfs(trie[ptr][i],ans));
		}
	}
	return max(ans,t);
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%s",s);
		trie_ins(s);
	}
	printf("%d",dfs(0,0));
	return 0;
}