hdu 2896 AC自动机模板题

题意：给出n个模式串，再给出m个文本串

　　　针对每一个文本串，假如文本串中包含模式串，就total++，然后输出包含的模式串，从小到达输出编号

思路：这道题有个地方需要注意一下，题目中给出的字符不单单只有小写字母，所以我们把trie的第二维开到128

　　　　其他地方没什么不同

　　　　但是！ 我自我感觉这样子做是会超时的，可是竟然没有超。。。

　　　　　　根据数据：每一个文本串1e4，那么查询当个的复杂度为 1e4+模式串个数（这里忽略不计）*(模式串的平均长度，我们假定100)

　　　　　　　　　　　　因此，单个查询复杂度就已经到达了1e6，然后有1000个查询。。。

　　　　　　大概是学艺不精把，毕竟初学ac自动机，所以其中缘由并不清楚

#include<bits/stdc++.h>
using namespace std;
const int maxn=1000001;
struct AC{
    int son[130];
    int flag,fail;
}trie[maxn];
int n,cnt;
char s[maxn];
queue<int>q;
int total;
int match[maxn];
int res[maxn];
int book;
void Insert(char* s,int base)
{
    int u=1,len=strlen(s);
    for(int i=0;i<len;i++){
        int v=s[i];
        if(!trie[u].son[v])
            trie[u].son[v]=++cnt;
        u=trie[u].son[v];
    }
    trie[u].flag++;
    match[u]=base;
}
void getFail()
{
    for(int i=0;i<=128;i++)
        trie[0].son[i]=1;          //初始化0的所有儿子都是1
    q.push(1);trie[1].fail=0;               //将根压入队列
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=0;i<=128;i++){              //遍历所有儿子
            int v=trie[u].son[i];           //处理u的i儿子的fail，这样就可以不用记父亲了
            int Fail=trie[u].fail;          //就是fafail，trie[Fail].son[i]就是和v值相同的点
            if(!v){
                trie[u].son[i]=trie[Fail].son[i];
                continue;
            }  //不存在该节点，第二种情况
            trie[v].fail=trie[Fail].son[i]; //第三种情况，直接指就可以了
            q.push(v);                      //存在实节点才压入队列
        }
    }
}
int query(char* s)
{
    int u=1,ans=0,len=strlen(s);
    for(int i=0;i<len;i++){
        int v=s[i];
        int k=trie[u].son[v];       //跳Fail
        while(k>1){   //经过就不统计了
            if(trie[k].flag>0){
                book++;
                res[book]=match[k];
            }
            k=trie[k].fail;         //继续跳Fail
        }
        u=trie[u].son[v];           //到下一个儿子
    }
    return ans;
}
int main(){
    cnt=1;scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",s);
        Insert(s,i);
    }
    getFail();
    int T;
    scanf("%d",&T);
    for(int Case=1;Case<=T;Case++){
        book=0;
        scanf("%s",s);
        query(s);
        if(book) total++;
        sort(res+1,res+1+book);
        if(book){
            printf("web %d: ",Case);
            for(int i=1;i<book;i++)
                printf("%d ",res[i]);
            printf("%d\n",res[book]);
        }
    }
    printf("total: %d\n",total);
    return 0;
}