Keywords Search(查找关键字)

hdoj 2222

题目大意：给出一些字符串 ，再给出一段文字，问文字中出现多少个单词

解决：AC自动机

 

#include <iostream>
#include <cstring>
#include <queue>
#include <cstdio>
using namespace std;
#define s  scanf
int top;
struct node
{
    int cnt;
    int fail;
    int next[26];
};
node tree[400000];
char mai[1000005];
void init()
{
    top=1;
    memset(tree[0].next,0,sizeof(tree[0].next));
   //刚开始的时候初始化，是在node的构造函数中实现的，但是由于是多组测试数据，所以
  //错误了好几次这次，就应该在初始化，和需要新的结点的时候在初始化
    tree[top].cnt=0;
}
void insert_trie(char str[])
{
    int id,i=0;
    while(*str)
    {
        id=*str-'a';
        if(!tree[i].next[id])
        {
            tree[i].next[id]=top;
            memset(tree[top].next,0,sizeof(tree[top].next));
            tree[top].cnt=0;
            top++;
        } 
        i=tree[i].next[id];
        str++;
    }
    tree[i].cnt++;
}
void build_fail()
{
    tree[0].fail=-1;
    queue<int> q;
    q.push(0);
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        for(int i=0;i<26;i++)
        {
            int tmp=tree[now].next[i];
            if(tmp)
            {
                int p;
                for(p=tree[now].fail;p!=-1;p=tree[p].fail)
                   if(tree[p].next[i])
                   {
                        tree[tmp].fail=tree[p].next[i];
                        break;
                   }
                   
                if(p==-1){tree[tmp].fail=0;}
                q.push(tmp);
            }
        }
    }
}
void query(char str[])
{
    int id,sum=0,p=0;
    while(*str)
    {
        id=*str-'a';
      //不匹配的情况，找到根节点，或者是找到匹配的地方
        while(tree[p].next[id]==0 && p)p=tree[p].fail;
        if(tree[p].next[id]==0)p=0;
        else p=tree[p].next[id];
    //匹配或者是根节点的情况，沿着fail一直找到跟，将所有的一该字母结束的串的情况 都   算上。
        int tmp=p;   
        while(tmp && tree[tmp].cnt)
        {
            sum+=tree[tmp].cnt;
            tree[tmp].cnt=0;
            tmp=tree[tmp].fail;
        }
        str++;
    }
    printf("%d\n",sum);
}

int main()
{
    int icase;
    s("%d",&icase);
    while(icase--)
    {
        init();
        int n;
        char str[60];
        s("%d",&n);
        while(n--)
        {
            s("%s",str);
            insert_trie(str);
        }
        build_fail();
        s("%s",mai);
        query(mai);
    }
 //   system("pause");
    return 0;
}