[FJOI2011] 前缀与后缀

---

题目描述

现有n个由小写字母组成的单词，然后给定m对小写的前缀与后缀。对于每对前缀与后缀，请你求出在n个单词中有多少个同时包含他们。

---

输入格式

输入第一行为n，接下去n行每行包含一个单词，合起来总字符数不超过100000个。
接下去一行为m，接下去m行每行包含一对空格隔开的前缀与后缀，合起来总字符数不超过200000个。

---

输出格式

对输出m行，每行一个整数为对应所求。

---

样例数据

样例输入

3
aaaaa
abacabaa
avtobus
6
a a
a aa
aa a
aaaaa aaaa
abac caba
abac a

样例输出

2
2
1
1
0
1

---

题目分析

将每个单词转化为：单词_单词
比如avtobus转为avtobus_avtobus
在前缀和后缀间添加下划线并反向插入AC自动机
比如abac a转为a_abac
接着依次匹配主串统计个数
注意可能有重复的字符串，需将统计写在树的内部

---

源代码

#include<algorithm>#include<iostream>#include<iomanip>#include<cstring>#include<cstdlib>#include<vector>#include<cstdio>#include<cmath>#include<queue>using namespace std;inline const int Get_Int() {    int num=0,bj=1;    char x=getchar();    while(x<'0'||x>'9') {        if(x=='-')bj=-1;        x=getchar();    }    while(x>='0'&&x<='9') {        num=num*10+x-'0';        x=getchar();    }    return num*bj;}const int maxn=400005;struct Tree {    int child[28],fail,tot,index,sum; //fail失败指针    void clear() {        memset(child,0,sizeof(child));        fail=0;        tot=0;        index=0;        sum=0;    }};int root=1,Map[400005];struct Aho_Corasick_Automaton { //AC自动机    int cnt;    Tree tree[maxn];    void init() {        cnt=1;        memset(tree,0,sizeof(tree));    }    void insert(string s,int id) {        int now=root,len=s.length();        for(int i=0; i<len; i++) {            int j;            if(s[i]=='_')j=27;            else j=s[i]-'a';            if(!tree[now].child[j]) {                tree[++cnt].clear();                tree[now].child[j]=cnt;            }            now=tree[now].child[j];        }        tree[now].tot=1;        Map[id]=now;    }    void buildfail() { //Bfs构造Fail指针        queue<int>Q;        Q.push(root);        while(!Q.empty()) {            int Now=Q.front();            Q.pop();            for(int i=0; i<28; i++) {                int Next=tree[Now].child[i];                if(Next==0)continue; //儿子不存在                Q.push(Next);                int fatherfail=tree[Now].fail; //父亲的失败指针                while(fatherfail&&!tree[fatherfail].child[i])fatherfail=tree[fatherfail].fail; //寻找可退回点                if(fatherfail)tree[Next].fail=tree[fatherfail].child[i]; //如果存在满足条件的点则设置失败指针                else tree[Next].fail=root; //否则指回root            }        }    }    int match(string s) { //匹配字符串s        int t=root,len=s.length(),ans=0;        for(int i=0; i<len; i++) {            int j;            if(s[i]=='_')j=27;            else j=s[i]-'a';            while(t&&!tree[t].child[j])t=tree[t].fail; //不匹配            if(t)t=tree[t].child[j]; //匹配            else t=root; //返回root            for(int j=t; j; j=tree[j].fail)                tree[j].sum+=tree[j].tot;        }        return ans;    }};Aho_Corasick_Automaton ac;int n,m;string s[400005];int main() {    ios::sync_with_stdio(false);    cin>>n;    for(int i=1; i<=n; i++) {        cin>>s[i];        s[i]=s[i]+'_'+s[i];    }    cin>>m;    ac.init();    for(int i=1; i<=m; i++) {        string a,b;        cin>>a>>b;        ac.insert(b+'_'+a,i);    }    ac.buildfail();    for(int i=1; i<=n; i++)ac.match(s[i]);    for(int i=1; i<=m; i++)printf("%d\n",ac.tree[Map[i]].sum);    return 0;}

---