首页 > 代码库 > BZOJ 3172 Tjoi2013 单词 fail树
BZOJ 3172 Tjoi2013 单词 fail树
题目大意及后缀数组做法见 http://blog.csdn.net/popoqqq/article/details/41042473
原来正解是fail树……难怪后缀数组被卡成这样
首先我们将给出的n个串构建AC自动机
朴素的做法是对于每个串将这个串每个节点沿着fail指针扫一遍,将路径上的所有点的cnt++
但是这样做会TLE
我们不妨反向思考 fail指针反向后是一棵树 沿着fail指针扫一遍就是沿着树边向根扫一遍
只在插入时将每个串的每个节点cnt++ 那么每个串终点所在fail树的子树中cnt的总和就是这个字符串的出现次数
于是每个点记录所在fail树的子树中cnt的数量 可以线性时间内出解
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define M 1001001 using namespace std; struct Trie{ Trie *son[26],*fail; int cnt; }*root,mempool[M],*C=mempool; int n; Trie *pos[210]; char s[M]; void Insert(Trie* &p,char *s,Trie* &pos) { if(!p) p=C++; p->cnt++; if(!*s) { pos=p; return ; } Insert(p->son[(*s)-'a'],s+1,pos); } void Build_Tree() { static Trie *q[M]; static int r,h; int i; for(i=0;i<26;i++) if(root->son[i]) root->son[i]->fail=root,q[++r]=root->son[i]; while(r!=h) { Trie *p=q[++h]; for(i=0;i<26;i++) if(p->son[i]) { Trie *temp=p->fail; while(temp!=root&&!temp->son[i]) temp=temp->fail; if(temp->son[i]) temp=temp->son[i]; p->son[i]->fail=temp; q[++r]=p->son[i]; } } for(i=r;i;i--) q[i]->fail->cnt+=q[i]->cnt; } int main() { int i; cin>>n; for(i=1;i<=n;i++) scanf("%s",s),Insert(root,s,pos[i]); Build_Tree(); for(i=1;i<=n;i++) printf("%d\n",pos[i]->cnt); return 0; }
BZOJ 3172 Tjoi2013 单词 fail树
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。