HDU - 2896 病毒侵袭(AC自动机)

题目大意：给出 n 个模式串，再给出 m 个匹配串，问有多少个模式串在匹配串中出现，需要分别对应上其编号

题目分析：对应编号问题我们可以直接开一个数组映射，也是比较经典的模板问题了，注意输出格式就好了，还有如果MLE的话，用指针的朋友可以用C++交一下试试，用数组的朋友不要滥用memset，剩下的就是模板了

代码：

#include<iostream>
#include<cstdio> 
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<unordered_map>
using namespace std;
 
typedef long long LL;
 
const int inf=0x3f3f3f3f;
 
const int N=1e5+100;
 
char s[N];
 
int fail[N],trie[N][130],cntword[N],rk[N],vis[N],cnt,tot;
 
void insert_word(int id)
{
	int len=strlen(s);
	int pos=0;
	for(int i=0;i<len;i++)
	{
		int to=s[i];
		if(!trie[pos][to])
			trie[pos][to]=++cnt;
		pos=trie[pos][to];
	}
	cntword[pos]++;
	rk[pos]=id;
}
 
void getfail()
{
	queue<int>q;
	for(int i=0;i<130;i++)
	{
		if(trie[0][i])
		{
			fail[trie[0][i]]=0;
			q.push(trie[0][i]);
		}
	}
	while(!q.empty())
	{
		int cur=q.front();
		q.pop();
		for(int i=0;i<130;i++)
		{
			if(trie[cur][i])
			{
				fail[trie[cur][i]]=trie[fail[cur]][i];
				q.push(trie[cur][i]);
			}
			else
				trie[cur][i]=trie[fail[cur]][i];
		}
	}
}

int search_word(int id)
{
	vector<int>ans;
	int len=strlen(s),pos=0;
	for(int i=0;i<len;i++)
	{
		int to=s[i];
		pos=trie[pos][to];
		int temp=pos;
		while(temp&&vis[temp]!=id)
		{
			vis[temp]=id;
			if(cntword[temp])//如果匹配成功
				 ans.push_back(rk[temp]);
			temp=fail[temp];
		}
	}
	if(ans.size())
	{
		tot++;
		sort(ans.begin(),ans.end());
		printf("web %d:",id);
		for(int i=0;i<ans.size();i++)
			printf(" %d",ans[i]);
		printf("\n");
	}
}

int main()
{
//	freopen("input.txt","r",stdin);
//	ios::sync_with_stdio(false);
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%s",s);
		insert_word(i);
	}
	getfail();
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%s",s);
		search_word(i);
	}
	printf("total: %d\n",tot);
	
	
 
 
 
 
 
	
	
	
	
	
	
	
	
	
	return 0;
}

来源：CSDN

作者：Frozen_Guardian

链接：https://blog.csdn.net/qq_45458915/article/details/104143936

标签

ac自动机