【洛谷】3796 【模板】AC自动机(加强版)

来源:互联网 发布:淘宝店办理营业执照 编辑:程序博客网 时间:2024/06/06 22:07

题目传送门

这题嘛,就是一个AC自动机的模板。主要是巩固一下今天学的AC自动机。

在构造Trie树的时候在每一个字符串结尾的叶节点上标记一下该节点属于第几个字符串,然后对文本串跑一边AC自动机,对每一个经过的节点计数一下。

然后把每一个模式串按计数从大到小sort一下,第一个模式串的计数就是答案,然后向后枚举和答案相同的模式串即可。

附上AC代码:

#include <cstdio>#include <cstring>#include <queue>#include <algorithm>using namespace std;const int N=151;struct note{int ans,wz;bool operator < (const note lyf) const {return ans==lyf.ans?wz<lyf.wz:ans>lyf.ans;}}a[N];struct Aho_Corasick_automaton{int lk[26],ed,nt;}AC[100010];int n,size=0;char s[N][1010],t[1000010];inline void clear(int x){memset(AC[x].lk,0,sizeof AC[x].lk),AC[x].ed=AC[x].nt=0;}inline void insert(char *s,int wz){int len=strlen(s+1),now=0;for (int i=1; i<=len; ++i){if (!AC[now].lk[s[i]-'a']) AC[now].lk[s[i]-'a']=++size,clear(size);now=AC[now].lk[s[i]-'a'];}AC[now].ed=wz;return;}inline void build(){queue <int> que;for (int i=0; i<26; ++i)if (AC[0].lk[i]) AC[AC[0].lk[i]].nt=0,que.push(AC[0].lk[i]);while (!que.empty()){int p=que.front();que.pop();for (int i=0; i<26; ++i)if (AC[p].lk[i]) AC[AC[p].lk[i]].nt=AC[AC[p].nt].lk[i],que.push(AC[p].lk[i]);else AC[p].lk[i]=AC[AC[p].nt].lk[i];}return;}inline void work(char *s){int len=strlen(s+1),now=0;for (int i=1; i<=len; ++i){now=AC[now].lk[s[i]-'a'];for (int j=now; j; j=AC[j].nt)++a[AC[j].ed].ans;}return;}int main(void){while (scanf("%d",&n)&&n){size=0,clear(0);for (int i=1; i<=n; ++i){scanf("%s",s[i]+1),a[i]=(note){0,i};insert(s[i],i);}AC[0].nt=0,build();scanf("%s",t+1),work(t);sort(a+1,a+1+n),printf("%d\n%s\n",a[1].ans,s[a[1].wz]+1);for (int i=2; a[i].ans==a[i-1].ans; ++i) printf("%s\n",s[a[i].wz]+1);}return 0;}

原创粉丝点击