2754: [SCOI2012]喵星球上的点名
Time Limit: 20 Sec Memory Limit: 128 MB
Submit: 1577 Solved: 710
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; }