HDU 爱爱爱爱 Keywords Search (Accepted自动机)

如此有爱的题目编号，如此令人心动的算法名字，怎能不让人反复回味。所以在写了两遍代码之后，我还要再~~~写一篇解题报告，当然解题报告里面的代码我是懒得重新再~~~~写一次的。过个几天复习时再写一遍吧。

 

总体流程呢，就是：

先建一个字典树的根节点，然后把字典里的单词一个个插进去，让这株小小的幼苗茁壮成长直到长成一棵大小不会超过(单词数*单词最大长度)的参天大树！！（准确来说，上限是for(i=1;i<=单词最大长度;i++)sum+=min(26^i,单词数)）

 

接着呢，类似KMP算法，要找出每个枝叶的失败指针。为了遍历所有枝叶，要从根开始广搜每个枝叶，然后对每个节点找他的失败指针。这里我要说一下，网上教程都把这叫做失败指针，这多难听啊，好好的AC自动机里面居然要用到“失败”！这个叫法真的是失败中的失败！！所以呢我要换一种更通俗易懂又贴切好听的叫法，叫做“啊，我那失散多年的兄弟”。那怎么找这个兄弟呢？先找到一个跟爸爸长得一模一样的叔叔（其实他就是你爸爸失散多年的兄弟哦），然后看这个叔叔有没有个跟自己长得一模一样的儿子，如果没有，就找下一个叔叔。那如果有呢？“啊，你就是我那失散多年的兄弟”。

 

现在找到兄弟了，爸爸也找到兄弟了，叔叔也找到兄弟了，爷爷也找到兄弟了，一大家子终于团聚了，他们要齐心协力对付最后共同的敌人——主串（主串泪目：同是字符串，匹配何太急）（各模式串：匹配一下而已嘛，又不会怀孕）。接下来大体跟KMP的匹配类似，两个指针，一个指主串开头，一个指字典树根节点（这里是解题报告，我说话要正常，就这样，正常），然后开始比较，比较失败就跳到相应的失败指针那里。但有点不一样，就是说如果比较时没有失败，不是加一次就够了。举个例子：模式串是acd、c，那字典树就是两条简单的直链acd和c，主串是acd。（很好，保持住）那么字典树里acd依次比较下来都对的，最后发现d上的值是1（表示是acd这个单词有1个），加个1就完了，输出了1。可实际上主串acd里把两个模式串都包括了，答案是2。那怎么办呢？比较没失败的时候，把字典树上这个点的失散多年的兄弟，失散多年的兄弟的失散多年的兄弟，失散多年的兄弟的失散多年的兄弟的失散多年的兄弟……（我要正常，放松，正常说话，冷静，正常点(拍脸)，正常……）都看一遍，把上面的值加上去，并且把那值置0，以防重复计算。

 

输出总值就好了。

 

学的时候看着也累，理解也累，写着也累，写完后觉得蛮顺了，却看到还有跟DP结合什么结合的，头巨大。

 

追加：刚看到了一个很重要的剪枝。把值加上去后不是置0，还是设为-1，表示已经加过了，下次加值时访问到这点直接跳出，因为接下去失败指针所能到的地方必也都加过了。600+瞬间降到200+

 

 

#include<iostream> #include<queue> using namespace std; #define Chr 26 struct node{ int next[Chr],v,fail; }; node trie[500000]; char str[1000001]; int trien; void set_node(int x){//节点初始化 int i; for(i=0;i<Chr;i++) trie[x].next[i]=0; trie[x].v=0; } void insert(char *x){//这里呢就是把单词插进来 int loc=0,i; for(i=0;x[i];i++){//遍历字母 if(!trie[loc].next[x[i]-'a']){//这个字母的点不存在就建一个 set_node(++trien); trie[loc].next[x[i]-'a']=trien; } loc=trie[loc].next[x[i]-'a'];//指到这个字母的位置，准备搞下一个字母 } trie[loc].v++;//最后个字母所在点的值加1 } void go(int pre,int c,int xid){//前个节点的id，这个节点的字母编号('a'=0)，这个节点的id if(!pre){//前一个是根节点，那失败指针也是根节点，即0 trie[xid].fail=0; return; } int loc=trie[pre].fail;//跟我爸爸一样的叔叔 while(loc&&!trie[loc].next[c])loc=trie[loc].fail; trie[xid].fail=trie[loc].next[c];//如果那叔叔的跟我一样的儿子存在，那我的失散多年的兄弟就是他(我草，我还真这么用了) //如果不存在也跳出来了，那就应该指向根节点。可由于不存在，所以trie[loc].next[c]肯定是0，直接指过去也可以，这样就不用额外判断loc是不是0了 } queue<int> id; void get_fail(){ id.push(0); int temp,i; while(id.size()){//广搜 temp=id.front();id.pop(); for(i=0;i<Chr;i++){ if(trie[temp].next[i]){ go(temp,i,trie[temp].next[i]);//给这个点找失败指针 id.push(trie[temp].next[i]); } } } } int ans(){ int ps=0,pd=0,count=0,temp; while(1){ while(str[ps]&&!trie[pd].next[str[ps]-'a']){ if(pd)pd=trie[pd].fail; else ps++; } if(!str[ps])break; pd=trie[pd].next[str[ps]-'a']; temp=pd; //-------------------- while(temp){ count+=trie[temp].v; trie[temp].v=0; //这里一块就是找失散多年的兄弟的失散多年的兄弟。。。。 temp=trie[temp].fail;//其实我也不想这么说的，可是不知道用指针怎么表述，失败指针所指的地方的失败指针？好麻烦 } //--------------------- ps++; } return count; } int main(){ int t,n; char dic[51]; scanf("%d",&t); while(t--){ scanf("%d",&n); set_node(0);trien=0;//建根节点，id为0 while(n--){ scanf("%s",dic); insert(dic); } get_fail(); scanf("%s",str); printf("%d/n",ans()); } return 0; }