3172: [Tjoi2013]单词
Time Limit: 10 Sec Memory Limit: 512 MBSubmit: 2707 Solved: 1276
[ Submit][ Status][ Discuss]
Description
某人读论文,一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次,现在想知道每个单词分别在论文中出现多少次。
Input
第一个一个整数N,表示有多少个单词,接下来N行每行一个单词。每个单词由小写字母组成,N<=200,单词长度不超过10^6
Output
输出N个整数,第i行的数字表示第i个单词在文章中出现了多少次。
Sample Input
3
a
aa
aaa
a
aa
aaa
Sample Output
6
3
1
3
1
HINT
Source
【题目分析】
思路十分巧妙,每次先匹配再插入会超时的,记录下标之后,从较深的结点向上递推累加,画出图来就很容易理解了。刚开始数组开小了,TLE(好奇竟然不是RE)。
【代码】
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
int cnt=0,list[1000005],fail[1000005],trie[1000005][26],pos[2001],v[1000005],tot=1,n;
char s[1000001];
inline void insert(int num)
{
scanf("%s",s);
int now=1,l=strlen(s)-1;
for (int i=0;i<=l;++i)
{
if (!trie[now][s[i]-'a']) trie[now][s[i]-'a']=++tot;
now=trie[now][s[i]-'a'];
v[now]++;//经过该节点的数目
}
pos[num]=now;//记录每一个字符串的最终位置
}
inline void bfs()
{
list[++cnt]=1;
queue<int>q;
q.push(1);
while(!q.empty()){
int x=q.front(); q.pop();
list[++cnt]=x;//记录bfx序
for (int i=0;i<=25;++i){
int j=fail[x];
while (j&&!trie[j][i]) j=fail[j];
if (trie[x][i]) {fail[trie[x][i]]=j?trie[j][i]:1;q.push(trie[x][i]);}
else trie[x][i]=j?trie[j][i]:1;
}
}
}
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;++i) insert(i);
bfs();
for (int i=cnt;i>=1;i--){
v[fail[list[i]]]+=v[list[i]];
}//逆bfs序递推
for (int i=1;i<=n;++i){
printf("%d\n",v[pos[i]]);
}
}