[BZOJ2754][SCOI2012]喵星球上的点名-AC自动机

喵星球上的点名

Description

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

HINT

【数据范围】

对于30%的数据,保证:

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

对于100%的数据,保证:

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


第二个喵星人咱不认识,但是

样例中第一个喵星人的名字怎么那么眼熟呢???


思路:

由于要求点名串是名字的子串,那么考虑对点名串建出AC自动机。

在插入点名串时,建立点名串与插入时遇到的最后一个AC自动机节点的映射,同时给这个节点作为点名串结尾字符的次数++。

然后考虑直接把所有名和姓在AC自动机上跑,每当走过一个节点,就沿着当前节点的fail指针跳到根节点。
沿路被访问的节点若曾经被作为点名串的结尾字符,并且还未在跑这个喵星人的名字的过程中访问到,那么这个喵星人的点名次数加上当前节点作为结尾字符的次数,同时这个节点被访问的次数++,且标记为已经被当前节点访问过。

喵星人的点名次数可直接输出,而对于每次点名的答到数可以通过查看每个点名串映射的AC自动机节点被访问的次数得到。

字符集过大所以要用map存,另外M貌似开50000多一点是会炸的。

那么这就做完了~

因为一时兴起(脑抽)就采用了sys这个名字命名namespace……

#include<bits/stdc++.h>

using namespace std;

inline int read()
{
    int x=0;char ch=getchar();
    while(ch<'0' || '9'<ch)ch=getchar();
    while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
    return x;
}

typedef map<int,int>::iterator mint;
const int N=20009;
const int M=90009;
const int L=1000009;

int n,m,ans[N];
vector<int> name[N][2],s;

namespace sys
{
    map<int,int> ch[L];
    int fail[L],ed[L],vis[M],tot;
    int rid[M],ans[M],q[L],l,r;

    inline void insert(const vector<int> &s,int id)
    {
        int now=0;
        for(int i=0,e=s.size();i<e;i++)
        {
            if(!ch[now].count(s[i]))
                ch[now][s[i]]=++tot;
            now=ch[now][s[i]];
        }
        ed[now]++;
        rid[id]=now;
    }

    inline void init()
    {
        l=q[r=1]=0;
        fail[0]=0;
        while(l<r)
        {
            int u=q[++l];
            for(mint i=ch[u].begin();i!=ch[u].end();i++)
            {
                int pos=fail[u];
                while(pos && !ch[pos].count((*i).first))
                    pos=fail[pos];
                if(u && ch[pos].count((*i).first))
                    fail[(*i).second]=ch[pos][(*i).first];
                else
                    fail[(*i).second]=0;
                q[++r]=(*i).second;
            }
        }
    }

    inline int count(const vector<int> &s,int id)
    {
        int now=0,ret=0;
        for(int i=0,e=s.size();i<e;i++)
        {
            while(now && !ch[now].count(s[i]))
                now=fail[now];
            if(ch[now].count(s[i]))
                now=ch[now][s[i]];
            for(int j=now;j;j=fail[j])
                if(ed[j] && vis[j]!=id)
                {
                    ret+=ed[j];
                    vis[j]=id;
                    ans[j]++;
                }
        }
        return ret;
    }

    inline int query(int id)
    {
        return ans[rid[id]];
    }
}

int main()
{
    n=read();m=read();
    for(int i=1,l;i<=n;i++)
        for(int k=0;k<=1;k++)
        {
            l=read();
            for(int j=1;j<=l;j++)
                name[i][k].push_back(read());
        }

    for(int i=1,l;i<=m;i++)
    {
        l=read();
        s.clear();
        for(int j=1;j<=l;j++)
            s.push_back(read());
        sys::insert(s,i);
    }

    sys::init();

    for(int i=1;i<=n;i++)
        for(int j=0;j<=1;j++)
            ans[i]+=sys::count(name[i][j],i);

    for(int i=1;i<=m;i++)
        printf("%d\n",sys::query(i));
    for(int i=1;i<=n;i++)
        printf("%d%c",ans[i]," \n"[i==n]);
    puts("");
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值