HDOJ1277 全文检索(AC自动机)

全文检索

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)

Total Submission(s): 1403 Accepted Submission(s): 458



[align=left]Problem Description[/align]
我们大家经常用google检索信息，但是检索信息的程序是很困难编写的；现在请你编写一个简单的全文检索程序。

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

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

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

[align=left]Sample Input[/align]

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

[align=left]Sample Output[/align]

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

[align=left]Author[/align]
Cai Minglun

[align=left]Source[/align]
杭电ACM集训队训练赛（VI）

[align=left]Recommend[/align]
lcy | We have carefully selected several similar problems for you: 1275 3068 2243 1274 1271

大一测试监考无聊水一把，题目不能一直老做一类啊，要不脑子都不灵活了，本来可以用字典树做的，而我却写了个AC自动机，写得长还费时间


ac代码

#include<stdio.h>
#include<string.h>
char str[60060],key[60],s[60060],k1[10],k2[10],k3[10];
int vis[10010],ans[10010],k,flag;
int head,tail;
struct node
{
node *fail;
node *next[10];
int cnt;
node()
{
fail=NULL;
cnt=0;
for(int i=0;i<10;i++)
next[i]=NULL;
}
}*q[602000];
node *root;
void insert(char *s,int id)
{
int temp,len,i;
node *p=root;
len=strlen(s);
for(i=0;i<len;i++)
{
temp=s[i]-'0';
if(p->next[temp]==NULL)
p->next[temp]=new node();
p=p->next[temp];
}
p->cnt=id;
}
void build_ac()
{
root->fail=NULL;
q[tail++]=root;
while(head!=tail)
{
node *p=q[head++];
node *temp=NULL;
for(int i=0;i<10;i++)
{
if(p->next[i]!=NULL)
{
if(p==root)
{
p->next[i]->fail=root;
}
else
{
temp=p->fail;
while(temp!=NULL)
{
if(temp->next[i]!=NULL)
{
p->next[i]->fail=temp->next[i];
break;
}
temp=temp->fail;
}
if(temp==NULL)
{
p->next[i]->fail=root;
}
}
q[tail++]=p->next[i];
}
}
}
}
void query()
{
int len=strlen(str);
node *p=root,*temp;
for(int i=0;i<len;i++)
{
int x=str[i]-'0';
while(p->next[x]==NULL&&p!=root)
{
p=p->fail;
}
p=p->next[x];
if(p==NULL)
{
p=root;
}
temp=p;
while(temp!=root&&temp->cnt!=0)
{
if(!vis[temp->cnt])
{
vis[temp->cnt]=1;
ans[k++]=temp->cnt;
flag=1;
}
temp=temp->fail;
}
}
}
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
int i;
head=tail=0;
root=new node();
k=0;
flag=0;
memset(vis,0,sizeof(vis));
for(i=0;i<n;i++)
{
scanf("%s",s);
strcat(str,s);
}
//printf("%s",str);
for(i=1;i<=m;i++)
{
int x;
scanf("[Key No. %d] ",&x);
scanf("%s%s%s%s",k1,k2,k3,key);
insert(key,i);
}
build_ac();
query();
if(!flag)
{
printf("No key can be found !\n");
}
else
{
printf("Found key:");
for(i=0;i<k;i++)
{
printf(" [Key No. %d]",ans[i]);
}
printf("\n");
}
}
}