【bzoj3172】: [Tjoi2013]单词 字符串-AC自动机

【bzoj3172】: [Tjoi2013]单词

先用所有单词构造一个AC自动机

题目要求的是每个单词在这个AC自动机里匹配到的次数

每次insert一个单词的时候把路径上的cnt++

那么点p->cnt就是以root到p这条路径为前缀的单词的个数

如果p->fail指向了点q，那么就会对q点产生p->cnt的贡献（root到q一定为root到p的后缀）

最后递推统计完所有fail的贡献，找到关键点输出就可以了

/* http://www.cnblogs.com/karl07/ */
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

struct trie{
    trie *next[26],*fail;
    int cnt,x;
}t[1000005];

int n;
char s[1000005];
trie *root,*NEW=t;
trie *Q[1000005],*wh[300];

trie *new1(int x){NEW++; NEW->cnt=0; NEW->x=x; return NEW;}

trie *insert(trie *p,int x){
    if (!p->next[x]) p->next[x]=new1(x);
    p->next[x]->cnt++;
    return p->next[x];
}

#define pnf p->next[i]->fail
void build_fail(){
    int l=0,r=0;
    for (int i=0;i<26;i++) if (root->next[i]) { Q[++r]=root->next[i]; root->next[i]->fail=root;}
    while (l!=r){
        trie *p=Q[++l];
        for (int i=0;i<26;i++){
            if (p->next[i]){
                Q[++r]=p->next[i];
                for (pnf=p->fail ; pnf!=root && !pnf->next[i] ; pnf=pnf->fail);
                if (pnf->next[i]) pnf=pnf->next[i];
            }
        }
    }
    for (int i=r;i>=1;i--) Q[i]->fail->cnt+=Q[i]->cnt;
}

int main(){
    scanf("%d",&n);
    root=new1(-1);
    for (int i=1;i<=n;i++){
        scanf("%s",s);
        wh[i]=root;
        int l=strlen(s);
        for (int j=0;j<l;j++) wh[i]=insert(wh[i],s[j]-'a'); 
    }
    build_fail();
    for (int i=1;i<=n;i++) printf("%d
",wh[i]->cnt);
    return 0;
}

一开始zz把strlen放到循环里慢了十几倍