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 ;}