BZOJ[3172][Tjoi2013]单词 Fail树

传送门ber～

先将所有的串建一个AC自动机，构建的Fail时候到一个点就将它所有的Fail都加上1
emmmm…TLE….

这时就要用到Fail树
将所有Fail指针反向，就可以弄出一个树一样的东西，可以发现，一个点一定是它子树点的后缀（如果串s能匹配到这里，s就会是子树所有点的后缀）

弱鸡说不明白了还是看这个吧QAQ
“这道题想找出每一个串在所有的串中匹配的次数，这时就可以利用fail树的这个性质，而且fail树也是建立在Trie上的，统计子树中所有结点个数，就代表其在所有串中的前缀中作为后缀出现的次数，即在所有串中的出现次数。在队列中越靠后的点就越靠近叶子，用队列倒着更新即可。”——YihAN_Z

代码如下：

#include<cstring>
#include<ctype.h>
#include<cstdio>
#include<queue>
#define N 1000005
using namespace std;
inline int read(){
    int x=0,f=1;char c;
    do c=getchar(),f=c=='-'?-1:f; while(!isdigit(c));
    do x=(x<<3)+(x<<1)+c-'0',c=getchar(); while(isdigit(c));
    return x*f;
}
struct Node{
    Node *ch[26],*nex;
    int cnt;
    Node():nex(NULL),cnt(0){
        for(int i=0;i<26;i++)
            ch[i]=NULL;
    }
}*root=new Node,*tail[205],*s[N];
queue<Node*>q;
char c[N];
int n,top;
struct Aho_Corasick_Automaton{
    inline void Insert(char *s,int id){
        int len=strlen(s+1);
        Node *x=root;
        for(int i=1;i<=len;i++){
            if(!x->ch[s[i]-'a']) x->ch[s[i]-'a']=new Node;
            x=x->ch[s[i]-'a'];
            x->cnt++;
        }
        tail[id]=x;
        return;
    }
    inline void GetFail(){
        for(int i=0;i<26;i++)
            if(root->ch[i]) q.push(root->ch[i]),root->ch[i]->nex=root;
            else root->ch[i]=root;
        while(!q.empty()){
            Node *x=q.front();q.pop();
            s[++top]=x;
            for(int i=0;i<26;i++)
                if(x->ch[i]) x->ch[i]->nex=x->nex->ch[i],q.push(x->ch[i]);
                else x->ch[i]=x->nex->ch[i];
        }
        while(top){
            s[top]->nex->cnt+=s[top]->cnt;
            top--;
        }
    }
}AC;
int main(){
    n=read();
    for(int i=1;i<=n;i++){
        scanf("%s",c+1);
        AC.Insert(c,i);
    }
    AC.GetFail();
    for(int i=1;i<=n;i++)
        printf("%d\n",tail[i]->cnt);
return 0;
}