BZOJ 1212 HNOI2004 L语言 AC自动机(Trie树)+动态规划

来源：互联网发布：淘宝的淘气值是干嘛的编辑：程序博客网时间：2024/05/22 00:16

题目大意：给定一个单词表和m个字符串问每个字符串的最长的前缀，满足这个前缀可以拆分成一些字符串使这些字符串都在单词表中出现过

再也不敢看错数据范围了……一道明明用Trie树能解决的问题居然被我写了AC自动机……

将单词表中的单词全都插入AC自动机每个单词所在的节点记录这个单词的长度

然后对于每个字符串用f[i]表示长度为i的前缀是否能拆分成单词表中的单词跑AC自动机

对于每个匹配的节点从这个节点开始到根的fail路径上的所有len f[i]|=f[i-len]

找到最大的为1的f[i]即是答案

由于单词长度最大为10 所以直接用Trie树暴力即可

#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>#define M 1050000using namespace std;struct Trie{int len;Trie *fail,*son[26];void* operator new (size_t size);}*root,*mempool,*C;int n,m;char s[M];void* Trie :: operator new (size_t size){if(C==mempool){C=new Trie[1<<15];mempool=C+(1<<15);memset(C,0,sizeof(Trie)*(1<<15) );}return C++;}void Insert(Trie*&p,char *pos,int dpt){if(!p) p=new Trie;if(!*pos){p->len=dpt;return ;}Insert(p->son[(*pos)-'a'],pos+1,dpt+1);}void BFS(){static Trie* q[1<<16];static unsigned short r,h;int i;Trie *temp;for(i=0;i<26;i++)if(temp=root->son[i]){temp->fail=root;q[++r]=temp;}while(r!=h){Trie *p=q[++h];for(i=0;i<26;i++)if(p->son[i]){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];}}}int Aho_Corasick_Automaton(){int i,re=0;Trie *p=root,*temp;static bool f[M];f[0]=1;for(i=1;s[i];i++){f[i]=0;while( p!=root && !p->son[s[i]-'a'] )p=p->fail;if(p->son[s[i]-'a']){p=p->son[s[i]-'a'];for(temp=p;temp!=root;temp=temp->fail)if(temp->len){f[i]|=f[i-temp->len];if(f[i]) break;}}if(f[i]) re=i;}return re;}int main(){int i;cin>>n>>m;for(i=1;i<=n;i++)scanf("%s",s+1),Insert(root,s+1,0);BFS();for(i=1;i<=m;i++){scanf("%s",s+1);cout<<Aho_Corasick_Automaton()<<endl;}}

0 0