免責事項:この記事はブロガーオリジナル記事です、続くBY-SAのCC 4.0を著作権契約、複製、元のソースのリンクと、この文を添付してください。
タイトル説明
紙を読んで誰かが、紙は多くの単語で構成されています。しかし、彼は何度も言葉を紙に表示され、今で各単語を紙に登場した回数を知りたいました。
入力説明:
最初の整数N、一つの単語のNライン続くワードの数、。各単語の小文字、N≤200は、ワード長が10 ^ 6を超えません
出力説明:
出力Nの整数、デジタルi行目には、記事で何度も登場し、私語を表します。``
あなただけが祈る木のように、失敗ツリーを構築する必要があります
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAXN = 1e6+50;
const int MOD = 1e9+7;
inline int read(){
int x=0,f=1; char ch=getchar();
while(!isdigit(ch)){ if(ch=='-') ch=-1; ch = getchar(); }
while(isdigit(ch)) x=x*10+ch-48,ch=getchar();
return x*f;
}
struct Trie{
int nxt[MAXN][26],fail[MAXN],cnt[MAXN],id[MAXN];
vector<int> g[MAXN];
int sz;
void Insert(char *s,int k){
int root=0,len=strlen(s);
for(int i=0;i<len;i++){
if(!nxt[root][s[i]-'a']) nxt[root][s[i]-'a'] = ++sz;
root = nxt[root][s[i]-'a'];
cnt[root]++;
}
id[k] = root;
}
void Build(){
queue<int> que;
for(int i=0;i<26;i++) if(nxt[0][i]) que.push(nxt[0][i]);
while(!que.empty()){
int u = que.front(); que.pop();
for(int i=0;i<26;i++){
if(nxt[u][i]) {
que.push(nxt[u][i]);
fail[nxt[u][i]] = nxt[fail[u]][i];
}
else nxt[u][i] = nxt[fail[u]][i];
}
}
}
void dfs(int u){
for(int i=0;i<(int)g[u].size();i++){
int v = g[u][i];
dfs(v);
cnt[u] += cnt[v];
}
}
void solve(int n){
for(int i=1;i<=sz;i++) g[fail[i]].push_back(i);
dfs(0);
for(int i=1;i<=n;i++) printf("%d\n",cnt[id[i]]);
}
}Automaton;
char str[MAXN];
int main(){
int n = read();
for(int i=1;i<=n;i++) scanf("%s",str),Automaton.Insert(str,i);
Automaton.Build();
Automaton.solve(n);
return 0;
}