【AC自动机】 hdu1277 全文检索

全文检索

http://acm.hdu.edu.cn/showproblem.php?pid=1277

Problem Description

我们大家经常用google检索信息，但是检索信息的程序是很困难编写的；现在请你编写一个简单的全文检索程序。
问题的描述是这样的：给定一个信息流文件，信息完全有数字组成，数字个数不超过60000个，但也不少于60个；再给定一个关键字集合，其中关键字个数不超过10000个，每个关键字的信息数字不超过60个，但也不少于5个；两个不同的关键字的前4个数字是不相同的；由于流文件太长，已经把它分成多行；请你编写一个程序检索出有那些关键字在文件中出现过。

 

Input

第一行是两个整数M，N；M表示数字信息的行数，N表示关键字的个数；接着是M行信息数字，然后是一个空行；再接着是N行关键字；每个关键字的形式是：[Key No. 1] 84336606737854833158。

 

Output

输出只有一行，如果检索到有关键字出现，则依次输出，但不能重复，中间有空格，形式如：Found key: [Key No. 9] [Key No. 5]；如果没找到，则输出形如：No key can be found !。

 

Sample Input

  
  

   
   20 10
646371829920732613433350295911348731863560763634906583816269
637943246892596447991938395877747771811648872332524287543417
420073458038799863383943942530626367011418831418830378814827
679789991249141417051280978492595526784382732523080941390128
848936060512743730770176538411912533308591624872304820548423
057714962038959390276719431970894771269272915078424294911604
285668850536322870175463184619212279227080486085232196545993
274120348544992476883699966392847818898765000210113407285843
826588950728649155284642040381621412034311030525211673826615
398392584951483398200573382259746978916038978673319211750951
759887080899375947416778162964542298155439321112519055818097
642777682095251801728347934613082147096788006630252328830397
651057159088107635467760822355648170303701893489665828841446
069075452303785944262412169703756833446978261465128188378490
310770144518810438159567647733036073099159346768788307780542
503526691711872185060586699672220882332373316019934540754940
773329948050821544112511169610221737386427076709247489217919
035158663949436676762790541915664544880091332011868983231199
331629190771638894322709719381139120258155869538381417179544
000361739177065479939154438487026200359760114591903421347697

[Key No. 1] 934134543994403697353070375063
[Key No. 2] 261985859328131064098820791211
[Key No. 3] 306654944587896551585198958148
[Key No. 4] 338705582224622197932744664740
[Key No. 5] 619212279227080486085232196545
[Key No. 6] 333721611669515948347341113196
[Key No. 7] 558413268297940936497001402385
[Key No. 8] 212078302886403292548019629313
[Key No. 9] 877747771811648872332524287543
[Key No. 10] 488616113330539801137218227609
  
  

 

Sample Output

  
  

   
   Found key: [Key No. 9] [Key No. 5]
  
  

题意：给一个长字符串和几个关键字，问哪些关键字在长字符串中出现过。

题解：AC自动机的模板题。

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
struct node
{
    node *next[10],*fail;
    int id;
    node()
    {
        memset(next,0,sizeof(next));
        fail=NULL;
        id=0;
    }
}*head;
vector<int> out;
char si[60005],temp[60005],pi[65];
bool flag;
void build(char *s,node *head,int id)
{
    int l=strlen(s),k;
    for(int i=0; i<l; ++i)
    {
        k=s[i]-'0';
        if(head->next[k]==NULL)
            head->next[k]=new node();
        head=head->next[k];
    }
    head->id=id;
}
void build_fail(node *head)
{
    node *now,*p;
    queue<node*> q;
    head->fail=NULL;
    q.push(head);
    for(; !q.empty();)
    {
        now=q.front();
        q.pop();
        for(int i=0; i<10; ++i)
            if(now->next[i])
            {
                p=now->fail;
                for(; p&&!p->next[i]; p=p->fail);
                now->next[i]->fail=p?p->next[i]:head;
                q.push(now->next[i]);
            }
    }
}
void ac_find(char *s,node *head)
{
    int len=strlen(s);
    node *p=head;
    for(int i=0;i<len;++i)
    {
        int k=s[i]-'0';
        for(;!p->next[k]&&p!=head;p=p->fail);
        p=p->next[k]==NULL?head:p->next[k];
        node *tmt=p;
        for(;tmt!=head;tmt=tmt->fail)
            if(tmt->id)
            {
               out.push_back(tmt->id);
               flag=false;
            }
    }
}
int main()
{
    int n,m,l;
    for(; ~scanf("%d%d",&m,&n);)
    {
        flag=true;
        head=new node();
        out.clear();
        for(int i=0;i<m;++i)
        {
            scanf("%s",temp);
            if(i==0)  strcpy(si,temp);
            else      strcat(si,temp);
        }
        for(int i=1; i<=n; ++i)
        {
            scanf("%s",pi);
            scanf("%s",pi);
            scanf("%d] %s",&l,pi);
            build(pi,head,i);
        }
        build_fail(head);
        ac_find(si,head);
        if(flag) puts("No key can be found !");
        else
        {
            printf("Found key:");
            for(int i=0;i<out.size();++i)
                printf(" [Key No. %d]",out[i]);
            printf("\n");
        }
    }
    return 0;
}