BZOJ 2754: [SCOI2012]喵星球上的点名 [AC自动机+map+暴力]

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

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 1902  Solved: 837
[Submit][Status][Discuss]

Description

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

现在你能帮助a180285统计每次点名的时候有多少喵星人答到，以及M次点名结束后每个喵星人答到多少次吗？  

Input

现在定义喵星球上的字符串给定方法：

先给出一个正整数L，表示字符串的长度，接下来L个整数表示字符串的每个字符。

输入的第一行是两个整数N和M。

接下来有N行，每行包含第i 个喵星人的姓和名两个串。姓和名都是标准的喵星球上的

字符串。

接下来有M行，每行包含一个喵星球上的字符串，表示老师点名的串。

Output

对于每个老师点名的串输出有多少个喵星人应该答到。

然后在最后一行输出每个喵星人被点到多少次。

HINT

【数据范围】

对于30%的数据，保证：

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

对于100%的数据，保证：

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

---

喵~ http://www.cnblogs.com/candy99/p/cat.html

---

首先想到两问建两个AC自动机，第一问是把喵名建自动机然后点名串统计fail子树，但是不对啊啊啊啊啊人家一个喵就算一次.....

所以就发现其他人用了个很暴力的方法，点名串建AC自动机然后喵在上面跑，每跑一个字符暴力统计更新fail指向的点名串......然后就过了

一些细节：

1.要用map  字符集太大(离散化并没有用) 这样建AC自动机的时候就不能用Trie图优化了用普通做法

2.一个喵统计点名串需要用个vis[]，从黄学长哪里学到了黑科技用个vector记录vis了那些，不用memset直接把vector里的还原行了

3.可能有重复点名串！！！！！

4.姓和名用个特殊数字连起来行了

980ms

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
const int N=1e5+,INF=1e9;
inline int read(){
    char c=getchar();int x=,f=;
    while(c<''||c>''){if(c=='-')f=-; c=getchar();}
    while(c>=''&&c<=''){x=x*+c-''; c=getchar();}
    return x*f;
}

int n,m,k;
vector<int> a[N];
struct node{
    map<int,int> ch;
    int fail,val;
    vector<int> s;
}t[N];
int sz;
void ins(int id){
    int u=,n=read();
    for(int i=;i<=n;i++){
        int c=read();
        if(!t[u].ch[c]) t[u].ch[c]=++sz;
        u=t[u].ch[c];
    }
    t[u].s.push_back(id);
}
int q[N],head,tail;
map<int,int>::iterator it,it2;
void getAC(){
    head=tail=;
    for(it=t[].ch.begin();it!=t[].ch.end();it++)
        q[tail++]=it->second;
    while(head!=tail){
        int u=q[head++];
        for(it=t[u].ch.begin();it!=t[u].ch.end();it++){
            int v=it->second,c=it->first;
            int now=t[u].fail;
            while(now&&!t[now].ch.count(c)) now=t[now].fail;
            if(t[now].ch.count(c)) t[v].fail=t[now].ch[c];
            else t[v].fail=now;//
            q[tail++]=v;
        }
    }
}

int a1[N],a2[N];
bool vis[N];
vector<int> V;
void update(int u,int id){
    for(;u;u=t[u].fail){
        if(vis[u]) break;
        vis[u]=;
        V.push_back(u);
        for(int i=;i<t[u].s.size();i++)
            a1[t[u].s[i]]++,a2[id]++;
    }
}
void runAC(int id){
    vector<int> &s=a[id];
    int u=,n=s.size();
    for(int i=;i<n;i++){
        while(u&&!t[u].ch.count(s[i])) u=t[u].fail;
        if(t[u].ch.count(s[i])) u=t[u].ch[s[i]];
        update(u,id);
    }
    for(int i=;i<V.size();i++) vis[V[i]]=;
    V.clear();
}
void solve(){
    getAC();
    for(int i=;i<=n;i++) runAC(i);
    for(int i=;i<=m;i++) printf("%d\n",a1[i]);
    for(int i=;i<=n;i++) {printf("%d",a2[i]);if(i!=n) putchar(' ');}
}
int main(){
    freopen("in","r",stdin);
    n=read();m=read();
    for(int i=;i<=n;i++){
        k=read();while(k--) a[i].push_back(read());
        a[i].push_back(-);
        k=read();while(k--) a[i].push_back(read());
    }
    for(int i=;i<=m;i++) ins(i);
    solve();
}