hdu 2222 Keywords Search

题意：上面的字符串在下面的文章中有多少个单词出现过。
思路：AC自动机
  1 #include <cstdio>
  2 #include <cstring>
  3 #include <queue>
  4 #include <algorithm>
  5 using namespace std;
  6 
  7 char str[1000010];
  8 int t,n;
  9 int ans;
 10 
 11 struct node
 12 {
 13     int v;
 14     struct node *next[27],*fail;
 15 }*root;
 16 
 17 void insert(char str[])
 18 {
 19     struct node *head=root;
 20     int k=strlen(str);
 21     for(int i=0; i<k; i++)
 22     {
 23         int x=str[i]-'a';
 24         if(head->next[x]==NULL)
 25         {
 26             struct node *p=(struct node*)malloc(sizeof(struct node));
 27             p->v=0;
 28             p->fail=root;
 29             for(int j=0; j<26; j++)
 30             {
 31                 p->next[j]=NULL;
 32             }
 33             head->next[x]=p;
 34         }
 35         head=head->next[x];
 36         if(i==k-1) head->v+=1;
 37     }
 38 }
 39 
 40 node *getFail(node *p,int x)
 41 {
 42     if(p->next[x]!=NULL)
 43     {
 44         return p->next[x];
 45     }
 46     else
 47     {
 48         if(p==root) return root;
 49         else
 50             return getFail(p->fail,x);
 51     }
 52 }
 53 
 54 void Build_AC()
 55 {
 56     node *head,*p;
 57     queue<node *>q;
 58     q.push(root);
 59     while(!q.empty())
 60     {
 61         head=q.front();
 62         q.pop();
 63         for(int j=0; j<26; j++)
 64         {
 65             if(head->next[j]!=NULL)
 66             {
 67                 q.push(head->next[j]);
 68                 if(head==root)
 69                 {
 70                     head->next[j]->fail=root;
 71                 }
 72                 else
 73                 {
 74                     bool flag=false;
 75                     p=head->fail;
 76                     while(p->next[j]==NULL)
 77                     {
 78                        if(p==root)
 79                        {
 80                           flag=true;
 81                           break;
 82                        }
 83                        p=p->fail;
 84                     }
 85                     if(flag) head->next[j]->fail=root;
 86                     else
 87                     head->next[j]->fail=p->next[j];
 88                 }
 89             }
 90 
 91         }
 92     }
 93 }
 94 
 95 int Find(char str[])
 96 {
 97     node *head=root;
 98     int k=strlen(str);
 99     int cnt=0;
100     for(int i=0; i<k; i++)
101     {
102         int x=str[i]-'a';
103         while(head->next[x]==NULL&&head!=root) head=head->fail;
104         if(head->next[x]==NULL) head=root;
105         else head=head->next[x];
106         node *p=head;
107         while(p!=root)
108         {
109             cnt+=p->v;
110             p->v=0;
111             p=p->fail;
112         }
113     }
114     return cnt;
115 }
116 
117 
118 int main()
119 {
120     scanf("%d",&t);
121     while(t--)
122     {
123         ans=0;
124         scanf("%d",&n);
125         root=(struct node *)malloc(sizeof(struct node));
126         root->v=0;
127         root->fail=root;
128         for(int i=0; i<26; i++)
129         {
130             root->next[i]=NULL;
131         }
132         for(int i=1; i<=n; i++)
133         {
134             scanf("%s",str);
135             insert(str);
136         }
137         Build_AC();
138         scanf("%s",str);
139         ans=Find(str);
140         printf("%d
",ans);
141     }
142     return 0;
143 }
View Code