BZOJ 3172 [Tjoi2013] 单词 Fail树

题目大意:给出n个单词,输出每个单词在所有单词中出现的次数。

fail树真的是神奇…这里讲的还不错,可以看看(原来我写的AC自动机是Trie图)

Fail树的一个性质是,某个结点对应的字符串肯定是其子树结点对应的字符串的后缀。

这道题想找出每一个串在所有的串中匹配的次数,这时就可以利用fail树的这个性质,而且fail树也是建立在Trie上的,统计子树中所有结点个数,就代表其在所有串中的前缀中作为后缀出现的次数,即在所有串中的出现次数。在队列中越靠后的点就越靠近叶子,用队列倒着更新即可。

#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
struct Node {
    Node *ch[26],*nxt;
    int val;
    Node():nxt(0x0),val(0){memset(ch,0,sizeof ch);}
}*root=new Node(),*p[205];
char s[1000005];
Node* Insert() {
    int len=strlen(s);
    Node* o=root;
    for(int i=0;i<len;i++) {
        o->val++;
        int z=s[i]-'a';
        if(!o->ch[z]) o->ch[z]=new Node();
        o=o->ch[z];
    }
    o->val++;
    return o;
}
void getFail() {
    static Node* q[1000005];
    int l,r; l=r=0;
    Node* o=root;
    for(int i=0;i<26;i++)
        if(o->ch[i]) q[r++]=(o->ch[i]) , o->ch[i]->nxt=o;
        else o->ch[i]=o;
    while(l<r) {
        o=q[l++];
        for(int i=0;i<26;i++)
            if(o->ch[i]) q[r++]=(o->ch[i]) , o->ch[i]->nxt=o->nxt->ch[i];
            else o->ch[i]=o->nxt->ch[i];
    }
    for(int i=r-1;~i;i--)
        q[i]->nxt->val+=q[i]->val;
    return ;
}
int main() {
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%s",s) , p[i]=Insert();
    getFail();
    for(int i=1;i<=n;i++) printf("%d\n",p[i]->val);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值