F08【模板】AC自动机
F08【模板】AC自动机——信息学竞赛算法_哔哩哔哩_bilibili
题目:给定 n 个模式串和一个文本串,求有多少个不同的模式串在文本串里出现过。两个模式串不同当且仅当他们编号不同
// AC自动机 O(n) #include<bits/stdc++.h> using namespace std; const int N=500010; int n;char str[1000010]; int ch[N][26],cnt[N],idx; int ne[N]; void insert(char *s){ //建树 int p=0; for(int i=0;s[i];i++){ int j=s[i]-'a'; if(!ch[p][j]) ch[p][j]=++idx; p=ch[p][j]; } cnt[p]++; } void build(){ //建AC自动机 queue<int> q; for(int i=0;i<26;i++) if(ch[0][i])q.push(ch[0][i]); while(q.size()){ int u=q.front();q.pop(); for(int i=0;i<26;i++){ int v=ch[u][i]; if(v)ne[v]=ch[ne[u]][i],q.push(v); else ch[u][i]=ch[ne[u]][i]; } } } int query(char *s){ int ans=0; for(int k=0,i=0;s[k];k++){ i=ch[i][s[k]-'a']; for(int j=i;j&&~cnt[j];j=ne[j]) ans+=cnt[j],cnt[j]=-1; } return ans; } int main(){ cin>>n; for(int i=0;i<n;i++) cin>>str,insert(str); build(); cin>>str; cout<<query(str)<<endl; return 0; }
Luogu P3121 [USACO15FEB] Censoring G
浙公网安备 33010602011771号