BZOJ3172 单词

• BZOJ3172 单词
• 题解
• ｃｏｄｅ

BZOJ3172 单词

题目传送门

题解

裸的AC自动机，构造出AC自动机之后，将每一个节点的出现次数都累加到它的\(fail\)节点上去，最后得到的就是答案了。

ｃｏｄｅ

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
bool Finish_read;
template<class T>inline void read(T &x){Finish_read=0;x=0;int f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;if(ch==EOF)return;ch=getchar();}while(isdigit(ch))x=x*10+ch-'0',ch=getchar();x*=f;Finish_read=1;}
template<class T>inline void print(T x){if(x/10!=0)print(x/10);putchar(x%10+'0');}
template<class T>inline void writeln(T x){if(x<0)putchar('-');x=abs(x);print(x);putchar('\n');}
template<class T>inline void write(T x){if(x<0)putchar('-');x=abs(x);print(x);}
/*================Header Template==============*/
const int maxn=1e6+500;
int n,w[210];
char s[maxn];
/*==================Define Area================*/
struct ac_automaton{
    int tot,ch[maxn][26],fail[maxn],q[maxn],head,tail,sum[maxn];
    void insert(int &x){
        int p=0;
        int len=strlen(s);
        for (int i=0;i<len;i++){
            if (!ch[p][s[i]-'a']) ch[p][s[i]-'a']=++tot;
            ++sum[p=ch[p][s[i]-'a']];
        }
        x=p;
    }
    void getfail(){
        head=0,q[tail=1],fail[0]=-1;
        while (head!=tail){
            int x=q[++head];
            for (int i=0;i<26;i++){
                if (ch[x][i]) q[++tail]=ch[x][i],fail[ch[x][i]]=x==0?0:ch[fail[x]][i];
                else ch[x][i]=x==0?0:ch[fail[x]][i];
            }
        }
    }
    void work(){
        for (int i=tail;i;i--) sum[fail[q[i]]]+=sum[q[i]];
        for (int i=1;i<=n;i++) printf("%d\n",sum[w[i]]);
    }
}T;
 
int main(){
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%s",s),T.insert(w[i]);
    T.getfail(),T.work();
    return 0;
}