[bzoj3172][Tjoi2013]单词——AC自动机

题目大意：

某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

思路：

第i个单词在整个文章中出现了多少次即i串的结尾可以被多少个串的节点给跳到。
于是吧fail看成每个节点唯一的父亲，每个节点的权值为有多少个单词的前缀经过了它，然后直接统计子树内的权值和即可。

#include<bits/stdc++.h>

#define REP(i,a,b) for(int i=a,i##_end_=b;i<=i##_end_;++i)
#define DREP(i,a,b) for(int i=a,i##_end_=b;i>=i##_end_;--i)
#define debug(x) cout<<#x<<"="<<x<<" "
#define fi first
#define se second
#define mk make_pair
#define pb push_back
typedef long long ll;

using namespace std;

void File(){
	freopen("bzoj3172.in","r",stdin);
	freopen("bzoj3172.out","w",stdout);
}

template<typename T>void read(T &_){
	_=0; T f=1; char c=getchar();
	for(;!isdigit(c);c=getchar())if(c=='-')f=-1;
	for(;isdigit(c);c=getchar())_=(_<<1)+(_<<3)+(c^'0');
	_*=f;
}

const int maxn=1e6+10;
int n,cnt_rank,ans[maxn],weight[maxn];
int ch[maxn][26],num[maxn],fail[maxn],cnt=1;
vector<int>rank[maxn];

void insert(char *s){
	int len=strlen(s+1),u=1,c;
	REP(i,1,len){
		c=s[i]-'a';
		if(!ch[u][c])ch[u][c]=++cnt;
		u=ch[u][c];
		++weight[u];
	}
	++num[u];
	rank[u].pb(++cnt_rank);
}

void build_fail(){
	int h=1,t=0,q[maxn];
	fail[1]=1;
	REP(i,0,25){
		int v=ch[1][i];
		if(v)q[++t]=v;
		if(v)fail[v]=1;
		else ch[1][i]=1;
	}
	while(h<=t){
		int u=q[h++];
		REP(i,0,25){
			int v=ch[u][i];
			if(v)q[++t]=v;
			if(v)fail[v]=max(ch[fail[u]][i],1);
			else ch[u][i]=max(ch[fail[u]][i],1);
		}
	}
}

int beg[maxn],to[maxn],las[maxn],cnte=1;
int sz[maxn];

void add(int u,int v){
	las[++cnte]=beg[u]; beg[u]=cnte; to[cnte]=v;
}

void dfs(int u){
	sz[u]=weight[u];
	for(int i=beg[u];i;i=las[i]){
		dfs(to[i]);
		sz[u]+=sz[to[i]];
	}
	REP(i,0,rank[u].size()-1)ans[rank[u][i]]=sz[u];
}

int main(){
	File();

	char s[maxn];

	read(n);
	REP(i,1,n){
		scanf("%s",s+1);
		//printf("%s
",s+1);
		insert(s);
	}

	build_fail();
	REP(i,2,cnt)add(fail[i],i);

	dfs(1);

	REP(i,1,n)printf("%d
",ans[i]);

	return 0;
}