2754: [SCOI2012]喵星球上的点名|AC自动机

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<vector>
#include<cstdio>
#include<queue>
#include<cmath>
#include<set>
#include<map>
#define N 100005
using namespace std;
int sc()
{
int i=0,f=1;char c=getchar();
while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
return i*f;
}
vector<int>a
,st
,V,M;
map<int,int>p
;
int ans1
,ans2
;
int q
,vis
,mark
,fail
;
int n,m,cnt=1;
void insert(int pos)
{
int l=sc(),x,now=1;
for(int i=1;i<=l;i++)
{
x=sc();
if(!p[now][x])p[now][x]=++cnt;
now=p[now][x];
}
st[now].push_back(pos);
}
void Ac_machine()
{
fail[q[1]=1]=0;
for(int l=1,r=2;l<r;l++)
{
int x=q[l];
for(map<int,int>::iterator i=p[x].begin();i!=p[x].end();i++)
{
int t=i->first,k=fail[x];
while(!p[k][t])k=fail[k];
fail[i->second]=p[k][t];
q[r++]=i->second;
}
}
}
void cal(int x,int y)
{
for(int i=y;i;i=fail[i])
if(!vis[i])
{
vis[i]=1;V.push_back(i);
for(int j=0;j<st[i].size();j++)
if(!mark[st[i][j]])
{
mark[st[i][j]]=1;
M.push_back(st[i][j]);
ans1[st[i][j]]++;
ans2[x]++;
}
}
else break;
}
void solve(int x)
{
int now=1;
for(int i=0;i<a[x].size();i++)
{
int t=a[x][i];
while(!p[now][t])now=fail[now];
now=p[now][t];
cal(x,now);
}
for(int i=0;i<V.size();i++)vis[V[i]]=0;
for(int i=0;i<M.size();i++)mark[M[i]]=0;
V.clear();M.clear();
}
int main()
{
n=sc(),m=sc();
for(int i=1;i<=n;i++)
{
int x=sc(),y;
while(x--)y=sc(),a[i].push_back(y);
a[i].push_back(10001);x=sc();
while(x--)y=sc(),a[i].push_back(y);
}
for(int i=0;i<=10001;i++)
p[0][i]=1;
for(int i=1;i<=m;i++)
insert(i);
Ac_machine();
for(int i=1;i<=n;i++)
solve(i);
for(int i=1;i<=m;i++)
printf("%d\n",ans1[i]);
for(int i=1;i<=n;i++)
{
printf("%d",ans2[i]);
if(i!=n)printf(" ");
}
return 0;
}