1. 程式人生 > >bzoj3172 單詞 AC自動機

bzoj3172 單詞 AC自動機

      (感覺以前寫過。。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