AC自助机

http://blog.henix.info/blog/trie-aho-corasick.html

这个博客说的很清楚了。自己实现了个，留个模版

View Code

  1 #include<iostream>

  2 #include<string>

  3 #include<algorithm>

  4 using namespace std;

  5 

  6 struct node

  7 {

  8     struct node *fail;

  9     struct node *next[26];

 10     int cnt;

 11     node()

 12     {

 13         cnt=0;

 14         fail=NULL;

 15         for(int i=0;i<26;i++)

 16             next[i]=NULL;

 17     }

 18 };

 19 node *root;

 20 char str[1000001];

 21 int n;

 22 

 23 void insert(char word[])

 24 {

 25     int i=0,len=strlen(word);

 26     node *tmp=root;

 27     if(tmp==NULL)

 28     {

 29         tmp=new node();

 30         root=tmp;

 31     }

 32     while(word[i])

 33     {

 34         int b=word[i]-'a';

 35         if(tmp->next[b]==NULL)

 36         {

 37             tmp->next[b]=new node();

 38         }

 39         if(i==len-1)

 40             tmp->next[b]->cnt++;

 41         tmp=tmp->next[b];

 42         i++;

 43     }

 44 }

 45 

 46 node *q[500010];

 47 int head,tail;

 48 

 49 void addFail()

 50 {

 51     head=tail=0;

 52     q[tail++]=root;

 53     while(head<tail)

 54     {

 55         node *x=q[head++];

 56         int i;

 57         for(i=0;i<26;i++)

 58         {

 59             if(x->next[i]!=NULL)

 60             {

 61                 q[tail++]=x->next[i];

 62                 node *t=x->fail;

 63                 while((t!=NULL) && t->next[i]==NULL)

 64                     t=t->fail;

 65                 if(t==NULL)

 66                     x->next[i]->fail=root;

 67                 else

 68                     x->next[i]->fail=t->next[i];

 69             }

 70         }

 71     }

 72 }

 73 

 74 int match(char word[])

 75 {

 76     int i=0,ans=0;

 77     node *tmp=root;

 78     while(word[i])

 79     {

 80         int b=word[i]-'a';

 81         while(tmp!=NULL && tmp->next[b]==NULL)

 82             tmp=tmp->fail;

 83         if(tmp!=NULL)

 84         {

 85             tmp=tmp->next[b];

 86             node *t=tmp;

 87             while(t!=NULL)

 88             {

 89                 if(t->cnt>0)

 90                 {

 91                     ans+=t->cnt;

 92                     t->cnt=0;

 93                 }

 94                 t=t->fail;

 95             }

 96         }

 97         else

 98             tmp=root;

 99         i++;

100     }

101     return ans;

102 }

103 

104 int main()

105 {

106     int i,cas;

107     freopen("D:\\in.txt","r",stdin);

108     scanf("%d",&cas);

109     while(cas--)

110     {

111         scanf("%d",&n);

112         root=NULL;

113         for(i=0;i<n;i++)

114         {

115             scanf("%*c%s",str);

116             insert(str);

117         }

118         scanf("%*c%s",str);

119         addFail();

120         int ans=match(str);

121         printf("%d\n",ans);

122     }

123     return 0;

124 }

 

View Code

  1 #include<iostream>

  2 #include<string>

  3 #include<algorithm>

  4 using namespace std;

  5 

  6 struct node

  7 {

  8     int cnt;

  9     struct node *fail;

 10     struct node *next[26];

 11 };

 12 

 13 node root[500010];

 14 int n,num;

 15 char str[1000001];

 16 

 17 void insert(char word[])

 18 {

 19     int i=0;

 20     node * tmp=root;

 21     while(word[i])

 22     {

 23         int b=word[i]-'a';

 24         if(tmp->next[b]==NULL)

 25         {

 26             tmp->next[b]=root+num;

 27             memset(root+num,0,sizeof(struct node));

 28             num++;

 29         }

 30         tmp=tmp->next[b];

 31         i++;

 32     }

 33     tmp->cnt++;

 34 }

 35 

 36 node *q[500010];

 37 int head,tail;

 38 

 39 void add_Fail()

 40 {

 41     head=tail=0;

 42     q[tail++]=root;

 43     while(head<tail)

 44     {

 45         node *x=q[head++];

 46         for(int i=0;i<26;i++)

 47         {

 48             if(x->next[i]!=NULL)

 49             {

 50                 q[tail++]=x->next[i];

 51                 node *t=x->fail;

 52                 while((t!=NULL) && t->next[i]==NULL)

 53                     t=t->fail;

 54                 if(t==NULL)

 55                     x->next[i]->fail=root;

 56                 else

 57                     x->next[i]->fail=t->next[i];

 58             }

 59         }

 60     }

 61 }

 62 

 63 int ans;

 64 

 65 void match(char word[])

 66 {

 67     int i=0;

 68     node *tmp=root;

 69     while(word[i])

 70     {

 71         int b=word[i]-'a';

 72         while(tmp!=NULL && tmp->next[b]==NULL)

 73             tmp=tmp->fail;

 74         if(tmp!=NULL)

 75         {

 76             tmp=tmp->next[b];

 77             node *t=tmp;

 78             while(t!=NULL)

 79             {

 80                 if(t->cnt>0)

 81                 {

 82                     ans+=t->cnt;

 83                     t->cnt=0;

 84                 }

 85                 t=t->fail;

 86             }

 87         }

 88         else

 89             tmp=root;

 90         i++;

 91     }

 92 }

 93 

 94 int main()

 95 {

 96     int cas,i;

 97     //freopen("D:\\in.txt","r",stdin);

 98     scanf("%d",&cas);

 99     while(cas--)

100     {

101         memset(root,0,sizeof(struct node));

102         num=1;

103         scanf("%d",&n);

104         for(i=0;i<n;i++)

105         {

106             scanf("%*c%s",str);

107             insert(str);

108         }

109         add_Fail();

110         ans=0;

111         scanf("%*c%s",str);

112         match(str);

113         printf("%d\n",ans);

114     }

115     return 0;

116 }