您的位置:首页 > 其它

hdu 2222(AC自动机 裸题)

2016-04-12 17:02 288 查看
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<map>
#include<queue>
#include <deque>
#include <list>
#include <ctime>
#include <stack>
#include <vector>
#include<iomanip>
#include<set>
#define Maxn 500010
#define mod
typedef long long ll;
#define FOR(i,j,n) for(int i=j;i<=n;i++)
#define DFR(i,j,k) for(int i=j;i>=k;--i)
#define lowbit(a) a&-a
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem(a) memset(a,0,sizeof(a))
#define eps 1e-9
const int inf = 0x3f3f3f3f;
const double pi = acos(-1.0);
using namespace std;
char str[1000010],keyword[51];

int head,tail;
struct node {
node * fail;
node * next[26];
int counter;
node()
{   fail=NULL;
counter=0;
FOR(i,0,25)next[i]=NULL;
}
}*q[Maxn];
node *root;
void Insert(char *str)
{   int temp,len;
node *p=root;
len=strlen(str);
FOR(i,0,len-1)
{   temp=str[i]-'a';
if(p->next[temp]==NULL)p->next[temp]=new node();
p=p->next[temp];
}
p->counter++;
}
void buildAC()
{   q[tail++]=root;
while(head!=tail)
{   node *p=q[head++];
node *temp=NULL;
FOR(i,0,25)
{   if(p->next[i]!=NULL)
{  if(p==root)p->next[i]->fail=root;
else
{  temp=p->fail;
while(temp!=NULL){
if(temp->next[i]!=NULL)
{p->next[i]->fail=temp->next[i];break;}
temp=temp->fail;}
if(temp==NULL)
p->next[i]->fail=root;
}
q[tail++]=p->next[i];
}
}

}

}
int query()
{   int index,len,result;
node *p=root;
result=0;
len=strlen(str);
FOR(i,0,len-1)
{   index=str[i]-'a';
while(p->next[index]==NULL&&p!=root)
p=p->fail;
p=p->next[index];
if(p==NULL)p=root;
node *temp=p;
while(temp!=root && temp->counter!=-1)
{   result += temp->counter;
temp->counter=-1;
temp=temp->fail;
}
}
return result;
}
int main()
{   int T,num;cin>>T;
FOR(i,1,T)
{head=tail=0;
root =new node();
scanf("%d",&num);
getchar();
FOR(j,1,num)
{gets(keyword);
Insert(keyword);
}
//  buildAC();
buildAC();
scanf("%s",str);
printf("%d\n",query());
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  AC自动机