[TJOI2013]单词

Description
某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

Input
第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6

Output
输出N个整数，第i行的数字表示第i个单词在文章中出现了多少次。

Sample Input
3
a
aa
aaa

Sample Output
6
3
1

---

直接对所有单词建立AC自动机，然后再把每个串去AC自动机里面跑一下更新答案，更新答案的时候记得打标记，最后再一次更新fail指针上所有的节点，否则时间复杂度退化，直接TLE

/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 0x7f7f7f7f
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline char gc(){
    static char buf[1000000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
inline int frd(){
    int x=0,f=1;char ch=gc();
    for (;ch<'0'||ch>'9';ch=gc())   if (ch=='-')    f=-1;
    for (;ch>='0'&&ch<='9';ch=gc()) x=(x<<1)+(x<<3)+ch-'0';
    return x*f;
}
inline int read(){
    int x=0,f=1;char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar())  if (ch=='-')    f=-1;
    for (;ch>='0'&&ch<='9';ch=getchar())    x=(x<<1)+(x<<3)+ch-'0';
    return x*f;
}
inline void print(int x){
    if (x<0)    putchar('-'),x=-x;
    if (x>9)    print(x/10);
    putchar(x%10+'0');
}
const int N=1e6;
int Ans[N+10];
vector<int>pos[N+10];
struct S1{
    int trie[N+10][26],fail[N+10],End[N+10],h[N+10],cnt[N+10];
    int tot,root;
    void insert(char *s,int ID){
        int len=strlen(s),p=root;
        for (int i=0;i<len;i++){
            if (!trie[p][s[i]-'a']) trie[p][s[i]-'a']=++tot;
            p=trie[p][s[i]-'a'];
        }
        if (!End[p])    End[p]=ID;
        pos[End[p]].push_back(ID);
    }
    void make_fail(){
        int head=1,tail=0;
        for (int i=0;i<26;i++)  if (trie[root][i])  h[++tail]=trie[root][i];
        for (;head<=tail;head++){
            int Now=h[head];
            for (int i=0;i<26;i++){
                if (trie[Now][i]){
                    int son=trie[Now][i];
                    fail[son]=trie[fail[Now]][i];
                    h[++tail]=son;
                }else   trie[Now][i]=trie[fail[Now]][i];
            }
        }
    }
    void check(char *s,int len){
        int p=root;
        for (int i=0;i<len;i++){
            p=trie[p][s[i]-'a'];
            cnt[p]++;
//          for (int x=p;x;x=fail[x])   Ans[End[x]]++;
        }
    }
    void update(){
        for (int i=tot;i;i--)   cnt[fail[h[i]]]+=cnt[h[i]];
        for (int i=1;i<=tot;i++)    Ans[End[i]]=cnt[i];
    }
}AC;//Aho-Corasick automation
char T[N+10];
int Len[N+10];
int main(){
    int n=read(),lenT=0;
    for (int i=1;i<=n;i++){
        scanf("%s",T+lenT);
        AC.insert(T+lenT,i);
        lenT+=(Len[i]=strlen(T+lenT));
    }
    AC.make_fail();
    lenT=0;
    for (int i=1;i<=n;i++){
        static char s[N+10];
        for (int j=0;j<Len[i];j++)  s[j]=T[j+lenT];
        lenT+=Len[i];
        AC.check(s,Len[i]);
    }
    AC.update();
    for (int i=1;i<=n;i++){
        if (!Ans[i])    continue;
        for (vector<int>::iterator it=pos[i].begin();it!=pos[i].end();it++) Ans[*it]=Ans[i];
    }
    for (int i=1;i<=n;i++)  printf("%d\n",Ans[i]);
    return 0;
}

转载于:https://www.cnblogs.com/Wolfycz/p/10485598.html