bzoj 3172:[Tjoi 2013]単語

1157 ワード

転送ゲート: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
",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; }