bzoj3172: [Tjoi2013]单词

传送门:http://www.lydsy.com:808/JudgeOnline/problem.php?id=3172

思路:每加入一个串,就给这个串的路径上所有点的sum+1,表示它的前缀都又出现了一次,然后从下向上把sum加到fail指针指向的点上就可以得到答案了。

#include<cstdio>
#include<cstring>
#include<algorithm>
const int maxn=1000010;
using namespace std;
int n,w[210];char s[maxn];

struct AC_DFA{
	int tot,ch[maxn][26],fail[maxn],q[maxn],head,tail,sum[maxn];
	void insert(int &x){
		int p=0;
		int len=strlen(s);
		for (int i=0;i<len;i++){
			if (!ch[p][s[i]-'a']) ch[p][s[i]-'a']=++tot;
			++sum[p=ch[p][s[i]-'a']];
		}
		x=p;
	}
	void getfail(){
		head=0,q[tail=1],fail[0]=-1;
		while (head!=tail){
			int x=q[++head];
			for (int i=0;i<26;i++){
				if (ch[x][i]) q[++tail]=ch[x][i],fail[ch[x][i]]=x==0?0:ch[fail[x]][i];
				else ch[x][i]=x==0?0:ch[fail[x]][i];
			}
		}
	}
	void work(){
		for (int i=tail;i;i--) sum[fail[q[i]]]+=sum[q[i]];
		for (int i=1;i<=n;i++) printf("%d\n",sum[w[i]]);
	}
}T;

int main(){
	scanf("%d",&n);
	for (int i=1;i<=n;i++) scanf("%s",s),T.insert(w[i]);
	T.getfail(),T.work();
	return 0;
}


posted @ 2015-07-26 15:17  orzpps  阅读(85)  评论(0编辑  收藏  举报