HDU 2896 AC自动机

最新推荐文章于 2020-07-27 19:43:52 发布

sevenster

最新推荐文章于 2020-07-27 19:43:52 发布

阅读量474

点赞数

CC 4.0 BY-SA版权

文章标签： insert query build web

本文链接：https://blog.youkuaiyun.com/bysen32/article/details/7390668

ACM[字符串处理] 专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一个基于AC自动机的应用案例，旨在通过构建自动机来快速查找并统计一系列病毒字符串在主串中的出现情况。文章提供了完整的C++实现代码，包括自动机的构建、插入病毒序列及查询过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

第二道AC自动机。和前面那道没有什么不同，都是基础的ＡＣ自动机模板。
题目大意：

有一些病毒序列，在主串中找到所有的病毒种类即可。

求出每个网站携带的病毒种类。

最后输出有多少个病毒网站。

还好输出不坑爹！

#include<iostream>
#include<cstdio>
#include<string.h>
#define MAX 128
using namespace std;

struct node
{
 	   int id,fail,next[MAX];
 	   void init(){
	   		memset( next,0,sizeof(next) );
	   		fail=-1,id=0;
   	   }
}Tire[111111];

int root,tot;
int queue[111111];
char str[11111];
int N,M;

void init(){
 	 root=tot=0;
 	 Tire[root].init();
}

void insert( int id,char *s )
{
 	 int i=0,k;
 	 int p=root;
 	 while( s[i] )
 	 {
	  		k=s[i++];
	  		if( !Tire[p].next[k] )
	  		{
			 	Tire[++tot].init();
			 	Tire[p].next[k]=tot;
		 	}
		 	p=Tire[p].next[k];
	 }
	 Tire[p].id=id;
}

void build_ac_automation()
{
 	 int head,tail;
 	 head=tail=0;
 	 queue[tail++]=root;
 	 while( head<tail )
 	 {
	  		int cur=queue[head++];
	  		for( int i=0;i<MAX;i++ )
	  		{
			 	 if( Tire[cur].next[i] )
			 	 {
				  	 int son=Tire[cur].next[i];
				  	 int p=Tire[cur].fail;
				  	 if( cur==root )
				  	 	 Tire[son].fail=root;
				  	 else
				  	 	 Tire[son].fail=Tire[p].next[i];
				  	 queue[tail++]=son;
			   	 }
			   	 else
			   	 {
				  	 int p=Tire[cur].fail;
				  	 if( cur==root )
				  	 	 Tire[cur].next[i]=root;
				  	 else
				  	 	 Tire[cur].next[i]=Tire[p].next[i];
			   	 }
 	 	 	}		
	 }
}

int query( char *s,int web )
{
 	int i=0,k;
 	int p=root,vis[555];
 	bool found=false;
 	memset( vis,0,sizeof(vis) );
 	while( s[i] )
 	{
	 	   k=s[i++];
	 	   while( !Tire[p].next[k]&&p!=root )
	 	   		  p=Tire[p].fail;
	 	   p=Tire[p].next[k];
	 	   if(p==-1)p=0;
	 	   int temp=p;
	 	   while( temp!=root )
	 	   {
		   		  if( Tire[temp].id>0 )
		   		  	  found=true;
		   		  vis[Tire[temp].id]=1;
		   		  temp=Tire[temp].fail;
  		   }
  	}
  	int cnt=0;
  	if( found )
  	{
  		printf( "web %d:",web );
  		for( int i=1;i<=N;i++ )
  		 if( vis[i] )
  		 	 printf( " %d",i ),cnt++;
  		 printf( "\n" );
	}
  	return cnt;
}



int main()
{
 	while( scanf( "%d\n",&N )!=EOF )
 	{
	 	   init();
	 	   for( int i=1;i<=N;i++ )
	 	   {
	 	   		gets(str);
				insert(i,str);
		   }
		   build_ac_automation();
	 	   scanf( "%d\n",&M );
	 	   int total=0;
	 	   for( int i=1;i<=M;i++ )
	 	   {
		   		gets(str);
		   		if( query(str,i) )
		   			total++;
	   	   }
	   	   printf( "total: %d\n",total );
  	}
 	return 0;
}