您的位置:首页 > 其它

(模板)AC自动机

2018-03-29 17:43 211 查看
说明链接:https://blog.csdn.net/creatorx/article/details/71100840

代码:(其中有个地方需要非常注意)

// luogu-judger-enable-o2
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<algorithm>
#include<cstring>
#include<string>
using namespace std;
struct node{
node*next1[26];
node*fail;
int sum;
node(){
memset(next1,0,sizeof(next1));fail=NULL;sum=0;
}
}*root=new node();
int cnt=0;
void insert(string str)
{
node*p=root;
for(int i=0;i<str.size();i++){
int x=str[i]-'a';
if(!p->next1[x]){
node*t=new node();p->next1[x]=t;
}
p=p->next1[x];
}
p->sum++;
}
void build_fail_pointer(void)
{
node*p,*temp;
queue<node*>que;que.push(root);
while(!que.empty()){
temp=que.front();que.pop();
for(int i=0;i<26;i++){
if(temp->next1[i]){
if(temp==root){
temp->next1[i]->fail=root;
}
else{
p=temp->fail;
while(p){
if(p->next1[i]){
temp->next1[i]->fail=p->next1[i];break;
}
p=p->fail;
}
if(!p)temp->next1[i]->fail=root;
}
que.push(temp->next1[i]);
}
}
}
}
void ac_automation(string str)
{
node*p=root;
for(int i=0;i<str.size();i++){
int x=str[i]-'a';
while(!p->next1[x]&&p!=root)p=p->fail;
p=p->next1[x];
if(!p)p=root;
node*temp=p;
while(temp!=root){
if(temp->sum){
cnt+=temp->sum;temp->sum=0;
}
else break;//**如果不只是需要知道模式串是否出现过而**
//**是要找出所有出现的位置的话,这一行是不能要的!!!!!!**
//并且,根据观察,没有这一行似乎并不会导致答案错误,只是时间会增加;
//但是,有这一行却可能导致答案错误
temp=temp->fail;
}
}
}
int main()
{
int n,i,j;string str;
cin>>n;
for(i=1;i<=n;i++){
cin>>str;insert(str);
}
cin>>str;
build_fail_pointer();ac_automation(str);
cout<<cnt<<endl;

return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: