[洛谷3808]【模板】AC自动机（简单版）

题目大意：
　　给定$n$个模式串$p(sum|p_i|le10^6)$和一个$t(|t|le10^6)$，求在$t$中被匹配的$p$的个数。

思路：
　　AC自动机模板题，注意$t$中一个字符可能对应自动机上多个结点，因此需要按照失配指针跳转统计。统计过的结点需要特殊标记，避免重复统计，否则会超时。

#include<queue>
#include<cstdio>
#include<cctype>
inline int getint() {
    register char ch;
    while(!isdigit(ch=getchar()));
    register int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int N=1e6+1,S=26;
char s[N];
int ans=0;
class AhoCorasick {
    private:
        std::queue<int> q;
        int val[N],ch[N][S],fail[N];
        int sz,new_node() {
            return ++sz;
        }
        int idx(const char &c) const {
            return c-'a';
        }
    public:
        void insert(const char s[]) {
            int p=0;
            for(register int i=0;s[i];i++) {
                if(!ch[p][idx(s[i])]) ch[p][idx(s[i])]=new_node();
                p=ch[p][idx(s[i])];
            }
            val[p]++;
        }
        void get_fail() {
            for(register int i=0;i<S;i++) {
                if(ch[0][i]) q.push(ch[0][i]);
            }
            while(!q.empty()) {
                const int &x=q.front();
                for(register int i=0;i<S;i++) {
                    if(!ch[x][i]) {
                        ch[x][i]=ch[fail[x]][i];
                        continue;
                    }
                    fail[ch[x][i]]=ch[fail[x]][i];
                    q.push(ch[x][i]);
                }
                q.pop();
            }
        }
        void find(const char s[]) {
            int p=0;
            for(register int i=0;s[i];i++) {
                p=ch[p][idx(s[i])];
                for(register int q=p;q&&~val[q];q=fail[q]) {
                    ans+=val[q];
                    val[q]=-1;
                }
            }
        }
};
AhoCorasick ac;
int main() {
    const int n=getint();
    for(register int i=0;i<n;i++) {
        scanf("%s",s);
        ac.insert(s);
    }
    ac.get_fail();
    scanf("%s",s);
    ac.find(s);
    printf("%d
",ans);
    return 0;
}