ac自动机(求每个单词在单词所组成的论文中出现的次数)

思路:其实就是求每个单词的前缀中有多少后缀出现过某个单词,如果正向通过单词来枚举每个后缀,复杂度过高,那么我们反着枚举,然后惊奇的发现,这不就是next数组吗!那么就是把原题转化成了一个dag图,只需要把某个后继点的值累加到每个前缀就可以了,最后在插入的时候记录下每个单词的的位置,由于是dag图,我们将bfs的序列反着遍历,把f数组的当前值累加到前驱即可,这样就等价于把他的值给了所有前驱节点。最后查询f[位置]即可。

#include<bits/stdc++.h>
using namespace std;
const int M=1000010;
char str[M];
int ne[M],tr[M][26],idx,cnt[M];
int f[M];
int id[210];
int q[M];
void insert(int x)
{
    int p=0;
    for(int i=0;str[i];i++)
    {
        int ii=str[i]-'a';
        if(!tr[p][ii])   tr[p][ii]=++idx;
        p=tr[p][ii];
        f[p]++;
    }
    id[x]=p;
}
void build()
{
    int hh=0,tt=-1;
    for(int i=0;i<26;i++)
    if(tr[0][i])
    q[++tt]=tr[0][i];
    while(hh<=tt)
    {
        int t=q[hh++];
        for(int i=0;i<26;i++)
        {
            int &p=tr[t][i];
            if(!p) p=tr[ne[t]][i];
            else
            {
                ne[p]=tr[ne[t]][i];
                q[++tt]=p;
            }
        }
    }
}
int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        scanf("%s",str);
        insert(i);
    }
   // exit(0);
    build();
    //return 0;
    for(int i=idx-1;i>=0;i--)
    {
        f[ne[q[i]]]+=f[q[i]];
    }
    for(int i=0;i<n;i++)
    cout<<f[id[i]]<<endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值