bzoj3172:[Tjoi2013]单词

传送门

看着就是裸的AC自动机是不是?会T一个点
然而。。。为什么放过暴力KMP和暴力find,理论复杂度\(O(nm)\)起步啊
AC自动机加一个树上差分就好啦,对于每个字符串的查询,实际上都是在fail树上每个点到根的路径上+1
fail树建出来就没了
代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
void read(int &x) {
	char ch; bool ok;
	for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
	for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=1e6+10;queue<int>q;
int ed[201],fail[maxn],n,rt=1,id=1,ch[maxn][26],ans,vis[maxn],pre[maxn*2],nxt[maxn*2],h[maxn],cnt;char s[201][maxn];
void add(int x,int y)
{
	pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt;
	pre[++cnt]=x,nxt[cnt]=h[y],h[y]=cnt;
}
void insert(char *s,int d)
{
	int len=strlen(s+1);rt=1;
	for(rg int i=1;i<=len;i++)
	{
		int now=s[i]-'a';
		if(!ch[rt][now])ch[rt][now]=++id;
		rt=ch[rt][now];vis[rt]++;
	}
	ed[d]=rt;
}
void bfs()
{
	q.push(1);
	while(!q.empty())
	{
		int x=q.front();q.pop();
		for(rg int i=0;i<26;i++)
		{
			if(!ch[x][i]){ch[x][i]=fail[x]?ch[fail[x]][i]:1;continue;}
			int j=fail[x],z=ch[x][i];q.push(z);
			while(j&&!ch[j][i])j=fail[j];
			if(j)fail[z]=ch[j][i],add(z,ch[j][i]);
			else fail[z]=1,add(1,z);
		}
	}
}
void solve(int x,int fa)
{
	for(rg int i=h[x];i;i=nxt[i])
		if(pre[i]!=fa)solve(pre[i],x),vis[x]+=vis[pre[i]];
}
int main()
{
	read(n);
	for(rg int i=1;i<=n;i++)scanf("%s",s[i]+1),insert(s[i],i);
	bfs();solve(1,0);
	for(rg int i=1;i<=n;i++)printf("%d\n",vis[ed[i]]);
}
posted @ 2019-02-25 19:43  蒟蒻--lichenxi  阅读(114)  评论(0编辑  收藏  举报