HDU2222(AC自動機模板)
阿新 • • 發佈:2018-11-10
物聯網課- - 又看了一遍AC自動機。
有趣
kuangbin巨巨的模板
#include<cstdio> #include<cstring> #include<queue> using namespace std; const int N=26; const int MAXN=500005; struct Trie{ int next[MAXN][N],fail[MAXN],end[MAXN]; int root; int tot; int newnode() { for(int i=0;i<N;i++) next[tot][i]=-1; end[tot++]=0; return tot-1; } void init() { tot=0; root=newnode(); } void insert(char buf[]) { int len=strlen(buf); int now=root; for(int i=0;i<len;i++) { int k=buf[i]-'a'; if(next[now][k]==-1) next[now][k]=newnode(); now=next[now][k]; } end[now]++; } void build() { queue<int>que; fail[root]=root; for(int i=0;i<N;i++) if(next[root][i]==-1) next[root][i]=root; else { fail[next[root][i]]=root; que.push(next[root][i]); } while(!que.empty()) { int now = que.front(); que.pop(); for(int i=0;i<N;i++) if(next[now][i]==-1) next[now][i]=next[fail[now]][i]; else { fail[next[now][i]]=next[fail[now]][i]; que.push(next[now][i]); } } } int query(char buf[]) { int len=strlen(buf); int now=root; int res=0; for(int i=0;i<len;i++) { now=next[now][buf[i]-'a']; int temp=now; while(temp!=root) { res+=end[temp]; end[temp]=0;//模式串只在主串中匹配一次就可以了 temp=fail[temp]; } } return res; } }; Trie ac; char buf[MAXN+MAXN]; int main() { int T; scanf("%d",&T); while(T--) { int n; scanf("%d",&n); ac.init(); for(int i=0;i<n;i++) { scanf("%s",buf); ac.insert(buf); } ac.build(); scanf("%s",buf); printf("%d\n",ac.query(buf)); } return 0; }