bzoj3172 單詞 AC自動機
阿新 • • 發佈:2019-02-01
(感覺以前寫過。。bzoj上不去我也不知道) 跑一遍AC自動機,每一個節點儲存一下屬於多少字串,為它的權值。然後一個節點表示的字串在整個字典中出現的次數相當於其在Fail樹中的子樹的權值的和。AC自動機不要寫掛就好了。
AC程式碼如下:
#include<iostream> #include<cstdio> #include<cstring> #define N 1100005 using namespace std; char s[N]; int n,a[N],h[N]; struct acam_node{ int cnt,last,ch[N][26],sz[N],fail[N]; void add(int x){ scanf("%s",s+1); int now=0,i,len=strlen(s+1); for (i=1; i<=len; i++){ int c=s[i]-'a'; if (!ch[now][c]) ch[now][c]=++cnt; now=ch[now][c]; sz[now]++; } a[x]=now; } void build(){ int i,head=0,tail=0; for (i=0; i<26; i++) if (ch[0][i]) h[++tail]=ch[0][i]; while (head<tail){ int x=h[++head],y; for (i=0; i<26; i++) if (y=ch[x][i]){ h[++tail]=y; fail[y]=ch[fail[x]][i]; } else ch[x][i]=ch[fail[x]][i]; } } void solve(){ int i; for (i=cnt; i>=0; i--) sz[fail[h[i]]]+=sz[h[i]]; for (i=1; i<=n; i++) printf("%d\n",sz[a[i]]); } }acam; int main(){ scanf("%d",&n); int i; for (i=1; i<=n; i++) acam.add(i); acam.build(); acam.solve(); return 0; }
by lych
2016.2.18