HDU 2896 AC自动机

中文题，题意就不解释了。

这道题把我坑了下，应该是做题不仔细的原因，一开始我以为是26个字母的（没认真读题，看样例的结果） ，然后RE了好几发，最多发现题目里的描述是ASCLL码表里面的可见字符，然后将建字典树过程中的0 - 25的循环改成0 - 127 就过了。

讲一下思路，这道题就是AC自动机的模版题，唯一需要注意的就是加一个id的域来存这个字符串的序号，最后输出的时候要按从小到大的顺序，我直接全部扔进set输出了。

不过我感觉数据有点水啊。。。总感觉A的不是很踏实。

加了释放内存，但是发现内存没比原来的少，难道数据只有一组o(︶︿︶)o 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <iomanip>
#define PI acos(-1.0)
#define Max 2505
#define inf 1<<28
#define LL(x) ( x << 1 )
#define RR(x) ( x << 1 | 1 )
#define REP(i,s,t) for( int i = ( s ) ; i <= ( t ) ; ++ i )
#define ll long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
#define PII pair<int,int>
using namespace std;


struct node {
    node *fail ;
    node *next[128] ;
    int count ;
    int id ;
    node() {
        fail = 0 ;
        count = 0 ;
        mem(next , 0) ;
        id = 0 ;
    }
}*qe[1000005] ;
node *root = 0 ;
//insert a[] .
void insert(char *a,int id) {
    node *p = root ;
    int l = strlen(a) ;
    for (int i = 0 ; i < l ; i ++ ) {
        int tt = a[i] ;
        if(p -> next[tt] == 0) {
            p -> next[tt] = new node() ;
        }
        p = p -> next[tt] ;
    }
    p -> count ++ ;
    p -> id = id ;
}
//build *fail .
void build() {
    root -> fail = 0 ;
    int h = 0 , t = 0 ;
    qe[h ++ ] = root ;
    while(h > t) {
        node *temp = qe[t ++ ] ;
        node *p = 0 ;
        for (int i = 0 ; i < 128 ; i ++ ) {
            if(temp -> next[i] != 0) {
                if(temp == root)temp -> next[i] -> fail = root ;
                else {
                    p = temp -> fail ;
                    while(p != 0) {
                        if(p -> next[i] != 0) {
                            temp -> next[i] -> fail = p -> next[i] ;//找到匹配
                            break ;
                        }
                        p = p -> fail ;
                    }
                    if(p == 0)temp -> next[i] -> fail = root ;//如果没找到，则将fail指向root
                }
                qe[h ++ ] = temp -> next[i] ;
            }
        }
    }
}
set<int>anss[1111] ;
int search(char *a ,int id) {
    int l = strlen(a) ;
    node *p = root ;
    int ans = 0 ;
    for (int i = 0  ; i < l ; i ++ ) {
        int tt = a[i] ;
        while(p -> next[tt] == 0 && p != root)p = p -> fail ;
        p = p -> next[tt] ;
        p = (p == 0) ? root : p ;
        node *temp = p ;
        while(temp != root && temp -> count != 0) {
            ans += temp -> count ;
            anss[id].insert(temp -> id) ;
            //temp -> count = -1 ;
            temp = temp -> fail ;
        }
    }
    return ans ;
}

void deleteAll(node *p){
    for (int i = 0 ; i < 128 ; i ++ )
    if(p -> next[i] != 0){
        deleteAll(p -> next[i]) ;
    }
    delete p ;
}
char a[11111] ;
int num[1111] ;
int main() {
    int n ;
    cin >> n ;
    getchar() ;
    root = new node() ;
    for (int i = 1 ; i <= n ;i ++ ){
        gets(a) ;
        insert(a ,i ) ;
    }
    int m ;
    cin >> m ;
    getchar() ;
    build() ;
    for (int i = 1 ; i <= m ;i ++ ){
        gets(a) ;
        num[i] = search(a , i) ;
    }
    int cc = 0 ;
    for (int i = 1 ; i <= m ;i ++ ){
        if(num[i]){
            printf("web %d:",i) ;
            for(set<int>::iterator p = anss[i].begin() ; p != anss[i].end() ; ++ p)
            cout <<" "<< *p ;
            cc ++ ;
            puts("") ;
        }
    }
    printf("total: %d\n",cc) ;
    deleteAll(root) ;
    return 0 ;
}

贴一个AC自动机静态的模版

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include <set>
using namespace std;
char ss[1000010];
int n,cnt,ans;
struct node {
    int num;
    int next[128];
    int fail;
    int id ;
    void init() {
        memset(next,-1,sizeof(next));
        fail=-1;
        num=0;
        id = 0 ;
    }
} a[500000];
void insert(char s[],int id) {
    int p=0,i=0;
    int len=strlen(s);
    while(i<len) {
        int k=s[i] ;
        if(a[p].next[k]==-1) {
            a[cnt].init();
            a[p].next[k]=cnt++;
        }
        i++;
        p=a[p].next[k];
    }
    a[p].num++;
    a[p].id = id ;
}
queue<int>q;
void ac_bfs() {
    q.push(0);
    while(!q.empty()) {
        int p=q.front();
        q.pop();
        int fail=0;
        for(int i=0; i<128; i++)if(a[p].next[i]!=-1) {
                if(p==0) {
                    a[a[p].next[i]].fail=p;
                } else {
                    fail=a[p].fail;
                    while(fail!=-1) {
                        if(a[fail].next[i]!=-1) {
                            a[a[p].next[i]].fail=a[fail].next[i];
                            break;
                        }
                        fail=a[fail].fail;
                    }
                    if(fail==-1)a[a[p].next[i]].fail=0;
                }
                q.push(a[p].next[i]);
            }
    }
}
set<int>fk[1111] ;
int query(char s[] ,int id) {
    int i=0,len=strlen(s);
    int p=0,ans=0;
    while(i<len) {
        int k=s[i] ;
        while(a[p].next[k]==-1&&p!=0)p=a[p].fail;
        p=a[p].next[k];
        p=(p==-1)?0:p;
        int tmp=p;
        while(tmp!=0 && a[tmp].num != 0) {
            ans+=a[tmp].num;
            fk[id].insert(a[tmp].id) ;
            tmp=a[tmp].fail;
        }
        i++;
    }
    return ans;
}


char virus[11111] ;
int fkk[1111] ;
int main() {
    int n ;
    cin >> n ;
    cnt = 1 ;
    a[0].init() ;
    getchar() ;
    for (int i = 1 ; i <= n ; i ++ ) {
        gets(virus) ;
        insert(virus,i) ;
    }
    int m ;
    cin >> m ;
    ac_bfs() ;
    getchar() ;
    int cc = 0 ;
    for (int i = 1 ; i <= m ; i ++ ) {
        gets(virus) ;
        fkk[i] = query(virus , i) ;
    }
    for (int i = 1 ; i <= m ; i ++ ) {
        if(fkk[i]) {
            printf("web %d:",i) ;
            for(set<int>::iterator p = fk[i].begin() ; p != fk[i].end() ; ++ p)
                cout <<" "<< *p ;
            cc ++ ;
            puts("") ;
        }
    }
    printf("total: %d\n",cc) ;
    return 0 ;
}