POJ 2896 AC自动机 or 暴力

DESCRIPTION ：大意是说。给你n个代表病毒的字符串。m个表示网站的字符串。让你计算有多少个网站被病毒感染了。被那些病毒感染了。

刚开始就想暴力。然而，忽略了条件：每个网站最多有三个病毒。于是。TLE了。于是换AC 自动机。于是MLE了。于是把最大的结构体指针数组换成队列。用时间来换空间。23333

应该注意结构体的初始化是必须的。

附代码：
AC 自动机：

#include<stdio.h> #include<string.h> #include<iostream> #include<queue> #include<algorithm> #include<map> using namespace std; #define T_SIZE 10000 struct trie{ trie* nexts[128]; trie* fail; int num1, num2; int mem; trie(){ for(int i=0;i<128;i++){ nexts[i]=NULL; } fail=NULL; num1= num2 = 0; } }; char T[T_SIZE+2]; queue<struct trie*>que; int ans[50]; map<int, int>mm; void insert(trie* root,char* s, int m){ trie* p=root; for(int i=0;s[i]!='\0';i++){ if(p->nexts[s[i]-0]==NULL) p->nexts[s[i]-0]=new trie; p=p->nexts[s[i]-0]; } p->num1++; p->mem = m; } void build_ac_automation(trie* root){ que.push(root); while(!que.empty()){ trie* fronts=que.front(); que.pop(); for(int i=0;i<128;i++){ if(fronts->nexts[i]!=NULL){ trie* p=fronts->fail; while(p!=NULL){ if(p->nexts[i]!=NULL){ fronts->nexts[i]->fail=p->nexts[i]; break; } p=p->fail; } if(p==NULL){ fronts->nexts[i]->fail=root; } que.push(fronts->nexts[i]); } } } } int ac_find(trie* root,char* T){ trie* p=root; int sum=0; memset(ans,0,sizeof(ans)); for(int i=0,len=strlen(T);i<len;i++){ while(p->nexts[T[i]-0]==NULL && p!=root) p=p->fail; if(p->nexts[T[i]-0]!=NULL){ p=p->nexts[T[i]-0]; } trie* temp=p; temp->num2 = temp->num1; while(temp!=root && temp->num2!=-1){ temp->num2 = temp->num1; if (!mm[temp->mem]) {ans[sum]=temp->mem+1; sum+=temp->num2;} temp->num2=-1; temp=temp->fail; } } return sum; } int main(){ int n; while(scanf("%d",&n)!= EOF){ mm.clear(); int summ=0; trie* root=new trie; getchar(); for(int i=0;i<n;i++){ memset(T,0,sizeof(T)); gets(T); insert(root,T, i); } build_ac_automation(root); int t; scanf("%d",&t); getchar(); for(int i=0;i<t;i++){ memset(T,0,sizeof(T)); gets(T); int num=ac_find(root,T); if(num!=0){ summ++; printf("web %d:",i+1); sort(ans, ans+num); printf(" %d", ans[0]); for(int j=1;j<num;j++){ if (ans[j] == ans[j-1]) continue; printf(" %d",ans[j]); } printf("\n"); } } printf("total: %d\n",summ); } return 0; }