【练习05】 AC自动机 1001 病毒侵袭

算法思路：AC自动机。

要点：

1.val存储病毒的编号；

2.find函数中当找到一个模板后，应该顺着失配边往回走，由last数组递归判断是否还有其他的子模板能够匹配，所有找到的模板的val值保存到数组save中。

3.对save数组进行sort排序，注意输出格式，依次打印病毒编号。

4.题中要求是“可见字符”，在不超内存的情况下，sigma_size最好开到127。

//模板开始
#include <string>
#include <vector>
#include <algorithm>
#include <iostream>
#include <sstream>
#include <fstream>
#include <map>
#include <set>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <iomanip>
#include <string.h>
#include <queue>
#define SZ(x) (int(x.size()))
using namespace std;

int toInt(string s){
istringstream sin(s);
int t;
sin>>t;
return t;
}
template<class T> string toString(T x){
ostringstream sout;
sout<<x;
return sout.str();
}
typedef long long int64;
int64 toInt64(string s){
istringstream sin(s);
int64 t;
sin>>t;
return t;
}
template<class T> T gcd(T a, T b){
if(a<0)
return gcd(-a, b);
if(b<0)
return gcd(a, -b);
return (b == 0)? a : gcd(b, a % b);
}
//模板结束（通用部分）

#define ifs cin

#define maxnode 100005
#define sigma_size 127

int save[maxnode];
int cnt;

struct Trie
{
int ch[maxnode][sigma_size];
int val[maxnode];
int sz;
int f[maxnode];
int last[maxnode];
Trie()
{
sz = 1;
memset(ch[0], 0, sizeof(ch[0]));
}
int idx(char c)
{
return (int)c;
}

void insert(char* s, int v)
{
int u = 0, n = strlen(s);
for(int i = 0; i < n; i++)
{
int c = idx(s[i]);
if(!ch[u][c])
{
memset(ch[sz], 0, sizeof(ch[sz]));
val[sz] = 0;
ch[u][c] = sz++;
}
u = ch[u][c];
}
val[u] = v;
}

//int search(char* s, int n)
//{
//	int u = 0;
//	int flag = 1;
//	for(int i = 0; i < n; i++)
//	{
//		int c = idx(s[i]);
//		if(ch[u][c] == 0)
//		{
//			flag = 0;
//			break;
//		}
//		u = ch[u][c];
//	}
//	if(flag && val[u])
//	{
//		return 1;
//	}
//	else
//	{
//		return 0;
//	}
//}

void getFail()
{
queue<int> q;
f[0] = 0;
for(int c = 0; c < sigma_size; c++)
{
int u = ch[0][c];
if(u)
{
f[u] = 0;
q.push(u);
last[u] = 0;

}

while(!q.empty())
{
int r = q.front();
q.pop();
for(int c = 0; c < sigma_size; c++)
{
int u = ch[r][c];
if(!u)
{
continue;
}
q.push(u);
int v = f[r];
while(v && !ch[v][c])
{
v = f[v];
}
f[u] = ch[v][c];
last[u] = val[f[u]] ? f[u] : last[f[u]];
}
}
}
}

void save_all(int j)
{
if(j)
{
save_all(last[j]);
save[cnt++] = val[j];
}
}

void find(char* T)
{
int n = strlen(T);
int j = 0;
for(int i = 0; i < n; i++)
{
int c = idx(T[i]);
while(j && !ch[j][c])
{
j = f[j];
}
j = ch[j][c];
if(val[j])
{
save_all(j);
}
else
{
if(last[j])
{
save_all(last[j]);
}
}
}
}
};

Trie t;
char a[505][205];
char b[10005];
int jishu;

//【练习05】 AC自动机 1001 病毒侵袭

int main()
{
//ifstream ifs("shuju.txt", ios::in);
int n;
while(ifs>>n)
{
for(int i = 1; i <= n; i++)
{
//ifs>>a[i];
scanf("%s", &a[i]);
t.insert(a[i], i);
}
t.getFail();

ifs>>n;
jishu = 0;
for(int i = 1; i <= n; i++)
{
//ifs>>b;
scanf("%s", &b);
cnt = 0;
memset(save, 0, sizeof(save));
t.find(b);
if(!save[0])
{
continue;
}
else
{
jishu++;
}
sort(save, save + cnt);
cout<<"web "<<i<<":";
for(int j = 0; j < cnt; j++)
{
cout<<" "<<save[j];
}
cout<<endl;
}
cout<<"total: "<<jishu<<endl;
}

return 0;
}