SCOI 2012 Day 1 喵星球上的点名

题目摘自：http://61.187.179.132/JudgeOnline/problem.php?id=2754

 

题目描述：

Time Limit: 20 Sec  Memory Limit: 128 MB

 

　　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

【数据范围】 

 对于30%的数据，保证： 

1<=N,M<=1000，喵星人的名字总长不超过4000，点名串的总长不超过2000。

对于100%的数据，保证：

1<=N<=20000，1<=M<=50000，喵星人的名字总长和点名串的总长分别不超过100000，保证喵星人的字符串中作为字符存在的数不超过10000。

  

　　这是四川省今年的省选题。话说当时我去做的时候什么都不知道，连KMP都没写过就直接面对这个伟大的匹配问题，自然只有裸了，结果得了我考试中唯一的30分。下来后自己看了看KMP，又写了一下，对字符串匹配有了基本的了解，然后准备挑战这道题。据说是可以用后缀数组做的，但是MS我同学在OJ上没有过，于是果断套上AC自动机。本题其实就是把m个串建成自动机，然后拿n个串依次询问即可。

 

　　参考了zhx大神的代码，自己又拿回去写了一遍，终于通过了。算是来个纪念吧。

 

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<map>
#include<queue>

using namespace std;

#define maxm 50001
#define maxn 20001
#define maxl 100001

int n,m,l,q[maxm],nowname[maxl],next[maxm],all,check[maxm],v[maxl],answer[maxn],ask[maxm];

struct na
{
    int p1,l1,p2,l2;
}name[maxn];

struct node
{
    int count;
    map<int,node*> next;
    node *fail;

    node()
    {
        count=0;
        fail=NULL;
    }
};

queue<node*> que;

void insert(node *root,int *nowstr,int now)
{
    int l=0;
    node* p=root;

    while(nowstr[l]!=-1)
    {
        if(p->next[nowstr[l]]==NULL) p->next[nowstr[l]]=new node;
        p=p->next[nowstr[l]];
        l++;
    }

    all++;
    q[all]=now;
    next[all]=p->count;
    p->count=all;
}

void build_AC(node* root)
{
    que.push(root);
    root->fail=NULL;

    while(!que.empty())
    {
        node* p=que.front();
        que.pop();

        for(map<int,node*>::iterator i=p->next.begin();i!=p->next.end();i++)
            if(i->second!=NULL)
            {
                node *x=i->second;
                int v=i->first;

                node *last=p->fail;
                while(last!=NULL)
                {
                    if(last->next[v]!=NULL)
                    {
                        x->fail=last->next[v];
                        break;
                    }
                    last=last->fail;
                }

                if(last==NULL) x->fail=root;
                int nowp=x->count,lastp=0;
                while(nowp!=0)
                {
                    lastp=nowp;
                    nowp=next[nowp];
                }
                if(lastp==0) x->count=x->fail->count;
                else next[lastp]=x->fail->count;
                que.push(x);
            }
    }
}

void query(node *root,int now)
{
    int l=name[now].p1;
    node *p=root;

    while(l<=name[now].p1+name[now].l1-1)
    {
        while(p->next[v[l]]==NULL&&p!=root)
            p=p->fail;

        p=p->next[v[l]];
        if(p==NULL) p=root;
        else
        {
            int nowp=p->count;
            while(nowp!=0)
            {
                if(check[q[nowp]]!=now)
                {
                    check[q[nowp]]=now;
                    answer[now]++;
                    ask[q[nowp]]++;
                }
                nowp=next[nowp];
            }
        }
        l++;
    }

    l=name[now].p2;
    p=root;

    while(l<=name[now].p2+name[now].l2-1)
    {
        while(p->next[v[l]]==NULL&&p!=root)
            p=p->fail;

        p=p->next[v[l]];
        if(p==NULL) p=root;
        else
        {
            int nowp=p->count;
            while(nowp!=0)
            {
                if(check[q[nowp]]!=now)
                {
                    check[q[nowp]]=now;
                    answer[now]++;
                    ask[q[nowp]]++;
                }
                nowp=next[nowp];
            }
        }
        l++;
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    l=-1;

    for(int i=1;i<=n;i++)
    {
        scanf("%d",&name[i].l1);
        name[i].p1=l+1;
        for(int j=1;j<=name[i].l1;j++)
        {
            l++;
            scanf("%d",&v[l]);
        }

        scanf("%d",&name[i].l2);
        name[i].p2=l+1;
        for(int j=1;j<=name[i].l2;j++)
        {
            l++;
            scanf("%d",&v[l]);
        }
    }

    node *root=new node;
    for(int i=1;i<=m;i++)
    {
        int nowl;
        scanf("%d",&nowl);
        for(int j=0;j<nowl;j++)
            scanf("%d",&nowname[j]);
        nowname[nowl]=-1;
        insert(root,nowname,i);
    }

    build_AC(root);

    for(int i=1;i<=n;i++) query(root,i);
    for(int i=1;i<=m;i++) printf("%d\n",ask[i]);
    for(int i=1;i<=n;i++)
    {
        printf("%d",answer[i]);
        if(i==n) printf("\n");
        else printf(" ");
    }

    return 0;
}