病毒侵袭 HDU - 2896 AC自动机

AC自动机基础题，

每个字符存一下是哪几个病毒的结尾字符。

然后查询时直接存下来即可。

注意每次flag要重新更新。

#include <cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
#define pb push_back
const int M = 1e5+7;
vector<int>pr,ty[M];
struct AC
{
	int tr[M][130],flag[M],fail[M];
	int cnt=1;queue<int>q;
	void init()
	{
		for(int i=0;i<=cnt;i++)fail[i]=flag[i]=0;
		for(int i=1;i<=cnt;i++)
			for(int j=0;j<128;j++)tr[i][j]=0;
		cnt=1;
	}
	void in(char *s,int id)
	{
		int len=strlen(s),u=1;
		for(int i=0;i<len;i++)
		{
			int v=s[i];
			if(!tr[u][v])tr[u][v]=++cnt;//新建节点
			u=tr[u][v]; 
		}
		flag[u]++;//这个节点字符是几个字符串的结尾 
		ty[u].pb(id);
	}
	void get_fail()
	{
		for(int i=0;i<128;i++)tr[0][i]=1;
		q.push(1);fail[1]=0;
		while(!q.empty())
		{
			int u=q.front();q.pop();
			for(int i=0;i<128;i++)
			{
				int v=tr[u][i];//遍历u所有儿子，这样不同记录fa 
				int Fail=fail[u];//由于BFS遍历，fail[u]已经处理好了，现在是找v的fail
				if(!v)tr[u][i]=tr[Fail][i];//不存在节点v，这样做的目的是，后面fail指针失配时直接不断返回fail 
				else fail[v]=tr[Fail][i],q.push(v);//存在实节点才入队列 
			}
		}
	}
	int qu(char *s)
	{
		int u=1,ans=0,len=strlen(s);
		for(int i=0;i<len;i++)
		{
			int v=s[i];
			int k=tr[u][v];//跳fail 
			while(k>1&&flag[k]!=-1)//经过就不统计了
			{
				ans+=flag[k],flag[k]=-1;
				for(auto x:ty[k])pr.pb(x);
				k=fail[k];
			}
			u=tr[u][v]; 
		}
		return ans;
	}
}ac;
char s[M],p[M];
int ft[M];
int main()
{
	ios::sync_with_stdio(false);
  	cin.tie(0);
  	int n,m;
  	cin>>n;
  	ac.init();
  	for(int T=1;T<=n;T++)
  	{
  		cin>>s;
  		ac.in(s,T);
	}
	ac.get_fail();
	for(int i=0;i<=ac.cnt;i++)ft[i]=ac.flag[i];
	cin>>m;
	int ans=0;
	for(int T=1;T<=m;T++)
	{
		cin>>p;
		for(int i=0;i<=ac.cnt;i++)ac.flag[i]=ft[i];
		pr.clear();
		ac.qu(p);
		if(pr.size())
		{
			cout<<"web "<<T<<":";
			ans++;
			sort(pr.begin(),pr.end());
			for(int i=0;i<pr.size();i++)cout<<" "<<pr[i];
			cout<<endl;
		}
	}
	cout<<"total: "<<ans<<endl;
	return 0;
}