bzoj 2754: [SCOI2012]喵星球上的点名 （fail树+树状数组+lca+dfs序）

2754: [SCOI2012]喵星球上的点名

Time Limit: 20 Sec   Memory Limit: 128 MB
Submit: 1710   Solved: 765
[ 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

[ Submit][ Status][ Discuss]

题解：map加next数组建立AC自动机，然后建立fail树。注意fail树的边的指向与fail指针是反向的。

判断A串在B串内出现了多少次，就需要统计fail树中A串的结尾节点的子树中有多少个节点属于B串。

这道题第一问其实就是需要求出fail的子树中有多少不同人的节点，我们对于fail求出dfs序，问题就转换成了求区间颜色中数的经典问题，参见hh的项链。

第二问的话我们把询问串结尾节点标记，然后将姓名串所有的节点按照dfs序排序，计算每个点到根的路径上经过了多少标记，因为有可能重复计算，所有我们我们加入一点根路径上的标记数的同时，还需要减去他与离他最近点的lca到根的标记数，算是一个简单的容斥吧。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue>
#include<map>
#define N 300003
using namespace std;
int m,n,a[N],fail[N];
int next[N],point[N],v[N],belong[N],nextn[N],pointn[N],vn[N],u[N];
int head[N],nxt[N],c[N],k,tot,totn,tot1,tot2,cnt,size;
int end[N],q[N],l[N],r[N],dfsn[N],mark[N],h[N],count1[N],mi[N],number[N],ci[N];
int ok[N],pt[N],col[N],pos[N],tr[N],deep[N],fa[N],pd[N],ky[N];
bool ext[N];
map<int,int> mp[200003];
struct data
{
	int l,r,ans,num,pos;
}b[N];
void add(int x,int y,int z)
{
	tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; u[tot]=z; 
	//cout<<x<<" "<<y<<endl;
}
void build(int x,int y)
{
	totn++; nextn[totn]=pointn[x]; pointn[x]=totn; vn[totn]=y;
	//cout<<x<<" "<<y<<endl;
}
int insert(int len,int k)
{
	int now=0;
	for (int i=1;i<=len;i++)
	 {
	 	if (!mp[now][a[i]])  mp[now][a[i]]=++cnt,add(now,cnt,a[i]);
	 	now=mp[now][a[i]];  
	 	q[++size]=now; build(now,k);
	 }
	end[k]=size;
	return now;
}
void addfail(int x,int y)
{
	if (!y)  return;
	tot1++; nxt[tot1]=head[x]; head[x]=tot1; c[tot1]=y;
//	cout<<x<<" "<<y<<endl;
}
void dfs(int x)
{
	dfsn[++k]=x; pos[x]=k; l[x]=k;
	for (int i=head[x];i;i=nxt[i])
	 dfs(c[i]);
	r[x]=k;
}
void makefail()
{
	queue<int> p;
	for (int i=point[0];i;i=next[i])
	  p.push(v[i]),addfail(0,v[i]);
	while (!p.empty())
	{
		int now=p.front(); p.pop();
		for (int i=point[now];i;i=next[i])
		 {
		 	int k=fail[now];
		 	while (!mp[k][u[i]]&&k!=0) k=fail[k];
		 	fail[v[i]]=mp[k][u[i]];  addfail(mp[k][u[i]],v[i]);
		 	p.push(v[i]);
		 }
	}
}
int cmp(data a,data b)
{
	return a.ans<b.ans||a.ans==b.ans&&a.r<b.r;
}
void init(int x,int y,int z)
{
	tot2++; ok[tot2]=h[x];  h[x]=tot2; col[tot2]=y; ky[tot2]=z;
}
int lowbit(int x)
{
	return x&(-x);
}
void change(int x,int v)
{
	if (x==0)  return;
	for (int i=x;i<=k;i+=lowbit(i))
	 tr[i]+=v;
}
int sum(int x)
{
	if (x==0) return 0;
	int ans=0;
	for (int i=x;i>=1;i-=lowbit(i))
	 ans+=tr[i];
	return ans;
}
void buildtree(int x,int num)
{
	number[x]=num+count1[x]; 
	for (int i=head[x];i;i=nxt[i])
	 {
	 	deep[c[i]]=deep[x]+1;
	 	buildtree(c[i],num+count1[x]);
	 	fa[c[i]]=x;
	 }
}
int cmp1(data a,data b)
{
	return a.num<b.num;
}
int cmp2(int x,int y)
{
	return pos[x]<pos[y];
}
int lca(int x,int y)
{
	if (deep[x]<deep[y])  swap(x,y);
	for (int i=1;i=deep[x]-deep[y];i++)
	 x=fa[x];
	if (x==y)  return x;
	while (fa[x]!=fa[y])
	 {
	 	x=fa[x]; 
	 	y=fa[y];
	 }
	return fa[x];
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("my.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
	{
		int x; scanf("%d",&x);
		for (int j=1;j<=x;j++)
		 scanf("%d",&a[j]);
		insert(x,i);
		scanf("%d",&x);
		for (int j=1;j<=x;j++)  scanf("%d",&a[j]); 
		insert(x,i);
	}
	for (int i=1;i<=m;i++)
	 {
	 	int x; scanf("%d",&x);
	 	for (int j=1;j<=x;j++)  scanf("%d",&a[j]);
	    b[i].pos=insert(x,n+i);
	    count1[b[i].pos]++;
	 }
    makefail();  
    k=0; dfs(0); 
    buildtree(0,0); 
    for (int i=1;i<=m;i++)
	 b[i].l=l[b[i].pos],b[i].r=r[b[i].pos],b[i].num=i;
	sort(b+1,b+m+1,cmp);
    for (int i=1;i<=k;i++)
	 {
	 	for (int j=pointn[dfsn[i]];j;j=nextn[j])
	 	 init(i,ci[vn[j]],vn[j]),ci[vn[j]]=i;
	 }
	int top=1;
	while (b[top].ans==-1) top++;
	for (int i=1;i<=k;i++)
	  {
	  	for (int j=h[i];j;j=ok[j])
	  	 if (ky[j]<=n)
	  	  change(col[j],-1),change(i,1);
	  	while (b[top].r<=i&&top<=m)
	  	  {
	  	  	 b[top].ans=sum(b[top].r)-sum(b[top].l-1);
	  	  	 top++;
		  }
	  }
	sort(b+1,b+m+1,cmp1);
	for (int i=1;i<=m;i++)   printf("%d\n",b[i].ans); 
	for (int i=1;i<=n;i++)
	 {
	 	sort(q+end[i-1]+1,q+end[i]+1,cmp2);
	 	int ans=number[q[end[i-1]+1]]; ext[q[end[i-1]+1]]=1;
	 	for (int j=end[i-1]+2;j<=end[i];j++)
	 	 {
	 	 if (ext[q[j]])  continue;
		 int t=lca(q[j],q[j-1]); 
	 	 ans+=number[q[j]]-number[lca(q[j],q[j-1])];
	 	 ext[q[j]]=1;
	     }
	    for (int j=end[i-1]+1;j<=end[i];j++)  ext[q[j]]=0;
	 	if (i!=n)  printf("%d ",ans);
	 	else printf("%d\n",ans);
	 }
}