hdu-2896-病毒侵袭（AC自动机，两种实现）

题目传送门

题意不解释   ，由于是可见字符所以只用开到94个即可。AC自动机裸题。

指针实现code：（只能用C++提交 G++是64位指针会超内存）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define MAXN 94
using namespace std;
struct node
{
    int cut;
    node *son[MAXN] ,*fail;
    node(){
        cut=0;
        fail=NULL;
        for(int i=0;i<MAXN;i++)
            son[i]=NULL;
    }
};
bool vis[510];
node *root;
void Insert(char *str,int x)
{
    node *p=root;
    int i=0;
    while(str[i]!='\0'){
        int id=str[i]-' ';
        if(p->son[id]==NULL)
            p->son[id]=new node;
        p=p->son[id];
        i++;
    }
    p->cut=x;
}
void fail_build()
{
    node *p=root;
    queue<node *>q;
    q.push(p);
    while(!q.empty()){
        p=q.front();
        q.pop();
        for(int i=0;i<MAXN;i++)
        {
            if(p->son[i]!=NULL)
            {
                if(p==root)
                    p->son[i]->fail=root;
                else
                {
                    node *temp=p->fail;
                    while(temp!=NULL){
                        if(temp->son[i]!=NULL){
                            p->son[i]->fail=temp->son[i];
                            break;
                        }
                        temp=temp->fail;
                    }
                    if(temp==NULL)
                        p->son[i]->fail=root;
                }
                q.push(p->son[i]);
            }
        }
    }
}
bool AC_find(char *str)
{
    node *p=root;
    bool falg=false;
    int i=0;
    while(str[i]!='\0'){
        //printf("%d\n",i);
        int id=str[i]-' ';
        while(p->son[id]==NULL&&p!=root)
            p=p->fail;

        p=p->son[id];
        if(p==NULL) p=root;
        node *temp=p;
        while(temp!=NULL&&temp->cut!=0){
            vis[temp->cut]=true;
            //temp->cut=-1;
            temp=temp->fail;
            falg=true;
        }
        i++;
    }
    return falg;
}
void del(node *p)
{
    for(int i=0;i<MAXN;i++)
        if(p->son[i]!=NULL)
            del(p->son[i]);
    delete(p);
}
int main()
{
    char str[10010],str1[210];
    int n,m;
    scanf("%d",&n);
    root=new node;
    for(int i=1;i<=n;i++){
        scanf("%s",str1);
        Insert(str1,i);
    }
    fail_build();
    scanf("%d",&m);
    int ans=0;
    for(int i=1;i<=m;i++){
        scanf("%s",str);
        bool falg=0;
        memset(vis,0,sizeof(vis));
        if(AC_find(str)){ans++;falg=1;}
        if(falg) printf("web %d:",i);
        for(int i=0;i<=n;i++){
            if(vis[i]) printf(" %d",i);
        }
        if(falg) printf("\n");
    }
    printf("total: %d\n",ans);
    return 0;
}

数组实现code：（数组实现，一维数组代表 节点个数）数组既节省时间也节省空间  但不太好理解 如何取舍自己看吧

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <queue>
#include <algorithm>


using namespace std;


const int allSon = 130;    ///包含所有的字符
const int maxn = 100005;
int node[maxn][allSon];    ///字典树节点
int fail[maxn];            ///fail指针
int num[maxn];             ///num[i]代表以第i个节点为结尾的单词个数
int vis[maxn];             ///标记
int index[maxn];           ///用来存放模式串的编号
char patten[maxn];         ///模式串
char text[maxn];           ///文本串
int ans[maxn];
int id;                    ///用来给节点编号
int total;                 ///统计总共有多少个网站感染病毒
///将模式串插入到字典树,传入的模式串的编号
void insertPatten(int identifier)
{
    int p = 0;      ///指向根节点
    int pos = 0;    ///下标
    int len = strlen(patten);
    while(pos < len)
    {
        int ch = patten[pos];
        if(node[p][ch]==-1)     ///节点不存在
        {
            memset(node[id],-1,sizeof(node[id]));  ///孩子节点都置空
            num[id] = 0;
            fail[id] = -1;
            vis[id] = -1;
            node[p][ch] = id++;
        }
        p = node[p][ch];
        pos++;
    }
    num[p]++;
    index[p] = identifier;      ///存储编号
}
///找fail指针，构造AC自动机
void build_AC_automaton()
{
    queue<int>qu;
    int p = 0;          ///最开始指向根节点
    qu.push(p);
    while(!qu.empty())
    {
        p = qu.front();
        qu.pop();
        for(int i = 0; i < allSon; i++)
        {
            if(node[p][i]!=-1)  ///第i个孩子存在
            {
                if(p == 0)  ///如果p是根，根的孩子fail指向根
                {
                    fail[node[p][i]] = 0;
                }
                else
                {
                    int temp = fail[p];
                    while(temp != -1)
                    {
                        if(node[temp][i] != -1)
                        {
                            fail[node[p][i]] = node[temp][i];
                            break;
                        }
                        temp = fail[temp];
                    }
                    if(temp == -1)
                    {
                        fail[node[p][i]] = 0;
                    }
                }
                qu.push(node[p][i]);
            }
        }
    }
}
///在AC自动机中进行查询，传入参数是第i个网站的编号。
void find_in_AC_automaton(int identify)
{
    int p = 0;
    int pos = 0;
    int t = 0;
    while(text[pos]!='\0')
    {
        int ch = text[pos];
        while(node[p][ch]==-1 && p!=0)
            p = fail[p];
        p = node[p][ch];
        if(p == -1) p = 0;
        int temp = p;
        /**由于别的字符串还要过来匹配，所以节点的num
        值不能进行改变，我们可以用vis数组来实现标记，避免
        重复统计**/
        while(temp!=0 && vis[temp]!=identify)
        {
            vis[temp] = identify;
            if(num[temp])   ///代表网站有一个病毒
            {
                ans[t++] = index[temp];
            }
            temp = fail[temp];
        }
        ///由于一个网站病毒最多三个，所以找够三个就出来，不要浪费时间做无用功。
        if(t >= 3)
            break;
        pos++;
    }
    if(t != 0)
    {
        total++;
        printf("web %d:",identify);
        sort(ans,ans+t);
        for(int i = 0; i < t; i++)
            printf(" %d",ans[i]);
        printf("\n");
    }
}
void init()
{
     memset(node[0],-1,sizeof(node[0]));
     num[0] = 0;
     fail[0] = -1;
     index[0] = -1;
     vis[0] = -1;
     id = 1;
     total = 0;
}
int main()
{
    int N,M;
    while(~scanf("%d",&N))
    {
        init();   ///初始化根节点
        for(int i = 1; i <= N; i++)
        {
            scanf("%s",patten);
            insertPatten(i);
        }
        build_AC_automaton();
        scanf("%d",&M);
        for(int i = 1; i <= M; i++)
        {
            scanf("%s",text);
            find_in_AC_automaton(i);
        }
        printf("total: %d\n",total);
    }
    return 0;
}