SCOI2012喵星球上的点名

http://codevs.cn/problem/2403/

2012年省队选拔赛四川

 时间限制: 2 s

 空间限制: 128000 KB

 

题目描述 Description

a180285幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非常有趣。 
假设课堂上有 N 个喵星人，每个喵星人的名字由姓和名构成。喵星球上的老师会选择M 个串来点名，每次读出一个串的时候，如果这个串是一个喵星人的姓或名的子串，那么这个喵星人就必须答到。 
然而，由于喵星人的字码过于古怪，以至于不能用 ASCII 码来表示。为了方便描述，a180285决定用数串来表示喵星人的名字。 
现在你能帮助 a180285 统计每次点名的时候有多少喵星人答到，以及 M 次点名结束后每个喵星人答到多少次吗？

 

输入描述 Input Description

现在定义喵星球上的字符串给定方法： 
先给出一个正整数 L，表示字符串的长度，接下来L个整数表示字符串的每个字符。 
输入的第一行是两个整数N和M。 
接下来有N行，每行包含第 i个喵星人的姓和名两个串。姓和名都是标准的喵星球上的字符串。 
接下来有M行，每行包含一个喵星球上的字符串，表示老师点名的串。

 

输出描述 Output Description

对于每个老师点名的串输出有多少个喵星人应该答到。 
然后在最后一行输出每个喵星人被点到多少次。

 

样例输入 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

 

数据范围及提示 Data Size & Hint

对于30%的数据，保证： 
1 <= M, N <= 1000
喵星人的名字总长不超过 4000，点名串的总长不超过 2000， 
对于100%的数据，保证： 
1<=N<=20000
1<=M<=50000
喵星人的名字总长和点名串的总长分别不超过
100000 保证喵星人的字符串中作为字符存在的数不超过 10000

 

AC自动机

学了几招：

1、用map搞trie

2、bool数组初始化，数组很大，需清空的数目不多，用vector记下来，枚举vector里的元素，清零

（貌似CDQ分治、整体二分就是这样搞的）

#include<iostream> 
#include<cstdio>
#include<queue>
#include<map>
#include<vector>
#define N 100010
using namespace std;
map<int,int>trie[N];
vector<int>a[20001],st[N],USE,MARK;
int f[N];
int len,tot=1,n,m;
queue<int>q;
bool mark[N],use[50001];
int call[50001],miao[20001];
struct aa
{
    void insert(int i)
    {
        int x,root=1;
        while(len--)
        {
            scanf("%d",&x);
            if(!trie[root][x]) trie[root][x]=++tot;
            root=trie[root][x];
        }
        st[root].push_back(i);
    }
    void getfail()
    {
        q.push(1);int j;
        while(!q.empty()) 
        {
            int now=q.front();q.pop();
            map<int,int>::iterator it;
            for(it=trie[now].begin();it!=trie[now].end();it++)
            {
                int w=it->second,e=it->first;
                q.push(w);
                j=f[now];
                while(!trie[j][e])   j=f[j];
                f[w]=trie[j][e];
            } 
        }
    }
    void get(int now,int who)
    {
        for(int i=now;i;i=f[i])
        {
            if(mark[i]) return;
            else
            {
                mark[i]=true;
                MARK.push_back(i);
                for(int j=0;j<st[i].size();j++)
                {
                    if(!use[st[i][j]])
                    {
                        use[st[i][j]]=true;
                        USE.push_back(st[i][j]);
                        miao[who]++;
                        call[st[i][j]]++;
                    }
                }
            }
        }
    }
    void solve(int k)
    {
        int id,root=1;
        for(int i=0;i<a[k].size();i++)
        {
            id=a[k][i];
            while(!trie[root][id]) root=f[root];
            root=trie[root][id];
            get(root,k);
        }
        for(int i=0;i<MARK.size();i++) mark[MARK[i]]=false;
        for(int i=0;i<USE.size();i++) use[USE[i]]=false;
        MARK.clear();USE.clear();
    }
}tree;
int main()
{
    /*freopen("name.in","r",stdin);
    freopen("name.out","w",stdout);*/
    int x;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&len);
        while(len--) 
        {
            scanf("%d",&x);
            a[i].push_back(x);
        }
        a[i].push_back(-1);
        scanf("%d",&len);
        while(len--)
        {
            scanf("%d",&x);
            a[i].push_back(x);
        }
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&len);
        tree.insert(i);
    }
    for(int i=-1;i<=10000;i++) trie[0][i]=1;
    tree.getfail();
    for(int i=1;i<=n;i++)    
    tree.solve(i);
    for(int i=1;i<=m;i++) printf("%d\n",call[i]);
    for(int i=1;i<n;i++) printf("%d ",miao[i]);printf("%d",miao[n]);
    return 0;
}