2754: [SCOI2012]喵星球上的点名
Time Limit: 20 Sec
Memory Limit: 128 MB
Submit: 1577
Solved: 710
[ Submit][ Status][ Discuss]
Description
a180285幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非
常有趣。
假设课堂上有N个喵星人,每个喵星人的名字由姓和名构成。喵星球上的老师会选择
M个串来点名,每次读出一个串的时候,如果这个串是一个喵星人的姓或名的子串,那么
这个喵星人就必须答到。
然而,由于喵星人的字码过于古怪,以至于不能用ASCII码来表示。为了方便描述,
a180285决定用数串来表示喵星人的名字。
现在你能帮助a180285统计每次点名的时候有多少喵星人答到,以及M次点名结束后
每个喵星人答到多少次吗?
Input
现在定义喵星球上的字符串给定方法:
先给出一个正整数L,表示字符串的长度,接下来L个整数表示字符串的每个字符。
输入的第一行是两个整数N和M。
接下来有N行,每行包含第i 个喵星人的姓和名两个串。姓和名都是标准的喵星球上的
字符串。
接下来有M行,每行包含一个喵星球上的字符串,表示老师点名的串。
Output
对于每个老师点名的串输出有多少个喵星人应该答到。
然后在最后一行输出每个喵星人被点到多少次。
Sample Input
2 3
6 8 25 0 24 14 8 6 18 0 10 20 24 0
7 14 17 8 7 0 17 0 5 8 25 0 24 0
4 8 25 0 24
4 7 0 17 0
4 17 0 8 25
Sample Output
2
1
0
1 2
【提示】
事实上样例给出的数据如果翻译成地球上的语言可以这样来看
2 3
izayoi sakuya
orihara izaya
izay
hara
raiz
HINT
【数据范围】
对于30%的数据,保证:
1<=N,M<=1000,喵星人的名字总长不超过4000,点名串的总长不超过2000。
对于100%的数据,保证:
1<=N<=20000,1<=M<=50000,喵星人的名字总长和点名串的总长分别不超过100000,保证喵星人的字符串中作为字符存在的数不超过10000。
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;
}