bzoj2754【SCOI2012】喵星球上的点名

标签：

Description

Input

Output

Sample Input

Sample Output

HINT

AC自动机

一个小技巧：对于一个喵星人，姓和名的子串都符合要求，可以将姓和名拼在一起，中间加一个没有出现过的数。这样就把两个串的问题变成一个串。

字符集太大，用map存储。

初始化太慢，用vector保存修改了哪些位置，计算完之后再修改。

然后就是比较基础的问题了：给出一些字符串，每次给一个字符串s，问有多少个串含有子串s。

方法是将模式串放入AC自动机(和KMP类似)，然后去跑文本串。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define maxn 100005
using namespace std;
int n,m,cnt=1,ans1[maxn],ans2[maxn],fail[maxn];
bool vst[maxn],mark[maxn];
vector<int> a[maxn],tag[maxn],V,M;
map<int,int> t[maxn];
queue<int> q;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void insert(int id)
{
	int len=read(),x,now=1;
	F(i,1,len)
	{
		x=read();
		if (!t[now][x]) t[now][x]=++cnt;
		now=t[now][x];
	}
	tag[now].push_back(id);
}
void getfail()
{
	q.push(1);
	while (!q.empty())
	{
		int x=q.front();q.pop();
		for(map<int,int>::iterator i=t[x].begin();i!=t[x].end();i++)
		{
			int tmp=i->first,y=i->second,j=fail[x];
			while (j&&!t[j][tmp]) j=fail[j];
			fail[y]=j?t[j][tmp]:1;
			q.push(y);
		}
	}
}
void getans(int id,int x)
{
	for(int i=x;i;i=fail[i]) if (!vst[i])
	{
		vst[i]=true;V.push_back(i);
		for(int j=0;j<tag[i].size();j++) if (!mark[tag[i][j]])
		{
			mark[tag[i][j]]=true;M.push_back(tag[i][j]);
			ans1[tag[i][j]]++;ans2[id]++;
		}
	}
}
void solve(int id)
{
	int now=1;
	for(int i=0;i<a[id].size();i++)
	{
		int x=a[id][i];
		while (now&&!t[now][x]) now=fail[now];
		now=now?t[now][x]:1;
		getans(id,now);
	}
	for(int i=0;i<V.size();i++) vst[V[i]]=false;
	for(int i=0;i<M.size();i++) mark[M[i]]=false;
	V.clear();M.clear();
}
int main()
{
	n=read();m=read();
	int len,x;
	F(i,1,n)
	{
		len=read();F(j,1,len) x=read(),a[i].push_back(x);
		a[i].push_back(100001);
		len=read();F(j,1,len) x=read(),a[i].push_back(x);
	}
	F(i,1,m) insert(i);
	getfail();
	F(i,1,n) solve(i);
	F(i,1,m) printf("%d\n",ans1[i]);
	F(i,1,n-1) printf("%d ",ans2[i]);
	printf("%d\n",ans2[n]);
	return 0;
}

bzoj2754【SCOI2012】喵星球上的点名

标签：

原文地址：http://blog.csdn.net/aarongzk/article/details/51494726