洛谷P3808 AC自动机复杂版

最新推荐文章于 2022-11-09 05:15:59 发布

原创最新推荐文章于 2022-11-09 05:15:59 发布 · 270 阅读

0 ·

CC 4.0 BY-SA版权

模板同时被 3 个专栏收录

53 篇文章

订阅专栏

字符串

12 篇文章

订阅专栏

AC自动机

3 篇文章

订阅专栏

博客讲述了在解决洛谷P3808题目时遇到的一个复杂版AC自动机问题，作者在处理特定鬼畜数据时遇到了困难，详细描述了调试过程中的困惑和最终解决方案。

这题有个数据非常鬼畜，反正我的板子莫名会wa一个地方，改改虽然对了但是还是有点懵逼。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
#include<queue>
#include<cstring>
#include<string>
#include<map>
using namespace std;
struct node{
    node*next1[26];
    node*fail;
    int tag,cnt;string strinfo;
    node(){
        memset(next1,0,sizeof(next1));fail=NULL;tag=0;cnt=0;
    }
}*root=new node();
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->tag)++;p->strinfo=str;
    //if(str=="ab")cout<<1<<endl;
}
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->tag){
                //if(temp->strinfo=="ab")cout<<temp->cnt<<endl;
                //cout<<temp->strinfo<<endl;
                temp->cnt++;
            }
            //else break;这一句一定不能要!!!!!!!!!!!!!!!
            temp=temp->fail;
        }
    }
}
void dl(node*p)//清除所有节点
{
    for(int i=0;i<26;i++){
        if(p->next1[i])dl(p->next1[i]);
    }
    if(p!=root)delete p;
}
int maxn=0;vector<string>ans;
void find(node *p)
{
    for(int i=0;i<26;i++){
        if(p->next1[i]){
            find(p->next1[i]);
        }
    }
    if(p->cnt){
        //if(p->strinfo=="ab")cout<<p->cnt<<endl;
        if(p->cnt>maxn){//如果大于现有答案,清空以前的答案
            maxn=p->cnt;ans.clear();
            for(int i=1;i<=p->tag;i++)
                ans.push_back(p->strinfo);
        }
        else if(p->cnt==maxn)
            for(int i=1;i<=p->tag;i++)
                ans.push_back(p->strinfo);
    }
}
int main()
{
    int n,i,j;string str;
    while(cin>>n&&n){
        map<string,int>m;int shunxu=0;//map保存加入顺序
        maxn=0;ans.clear();dl(root);
        for(i=0;i<26;i++)root->next1[i]=NULL;root->fail=NULL;root->cnt=0;root->tag=0;
        for(i=1;i<=n;i++){
            cin>>str;insert(str);m.insert(pair<string,int>(str,++shunxu));
        }
        build_fail_pointer();
        cin>>str;
        //cout<<str<<endl;
        ac_automation(str);find(root);
        cout<<maxn<<endl;
        map<int,string>m1;
        for(auto i=ans.begin();i!=ans.end();i++){
            string strt=*i;
            auto j=m.find(strt);m1.insert(pair<int,string>(j->second,strt));//按照加入顺序再次丢进一个从小到大的map中,方便按顺序输出
            m.erase(j);
        }
        for(auto i=m1.begin();i!=m1.end();i++)
            cout<<i->second<<endl;
    }

    return 0;
}

鬼畜数据：

10
qabqks
vimbirqy
cflwvxtp
klljfj
ab
nkeiid
fkypjfev
yvgp
evdhs
xaizql
qabqksatffqpjomzstjabfklljfjqevdhsqabqkscflwvxtpeevdhsmzonkeiid

标准输出：

3
ab

如果加上71行那句不能要的就会疯狂wa，一脸懵逼….