码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ3172: [Tjoi2013]单词

时间:2017-10-05 12:17:35      阅读:130      评论:0      收藏:0      [点我收藏+]

标签:队列   logs   更新   center   ret   blog   main   get   amp   

【传送门:BZOJ3172


简要题意:

  给出n个单词,你可以理解为将这些单词变成一个个段落,然后求出每个单词在所有段落中出现的次数


题解:

  刚开始不是很懂题目,结果发现将所有单词看成一篇文章,每个单词看成一个段落就懂了

  由于某种unbelievable的原因,我刚好做了AC自动机的专题训练,看到这道题就秒想AC自动机

  将每个单词放进AC自动机里,每个点的s表示有多少个单词经过,然后在构建失败指针的时候,通过队列来更新s值,怎么更新呢,假设有一个i点,它的失败指针指向j,设sj为从根到j点所构成的字符串,si为从根到i点所构成的字符串,那么我们可以知道sj为si的后缀,也说明了si中有sj的出现,那么就将j点的s值加上i点的s值,达到更新且不重复的目的来求出答案


参考代码:

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
using namespace std;
struct node
{
    int s,c[27],fail;
    node()
    {
        s=fail=0;
        memset(c,-1,sizeof(c));
    }
}t[1100000];
int tot,ans,n;
char a[1100000];
int ed[210];
void bt(int k,int root)
{
    int x=root,len=strlen(a+1);
    for(int i=1;i<=len;i++)
    {
        int y=a[i]-a+1;
        if(t[x].c[y]==-1)
        {
            t[x].c[y]=++tot;
        }
        x=t[x].c[y];
        t[x].s++;
    }
    ed[k]=x;
}
int list[1100000];
void bfs()
{
    int x;
    int head=1,tail=1;
    list[1]=0;
    while(head<=tail)
    {
        x=list[head];
        for(int i=1;i<=26;i++)
        {
            int son=t[x].c[i];
            if(son==-1)continue;
            if(x==0) t[son].fail=0;
            else
            {
                int j=t[x].fail;
                while(j!=0&&t[j].c[i]==-1) j=t[j].fail;
                t[son].fail=max(t[j].c[i],0);
                int x=t[son].fail,y=son;
            }
            list[++tail]=son;
        }
        head++;
    }
    for(int i=tail;i>=1;i--) t[t[list[i]].fail].s+=t[list[i]].s;
}
int main()
{
    ans=tot=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",a+1);
        bt(i,0);
    }
    bfs();
    for(int i=1;i<=n;i++)
    {
        printf("%d\n",t[ed[i]].s);
    }
    return 0;
}

 

 

 

BZOJ3172: [Tjoi2013]单词

标签:队列   logs   更新   center   ret   blog   main   get   amp   

原文地址:http://www.cnblogs.com/Never-mind/p/7628734.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!