hdoj3065(AC自动机简单题)
作者:互联网
题目链接:https://vjudge.net/problem/HDU-3065
题意:给定n个不相同的模式串,模式串总长<=5e4。给定文本串,长度<=2e6。求各个模式串在文本串中出现次数。
思路:
AC自动机简单题。因为要输出次数,所以用到了拓扑排序优化版本的AC自动机。用key[u]表示以u结点结束的模式串编号,res[u]表示文本中以u为结尾的次数(拓扑排序优化版本中不用沿fail循环,那样会T),用in[u]表示入度。
另外这题是多组输入,比较坑!
AC code:
#include<cstdio> #include<cstring> #include<algorithm> #include<queue> using namespace std; const int maxt=5e4+5; const int maxn=1e3+5; int n,cnt,trie[maxt][130],key[maxt],res[maxt],in[maxt],fail[maxt]; int ans[maxn]; char s[maxn][55],ss[2000005]; void build(char *s,int k){ int len=strlen(s),u=0; for(int i=0;i<len;++i){ int t=s[i]; if(!trie[u][t]){ ++cnt; memset(trie[cnt],0,sizeof(trie[cnt])); key[cnt]=0,fail[cnt]=0,res[cnt]=0,in[cnt]=0; trie[u][t]=cnt; } u=trie[u][t]; } key[u]=k; } void get_fail(){ queue<int> que; for(int i=0;i<130;++i) if(trie[0][i]){ fail[trie[0][i]]=0; que.push(trie[0][i]); } while(!que.empty()){ int u=que.front();que.pop(); for(int i=0;i<130;++i){ if(trie[u][i]){ fail[trie[u][i]]=trie[fail[u]][i]; ++in[trie[fail[u]][i]]; que.push(trie[u][i]); } else{ trie[u][i]=trie[fail[u]][i]; } } } } void query(char *s){ int len=strlen(s),u=0; for(int i=0;i<len;++i){ int t=s[i]; u=trie[u][t]; ++res[u]; } } void topu(){ queue<int> que; for(int i=1;i<=cnt;++i) if(!in[i]) que.push(i); while(!que.empty()){ int u=que.front();que.pop(); ans[key[u]]+=res[u]; int v=fail[u]; --in[v]; res[v]+=res[u]; if(!in[v]) que.push(v); } } int main(){ while(~scanf("%d",&n)){ memset(trie[0],0,sizeof(trie[0])); key[0]=0,res[0]=0,in[0]=0; cnt=0; for(int i=1;i<=n;++i) ans[i]=0; for(int i=1;i<=n;++i){ scanf("%s",s[i]); build(s[i],i); } fail[0]=0; get_fail(); scanf("%s",ss); query(ss); topu(); for(int i=1;i<=n;++i) if(ans[i]) printf("%s: %d\n",s[i],ans[i]); } return 0; }
标签:hdoj3065,AC,include,int,maxt,maxn,自动机 来源: https://www.cnblogs.com/FrankChen831X/p/12523968.html