HDU2222 Keywords Search 题解&代码

题意：多组数据，每组数据有n个字符串作为字典，然后给出另外一个字符串，询问这个新的字符串中有几个字典中的单词。
多个匹配串对单字符串匹配，AC自动机是标准解法，算是测试模板了【笑
然而RE了一发WA了一发…没看清数据范围

对于字典中的字符串建立trie树和fail指针，然后对待匹配串匹配即可
有一些奇怪的小细节譬如字典中可能有多个相同字符串，以及判重问题…恩还是很好解决的嘛

#include <iostream>#include <cstdio>#include <cstring>using namespace std;const int maxn = 500005;const int maxm = 26;int T, n, tot, ans, temp;int ch[maxn][maxm], fail[maxn], flag[maxn], q[maxn], vis[maxn];char s[maxn*2];int cal(char c){    return c-'a';}void newnode(int x,int temp){    if(temp!=-1) ch[x][temp] = ++tot;    for(int i = 0; i < maxm; i++)        ch[tot][i]=0;    flag[tot]=0;    fail[tot]=0;}void addtrie(char s[]){    int x = 0, p = 0;    int len = strlen(s);    while( p < len )    {        temp=cal(s[p]);        if( !ch[x][temp] ) newnode(x, temp);            x = ch[x][temp];            p++;        }        flag[x]++;}void getfail(void){    int h = 0, t = 0;        for(int i = 0; i < maxm; i++)            if( ch[0][i] ) q[t++] = ch[0][i];        while( h < t )        {            temp=q[h++];            for(int i = 0; i < maxm; i++)                if( !ch[temp][i] ) ch[temp][i] = ch[fail[temp]][i];                else                {                    q[t++] = ch[temp][i];                    fail[ch[temp][i]] = ch[fail[temp]][i];                }        }}int query(void){    int len = strlen(s), x = 0, p = 0;    temp = 0;    for(int i = 0; i < len; i++)    {        temp = s[i]-'a';        x = ch[x][temp];        p = x;        while( p && !vis[p] )        {            ans += flag[p];            flag[p] = 0;            vis[p] = 1;            p = fail[p];        }    }    return ans;}int main(void){    scanf("%d",&T);    while(T--)    {        tot = 0;        ans = 0;        newnode(0,-1);        memset(vis,0,sizeof(vis));        scanf("%d",&n);        for(int i = 0; i < n; i++)        {            scanf("%s",s);            addtrie(s);        }        getfail();        scanf("%s",s);        printf("%d\n",query());    }    return 0;}