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

好久没写AC自动机啦..终于熟练掌握了WA自动机..最近通过率掉的异常的感人QAQ
map加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[N],st[N],V,M;
map<int,int>p[N];
int ans1[N],ans2[N];
int q[N],vis[N],mark[N],fail[N];
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;
}