【刷题】BZOJ 2754 [SCOI2012]喵星球上的点名

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。

Solution

写了个暴力乱搞

把所有的串接在一起，用大数隔开，然后求SA

求出SA后，枚举所有点名串，暴力前后找含有点名串的（就是LCP大于等于点名串的长度），然后统计答案

这个做法原题数据是可过的，BZOJ上也可过

然而洛谷加了一组大数据，把这个做法卡掉了。唉，就交了BZOJ算了

#include<bits/stdc++.h>
#define ui unsigned int
#define ll long long
#define db double
#define ld long double
#define ull unsigned long long
const int MAXN=50000+10,MAXM=100000+10;
int s[MAXM<<2],ml[MAXM],rk[MAXM<<2],cnt[MAXM<<2],dpt,n,m,sn,sm,ans1[MAXM],ans2[MAXN],nxt[MAXM<<2],height[MAXM<<2],bel[MAXM<<2],vis[MAXM<<2],SA[MAXM<<2],stm[MAXM];
template<typename T> inline void read(T &x)
{
	T data=0,w=1;
	char ch=0;
	while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
	if(ch=='-')w=-1,ch=getchar();
	while(ch>='0'&&ch<='9')data=((T)data<<3)+((T)data<<1)+(ch^'0'),ch=getchar();
	x=data*w;
}
template<typename T> inline void write(T x,char ch='\0')
{
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
	if(ch!='\0')putchar(ch);
}
template<typename T> inline void chkmin(T &x,T y){x=(y<x?y:x);}
template<typename T> inline void chkmax(T &x,T y){x=(y>x?y:x);}
template<typename T> inline T min(T x,T y){return x<y?x:y;}
template<typename T> inline T max(T x,T y){return x>y?x:y;}
inline void GetSA()
{
	m=10010;
	for(register int i=1;i<=n;++i)rk[i]=s[i];
	for(register int i=1;i<=n;++i)cnt[rk[i]]++;
	for(register int i=1;i<=m;++i)cnt[i]+=cnt[i-1];
	for(register int i=n;i>=1;--i)SA[cnt[rk[i]]--]=i;
	for(register int k=1,ps;k<=n;k<<=1)
	{
		ps=0;
		for(register int i=n-k+1;i<=n;++i)nxt[++ps]=i;
		for(register int i=1;i<=n;++i)
			if(SA[i]>k)nxt[++ps]=SA[i]-k;
		for(register int i=1;i<=m;++i)cnt[i]=0;
		for(register int i=1;i<=n;++i)cnt[rk[i]]++;
		for(register int i=1;i<=m;++i)cnt[i]+=cnt[i-1];
		for(register int i=n;i>=1;--i)SA[cnt[rk[nxt[i]]]--]=nxt[i];
		std::swap(rk,nxt);
		rk[SA[1]]=1,ps=1;
		for(register int i=2;i<=n;rk[SA[i]]=ps,++i)
			if(nxt[SA[i]]!=nxt[SA[i-1]]||nxt[SA[i]+k]!=nxt[SA[i-1]+k])ps++;
		if(ps>=n)break;
		m=ps;
	}
	for(register int i=1,j,k=0;i<=n;height[rk[i++]]=k)
		for(k=k?k-1:k,j=SA[rk[i]-1];s[i+k]==s[j+k];++k);
}
int main()
{
	read(sn);read(sm);
	dpt=10005;
	for(register int i=1,ln;i<=sn;++i)
	{
		read(ln);
		for(register int j=1,x;j<=ln;++j)read(x),s[++n]=x,bel[n]=i;
		s[++n]=dpt;
		read(ln);
		for(register int j=1,x;j<=ln;++j)read(x),s[++n]=x,bel[n]=i;
		s[++n]=dpt;
	}
	for(register int i=1,ln;i<=sm;++i)
	{
		read(ln);stm[i]=n+1;ml[i]=ln;
		for(register int j=1,x;j<=ln;++j)read(x),s[++n]=x;
		s[++n]=dpt;
	}
	GetSA();
	for(register int i=1;i<=sm;++i)
	{
		int l=rk[stm[i]],r=rk[stm[i]],len=ml[i];
		while(height[l]>=len)--l;
		while(height[r+1]>=len)++r;
		for(register int j=l;j<=r;++j)
			if(bel[SA[j]]&&(vis[bel[SA[j]]]^i))vis[bel[SA[j]]]=i,ans1[i]++,ans2[bel[SA[j]]]++;
	}
	for(register int i=1;i<=sm;++i)write(ans1[i],'\n');
	for(register int i=1;i<=sn;++i)write(ans2[i],' ');puts("");
	return 0;
}