hdu 3065 病毒侵袭持续中 AC自动机

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3065

小t非常感谢大家帮忙解决了他的上一个问题。然而病毒侵袭持续中。在小t的不懈努力下，他发现了网路中的“万恶之源”。这是一个庞大的病毒网站，他有着好多好多的病毒，但是这个网站包含的病毒很奇怪，这些病毒的特征码很短，而且只包含“英文大写字符”。当然小t好想好想为民除害，但是小t从来不打没有准备的战争。知己知彼，百战不殆，小t首先要做的是知道这个病毒网站特征：包含多少不同的病毒，每种病毒出现了多少次。大家能再帮帮他吗？
Input
第一行，一个整数N（1<=N<=1000），表示病毒特征码的个数。
接下来N行，每行表示一个病毒特征码，特征码字符串长度在1—50之间，并且只包含“英文大写字符”。任意两个病毒特征码，不会完全相同。
在这之后一行，表示“万恶之源”网站源码，源码字符串长度在2000000之内。字符串中字符都是ASCII码可见字符（不包括回车）。
Output
按以下格式每行一个，输出每个病毒出现次数。未出现的病毒不需要输出。
病毒特征码: 出现次数
冒号后有一个空格，按病毒特征码的输入顺序进行输出。
 
题意描述：中文题，如上所述。
算法分析：AC自动机的简单题，大白书和很多博客都有AC自动机的详细介绍。AC自动机真心是个字符串处理的神奇东西，好好学习一下。
  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<cmath>
  5 #include<cstdlib>
  6 #include<algorithm>
  7 #define inf 0x7fffffff
  8 #define PI 3.141592654
  9 #define exp 1e-10
 10 using namespace std;
 11 const int maxn=1000+10;
 12 const int M = 2000000+10;
 13 
 14 struct node
 15 {
 16     node *fail;
 17     node *child[26];
 18     int count;
 19     int id;
 20     node () {
 21         fail=NULL;
 22         count=0;
 23         id=-1;
 24         memset(child,NULL,sizeof(child));
 25     }
 26 }*q[50000+100];
 27 int head,tail;
 28 int n;
 29 char keyword[maxn][55];
 30 int an[maxn];
 31 char str[M];
 32 
 33 void insert(node *root,char *str,int k)
 34 {
 35     node *p=root;
 36     int len=strlen(str);
 37     for (int i=0 ;i<len ;i++)
 38     {
 39         int index=str[i]-'A';
 40         if (p->child[index]==NULL) p->child[index]=new node();
 41         p=p->child[index];
 42     }
 43     p->count++;
 44     p->id=k;// no repeat
 45     return ;
 46 }
 47 void build_AC(node *root)
 48 {
 49     root->fail=NULL;
 50     q[head++]=root;
 51     while (head != tail)
 52     {
 53         node *temp=q[tail++];
 54         node *p=NULL;
 55         for (int i=0 ;i<26 ;i++)
 56         {
 57             if (temp->child[i]!=NULL)
 58             {
 59                 if (temp==root) temp->child[i]->fail=root;
 60                 else
 61                 {
 62                     p=temp->fail;
 63                     while (p!=NULL)
 64                     {
 65                         if (p->child[i]!=NULL)
 66                         {
 67                             temp->child[i]->fail=p->child[i];
 68                             break;
 69                         }
 70                         p=p->fail;
 71                     }
 72                     if (p==NULL) temp->child[i]->fail=root;
 73                 }
 74                 q[head++]=temp->child[i];
 75             }
 76         }
 77     }
 78     return ;
 79 }
 80 void find(node *root,char *str)
 81 {
 82     int len=strlen(str);
 83     node *p=root;
 84     for (int i=0 ;i<len ;i++)
 85     {
 86         if (str[i]>='A' && str[i]<='Z')
 87         {
 88             int index=str[i]-'A';
 89             while (p->child[index]==NULL && p!=root) p=p->fail;
 90             p=p->child[index];
 91             p= p==NULL ? root : p ;
 92             node *temp=p;
 93             while (temp!=root)
 94             {
 95                 if (temp->count) an[temp->id ]++;
 96                 temp=temp->fail;
 97             }
 98         }
 99         else p=root;
100     }
101     return ;
102 }
103 int main()
104 {
105     while (scanf("%d",&n)!=EOF)
106     {
107         head=tail=0;
108         memset(str,0,sizeof(str));
109         memset(keyword,0,sizeof(keyword));
110         memset(an,0,sizeof(an));
111         node *root=new node();
112 
113         for (int i=0 ;i<n ;i++)
114         {
115             scanf("%s",keyword[i]);
116             insert(root,keyword[i],i);
117         }
118         build_AC(root);
119         getchar() ;gets(str) ;
120         find(root,str);
121         for (int i=0 ;i<n ;i++)
122         if (an[i]) printf("%s: %d\n",keyword[i],an[i]);
123     }
124     return 0;
125 }

 

posted @ 2015-03-30 21:11  huangxf  阅读(187)  评论(0)  编辑  收藏  举报