【HDOJ 5384】Danganronpa

【HDOJ 5384】Danganronpa

AC自动机。。。当时感觉用字典树 标神也往自动机想来着。。手太生加上时间紧迫也没敲……回来一看题解什么AB同时建自动机。。。顿时愣了 什么叫同时建= =问了问财神说普通自动机。。。B串单建 立马疯了……这不就是模板题么。。。 B串建自动机 A串枚举查询 写完兴冲冲1T……立马想法优化 建fail时压缩一下 查询时直接累计 不再循环找fail 171ms。。。第二个自动机的题。。距上次蛮久了 这次一复习 感觉印象差不多有了

代码(模板)如下:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <queue>

using namespace std;

typedef struct Node
{
    int ch[26],cnt,fail;
}Node;

Node tr[2600011];
char a[100001][10002];
int tp;

int SetNode()
{
    memset(tr[tp].ch,-1,sizeof(tr[tp].ch));
    tr[tp].cnt = 0;
    return tp++;
}

void Add(int site,char *str)
{
    int i,k;
    for(i = 0; str[i]; ++i)
    {
        k = tr[site].ch[str[i] - 'a'];
        if(k == -1)  tr[site].ch[str[i] - 'a'] = k = SetNode();
        site = k;
    }
    tr[site].cnt++;
}

void Build_AC(int site)
{
    queue <int> q;
    q.push(site);
    int i,tmp;
    while(!q.empty())
    {
        site = q.front();
        q.pop();
        for(i = 0; i < 26; ++i)
        {
            if(tr[site].ch[i] == -1) continue;
            if(!site) tr[tr[site].ch[i]].fail = 0;
            else
            {
                tmp = tr[site].fail;
                while(tmp && tr[tmp].ch[i] == -1) tmp = tr[tmp].fail;
                tr[tr[site].ch[i]].fail = (tr[tmp].ch[i] == -1)? tmp: tr[tmp].ch[i];
                if(tr[tmp].ch[i] != -1)  ///压缩计数器
                    tr[tr[site].ch[i]].cnt += tr[tr[tmp].ch[i]].cnt;
            }
            q.push(tr[site].ch[i]);
        }
    }
}

int Search(int site,char *str)
{
    int i,k,ans = 0;
    for(i = 0; str[i]; ++i)
    {
        k = str[i] - 'a';
        while(site && tr[site].ch[k] == -1) site = tr[site].fail;
        if(tr[site].ch[k] != -1) site = tr[site].ch[k];
        ans += tr[site].cnt;
    }
    return ans;
}

int main()
{
    int t,na,nb,i;
    char str[100005];
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d",&na,&nb);
        tp = 0;
        SetNode();
        for(i = 0; i < na; ++i)
        {
            scanf("%s",a[i]);
        }
        while(nb--)
        {
            scanf("%s",str);
            Add(0,str);
        }
        Build_AC(0);
        for(i = 0; i < na; ++i)
        {
            printf("%d\n",Search(0,a[i]));
        }
    }
    return 0;
}