一中OJ | #1505 最长词链 | Trie树 / 栈
时限 1000MS/Case 内存 64MB/Case
题目描述
i
int
integer
但下面的单词不组成词链:
integer
intern
现在你要做的就是在一个给定的单词表中取出一些词,组成最长的词链,就是包含单词数最多的词链。将它的单词数统计出来。
输入格式
第一行一个整数N,为单词的数量。
接下来N行,每行给出一个单词。
输出格式
一个整数,为最长单词链包含单词数。
样例输入
5
internetint
integer
i
intern
样例输出
4
数据范围
30%的数据:N<100
50%的数据: N<=1000100%的数据: 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;
}