【刷题】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;

}