BZOJ 2754([SCOI2012]喵喵叫的星球-统计序列的后缀阵列中子序列出现次数)

2754: [SCOI2012]喵喵叫的星球

Time Limit: 20 Sec Memory Limit: 128 MB

Submit: 805 Solved: 380

[

id=2754" style="color:blue; text-decoration:none">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。

Source

search=" style="color:blue; text-decoration:none">

这题是后缀数组。

先把模式串P写成s11*s12*s21*s22...sn2 (*为特殊字符)

然后用find统计sa中出现上下界平均O(Len总logLen总+MlogLen总) 最坏O(MLen总)

可能会T但正解至今仍不知道

#include<cstdio>

#include<cstring>

#include<cstdlib>

#include<algorithm>

#include<functional>

#include<iostream>

#include<cmath>

#include<cctype>

#include<ctime>

using namespace std;

#define For(i,n) for(int i=1;i<=n;i++)

#define Fork(i,k,n) for(int i=k;i<=n;i++)

#define Rep(i,n) for(int i=0;i<n;i++)

#define ForD(i,n) for(int i=n;i;i--)

#define RepD(i,n) for(int i=n;i>=0;i--)

#define Forp(x) for(int p=pre[x];p;p=next[p])

#define Forpiter(x) for(int &p=iter[x];p;p=next[p])

#define Lson (x<<1)

#define Rson ((x<<1)+1)

#define MEM(a) memset(a,0,sizeof(a));

#define MEMI(a) memset(a,127,sizeof(a));

#define MEMi(a) memset(a,128,sizeof(a));

#define INF (2139062143)

#define F (100000007)

#define MAXN (1000000+10)

#define Sigma_size (10000+10)

#define sp_char (10001)

typedef long long ll;

ll mul(ll a,ll b){return (a*b)%F;}

ll add(ll a,ll b){return (a+b)%F;}

ll sub(ll a,ll b){return (a-b+(a-b)/F*F+F)%F;}

void upd(ll &a,ll b){a=(a%F+b%F)%F;}

// 约定 -1:/0	sp_char:*	cat[i]=0 该位置无猫

class SA

{

public:

	int s[MAXN];

	int sa[MAXN],t[MAXN],t2[MAXN],c[MAXN],n;

	SA(){}

	SA(int *_s,int _n){memcpy(s,_s,sizeof(s));n=_n;MEM(sa) MEM(t) MEM(t2) MEM(c) }

	void mem(int *_s,int _n){memcpy(s,_s,sizeof(s));n=_n;MEM(sa) MEM(t) MEM(t2) MEM(c) }

	void build_sa(int m)

	{

		int *x=t,*y=t2;

		Rep(i,m) c[i]=0;

		Rep(i,n) c[x[i]=s[i]]++; //x[i] 第i个字符串第一keyword排第几

		For(i,m-1) c[i]+=c[i-1];

		RepD(i,n-1) sa[--c[x[i]]]=i;

		for(int k=1;k<=n;k<<=1)

		{

			int p=0;

			Fork(i,n-k,n-1) y[p++]=i;

			Rep(i,n) if (sa[i]>=k) y[p++]=sa[i]-k; //y[i] 第二keyword排i的是第？字符串 

			Rep(i,m) c[i]=0;

			Rep(i,n) c[x[y[i]]]++; // x[y[i]] 第二keyword排i的字符串，第1keyword排第几

			For(i,m-1) c[i]+=c[i-1];

			RepD(i,n-1) sa[--c[x[y[i]]]]=y[i]; //第二keyword从大到小遍历。以第一keyword排序 ，最后得到以1，2keyword排序的

			//此时sa为第一keyword。第2keyword均排好序的

			swap(x,y);  //此时y变为第一keyword排序的rank，目标是把x变成第一，二keyword排序的rank

			p=1; x[sa[0]]=0;

			For(i,n-1)

				x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i]+k]==y[sa[i-1]+k] ? p-1:p++;

			if (p>=n) break;

			m=p;

		}

	}

	int m; //模板串P的长度要事先赋值

	int cmp_suffix(int *pattern,int p)

	{

		for(int i=0;i<m;i++)

		{

			if (pattern[i]!=s[sa[p]+i]) return pattern[i]-s[sa[p]+i];

		}

		return 0;

	}

	int find(int *P,int _m)

	{

		m=_m; //这里赋值也行

		if (cmp_suffix(P,0)<0||cmp_suffix(P,n-1)>0) return -1;

		int L=0,R=n-1;

		while(L<=R)

		{

			int M=(L+R)>>1;

			int res=cmp_suffix(P,M);

			if (!res) return M;

			else if (res<0) R=M-1;

			else L=M+1;

		}

		return -1;

	}

	int rank[MAXN],height[MAXN];

	void make_height()

	{

		int k=0;

		Rep(i,n) rank[sa[i]]=i;

		Rep(i,n)

		{

			if (rank[i]-1<0) continue;

			if (k) k--;

			int j=sa[rank[i]-1];

			while(s[i+k]==s[j+k]) ++k;

			height[rank[i]]=k;

		}

	}

}S;

int flag[MAXN]={0},cat[MAXN]={0},sum[MAXN]={0};

int len,s[MAXN];

int n,m;

int get_init(int &p,int i,bool is_cat)

{

	int len;

	scanf("%d",&len);

	while(len--)

	{

		scanf("%d",&s[++p]);

		if (is_cat) cat[p]=i;

	}

	return len;

}

int main()

{

//	freopen("bzoj2754.in","r",stdin);

//	freopen(".out","w",stdout);

	scanf("%d%d",&n,&m);

	int p=-1;

	For(i,n)

	{

		For(j,2)

		{

			get_init(p,i,1);

			s[++p]=sp_char;cat[p]=0;

		}

	}

	s[++p]=-1; 

	S.mem(s,p);

	S.build_sa(Sigma_size);

	S.make_height();

	For(i,m)

	{

		p=-1;

		get_init(p,i,0);

		s[++p]=-1;

		int K=S.find(s,p);

		if (K==-1)

		{

			cout<<"0\n";

			continue;

		}

		int ans=0;

		int t=S.sa[K];

		if (flag[cat[t]]^i) sum[cat[t]]++,flag[cat[t]]=i,++ans;			

		int k=K;

		while (k>0&&S.cmp_suffix(s,k-1)==0)

		{

			k--;

			int t=S.sa[k];

			if (flag[cat[t]]^i) sum[cat[t]]++,flag[cat[t]]=i,++ans;

		}

		k=K;

		while (k<S.n&&S.cmp_suffix(s,k+1)==0)

		{

			k++;

			int t=S.sa[k];

			if (flag[cat[t]]^i) sum[cat[t]]++,flag[cat[t]]=i,++ans;

		}

		printf("%d\n",ans);

	}

	printf("%d",sum[1]);

	Fork(i,2,n) printf(" %d",sum[i]);

	putchar('\n');

	return 0;

}