BZOJ2754【AC自动机】

最新推荐文章于 2020-01-08 15:08:12 发布

Lethelody

最新推荐文章于 2020-01-08 15:08:12 发布

阅读量456

点赞数

分类专栏：字符串

本文链接：https://blog.csdn.net/Lethelody/article/details/44906583

版权

字符串专栏收录该内容

15 篇文章 0 订阅

订阅专栏

可以用map来维护节点的next域.注意不能路径压缩.

/* I will wait for you */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iostream>
#include<fstream>
#include<vector>
#include<queue>
#include<deque>
#include<set>
#include<map>
#include<string>
#define make make_pair
#define fi first
#define se second

using namespace std;

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int maxn=100010;
const int maxm=1010;
const int maxs=26;
const int INF=1<<29;
const int P=1000000007;
const double error=1e-9;

int cnt,fail[maxn],q[maxn];
int vis[maxn],mark[maxn],ansa[maxn],ansb[maxn];
vector<int> na[maxn],end[maxn],ma,vi;
map<int,int> next[maxn];

void init()
{
	cnt=1;fail[1]=0;
	for(int i=-1;i<=10000;i++) next[0][i]=1;
}

void insert(int v)
{
	int n,p=1;scanf("%d",&n);
	for(int i=0;i<n;i++)
	{
		int c;scanf("%d",&c);
		if(!next[p][c]) next[p][c]=++cnt;
		p=next[p][c];
	}
	end[p].push_back(v);
}
		
int getfail()
{
	int head=1,tail=0;q[++tail]=1;
	
	while(head<=tail)
	{
		int u=q[head++]; 
		for(typeof(next[u].end()) i=next[u].begin();i!=next[u].end();i++)
		{
			int x=i->fi,c=fail[u];
			while(!next[c][x]) c=fail[c];
			fail[i->se]=next[c][x];q[++tail]=i->se;
		}
	}
}

void count(int x,int u)
{
	for(int i=x;i&&!vis[i];i=fail[i])
	{
		vis[i]=1;vi.push_back(i);
		for(int j=0;j<end[i].size();j++) if(!mark[end[i][j]])
			mark[end[i][j]]=1,ansa[end[i][j]]++,ansb[u]++,ma.push_back(end[i][j]);
	}
}

void slove(int u)
{
	int p=1,n=na[u].size();
	for(int i=0;i<n;i++)
	{
		int x=na[u][i];while(!next[p][x]) p=fail[p];
		p=next[p][x];count(p,u);
	}
	for(int i=0;i<vi.size();i++) vis[vi[i]]=0;
	for(int i=0;i<ma.size();i++) mark[ma[i]]=0;
	vi.clear();ma.clear();
}

int main()
{
	int n,m;scanf("%d%d",&n,&m);
	for(int i=1,u;i<=n;i++)
	{
		scanf("%d",&u);for(int j=0,c;j<u;j++) scanf("%d",&c),na[i].push_back(c);
		na[i].push_back(-1);
		scanf("%d",&u);for(int j=0,c;j<u;j++) scanf("%d",&c),na[i].push_back(c);
	}
	
	init();for(int i=1;i<=m;i++) insert(i);getfail();
	
	for(int i=1;i<=n;i++) slove(i);
	
	for(int i=1;i<=m;i++) printf("%d\n",ansa[i]);
	for(int i=1;i<=n;i++) printf("%d%c",ansb[i],i==n?10:' ');

	return 0;
}