2754: [SCOI2012]喵星球上的点名
Time Limit: 20 Sec Memory Limit: 128 MB
Submit: 680 Solved: 314
[Submit][Status]
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
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
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。
Source
题解：
没想到后缀数组的解法这么暴力也能过。。。
把姓+名+询问全部放到一个字符串里做后缀数组，对每一个询问向前、向后扫描并把答案放入set，set判重并输出size（）
一开始好像是中间填的字符越界了，导致一直WA了好几天。。。
本以为我对字符串重新编一下号就能使m减小而程序跑得更快，没想到还是慢成翔啊。。。
代码：
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<string>
#define inf 1000000000
#define maxn 1000000+1000
#define maxm 500+100
#define eps 1e-10
#define ll long long
#define pa pair<int,int>
#define for0(i,n) for(int i=0;i<=(n);i++)
#define for1(i,n) for(int i=1;i<=(n);i++)
#define for2(i,x,y) for(int i=(x);i<=(y);i++)
#define for3(i,x,y) for(int i=(x);i>=(y);i--)
#define mod 1000000007
using namespace std;
inline int read()
{
int x=,f=;char ch=getchar();
while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
while(ch>=''&&ch<=''){x=*x+ch-'';ch=getchar();}
return x*f;
}
set<int>s;
set<int>::iterator it;
int a[maxn],b[maxn],v[maxn],ans[maxn],rank[maxn],h[maxn],sa[maxn],len[maxn],pos[maxn],wr[maxn],wa[maxn],wb[maxn];
int n,m,cnt,tot,mid;
inline bool cmp(int *r,int a,int b,int l)
{
return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int n,int m)
{
int *x=wa,*y=wb,*t;
for0(i,m)wr[i]=;
for0(i,n)wr[x[i]=a[i]]++;
for1(i,m)wr[i]+=wr[i-];
for3(i,n,)sa[--wr[x[i]]]=i;
for(int p=,j=;p<=n;j<<=,m=p)
{
p=;
for2(i,n-j+,n)y[p++]=i;
for0(i,n)if(sa[i]>=j)y[p++]=sa[i]-j;
for0(i,m)wr[i]=;
for0(i,n)wr[x[y[i]]]++;
for1(i,m)wr[i]+=wr[i-];
for3(i,n,)sa[--wr[x[y[i]]]]=y[i];
int i;
for(t=x,x=y,y=t,p=,x[sa[]]=,i=;i<=n;i++)
x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
}
}
void geth(int n)
{
for1(i,n)rank[sa[i]]=i;
for(int i=,k=,j;i<n;h[rank[i++]]=k)
for(k?k--:,j=sa[rank[i]-];a[i+k]==a[j+k];k++);
}
int main()
{
freopen("input.txt","r",stdin);
freopen("output.txt","w",stdout);
n=read();m=read();tot=-;
for1(i,n)
{
int x=read();
for1(j,x)a[++tot]=read()+,b[tot]=i;
x=read();
a[++tot]=;
for1(j,x)a[++tot]=read()+,b[tot]=i;
a[++tot]=;
}
a[++tot]=;mid=tot;
for1(i,m)
{
len[i]=read();
pos[i]=tot+;
for1(j,len[i])a[++tot]=read()+;
a[++tot]=;
}
cnt=;
for0(i,tot)if(!v[a[i]])a[i]=v[a[i]]=++cnt;else a[i]=v[a[i]];
a[++tot]=;
da(tot,cnt);
geth(tot);
for1(i,m)
{
s.clear();
int x=rank[pos[i]],y=len[i];
for2(j,x+,tot)
{
if(h[j]<y)break;
if(sa[j]>=mid)continue;
if(b[sa[j]])s.insert(b[sa[j]]);
}
for3(j,x,)
{
if(h[j]<y)break;
if(sa[j-]>=mid)continue;
if(b[sa[j-]])s.insert(b[sa[j-]]);
}
for(it=s.begin();it!=s.end();it++)ans[*it]++;
printf("%d\n",s.size());
}
for1(i,n-)printf("%d ",ans[i]);printf("%d\n",ans[n]);
return ;
}