KMP与AC自动机模板

HDU 1711 Number Sequence（KMP模板题）

http://acm.hdu.edu.cn/showproblem.php?pid=1711

#include<bits/stdc++.h>
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define pqueue priority_queue
#define NEW(a,b) memset(a,b,sizeof(a))
const double pi=4.0*atan(1.0);
const double e=exp(1.0);
const int maxn=1e6+;
typedef long long LL;
typedef unsigned long long ULL;
//typedef pair<LL,LL> P;
const LL mod=1e9+;
using namespace std;
int a[maxn],b[maxn],Next[maxn];
int n,m;
void get_next(){
    Next[]=-;
    int k=-,i=;
    while(i<m){
        while(k>-&&b[k]!=b[i]){
            k=Next[k];
        }
        if(b[k]==b[i]||k==-){
            k++;
        }
        Next[++i]=k;
    }
    return;
}
int main(){
    fio;
    int t;
    cin>>t;
    while(t--){
       cin>>n>>m;
       for(int i=;i<n;i++){
            cin>>a[i];
       }
       for(int i=;i<m;i++){
            cin>>b[i];
       }
       get_next();
       int p=;
       int is=-;
       for(int i=;i<n;){
            if(a[i]==b[p]||p==-){
                p++;
                i++;
            }
            else{
                while(p!=-){
                    if(b[p]==a[i]){
                        break;
                    }
                    p=Next[p];

                }
            }
            if(p==m){
                is=i-m+;
                break;
            }
       }
       cout<<is<<endl;
    }
}

HDU 2222 Keywords Search（AC自动机模板题）

http://acm.hdu.edu.cn/showproblem.php?pid=2222

#include<bits/stdc++.h>
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define pqueue priority_queue
#define NEW(a,b) memset(a,b,sizeof(a))
const double pi=4.0*atan(1.0);
const double e=exp(1.0);
const int maxn=1e6+;
typedef long long LL;
typedef unsigned long long ULL;
//typedef pair<LL,LL> P;
const LL mod=1e9+;
using namespace std;
struct node{
    node *nxt[];
    node *fail;
    int is;
    void init(){
        is=;
        fail=;
        NEW(nxt,);
    }
};
node *root;
node *q[maxn];
int cnt;
void Insert(string s){
    node *p=root;
    int x;
    for(int i=;s[i];i++){
        x=s[i]-'a';
        if(p->nxt[x]==NULL){
            p->nxt[x]=new node;
            p->nxt[x]->init();
        }
        p=p->nxt[x];
    }
    p->is++;
}
void build_failpoint(){
    int head=;
    int tail=;
    q[]=root;
    node *tmp;
    node *f;
    while(head<tail){
        tmp=q[head++];
        for(int i=;i<;i++){
            if(tmp->nxt[i]){
                if(tmp==root){
                    tmp->nxt[i]->fail=root;
                }
                else{
                    f=tmp->fail;
                    while(f!=){
                        if(f->nxt[i]!=){
                            tmp->nxt[i]->fail=f->nxt[i];
                            break;
                        }
                        f=f->fail;
                    }
                    if(f==){
                        tmp->nxt[i]->fail=root;
                    }
                }
                q[tail++]=tmp->nxt[i];
            }
        }
    }
}
void ac_automation(string t){
    node *p=root;
    int x;
    for(int i=;t[i];i++){
        x=t[i]-'a';
        if(p->nxt[x]){
            p=p->nxt[x];
        }
        else{
            p=p->fail;
            while(p!=){
                if(p->nxt[x]){
                    p=p->nxt[x];
                    break;
                }
                p=p->fail;
            }
            if(p==){
                p=root;
            }
        }
        node *tmp=p;
        while(tmp->is!=-&&tmp!=root){
            cnt+=tmp->is;
            tmp->is=-;
            tmp=tmp->fail;
        }
    }
}
int main(){
    fio;
    int t;
    int n;
    cin>>t;
    string s;
    while(t--){
       cin>>n;
       cnt=;
       root=new node;
       root->init();
       for(int i=;i<n;i++){
            cin>>s;
            Insert(s);
       }
       build_failpoint();
       cin>>s;
       ac_automation(s);
       cout<<cnt<<endl;
    }
}

AC自动机改成了数组版（表示作为下标党用不惯指针）

#include<bits/stdc++.h>
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define pqueue priority_queue
#define NEW(a,b) memset(a,b,sizeof(a))
const double pi=4.0*atan(1.0);
const double e=exp(1.0);
const int maxn=1e6+;
typedef long long LL;
typedef unsigned long long ULL;
//typedef pair<LL,LL> P;
const LL mod=1e9+;
using namespace std;
struct node{
    int nxt[];
    int fail;
    int is;
    void init(){
        is=;
        fail=;
        NEW(nxt,);
    }
}q[maxn];
int root=,tot=;
int cnt,que[maxn];
void Insert(string s){
    int p=root;
    int x;
    for(int i=;s[i];i++){
        x=s[i]-'a';
        if(q[p].nxt[x]==){
            q[p].nxt[x]=++tot;
            q[q[p].nxt[x]].init();
        }
        p=q[p].nxt[x];
    }
    q[p].is++;
}
void build_failpoint(){
    int head=;
    int tail=;
    que[]=root;
    int tmp;
    int f;
    while(head<tail){
        tmp=que[head++];
        for(int i=;i<;i++){
            if(q[tmp].nxt[i]){
                if(tmp==root){
                    q[q[tmp].nxt[i]].fail=root;
                }
                else{
                    f=q[tmp].fail;
                    while(f!=){
                        if(q[f].nxt[i]!=){
                            q[q[tmp].nxt[i]].fail=q[f].nxt[i];
                            break;
                        }
                        f=q[f].fail;
                    }
                    if(f==){
                        q[q[tmp].nxt[i]].fail=root;
                    }
                }
                que[tail++]=q[tmp].nxt[i];
            }
        }
    }
}
void ac_automation(string t){
    int p=root;
    int x;
    for(int i=;t[i];i++){
        x=t[i]-'a';
        if(q[p].nxt[x]){
            p=q[p].nxt[x];
        }
        else{
            p=q[p].fail;
            while(p!=){
                if(q[p].nxt[x]){
                    p=q[p].nxt[x];
                    break;
                }
                p=q[p].fail;
            }
            if(p==){
                p=root;
            }
        }
        int tmp=p;
        while(q[tmp].is!=-&&tmp!=root){
            cnt+=q[tmp].is;
            q[tmp].is=-;
            tmp=q[tmp].fail;
        }
    }
}
int main(){
    fio;
    int t;
    int n;
    cin>>t;
    string s;
    while(t--){
       cin>>n;
       cnt=;
       q[root].init();
       for(int i=;i<n;i++){
            cin>>s;
            Insert(s);
       }
       build_failpoint();
       cin>>s;
       ac_automation(s);
       cout<<cnt<<endl;
    }
}