hdu 2896 病毒侵袭 ac自动机
#include <cstdio> #include <cstring> #include <queue> using namespace std; #define MAXNODE 100100 #define SIGMA_SIZE 135 int ch[MAXNODE][SIGMA_SIZE]; int f[MAXNODE]; // fail函数 int val[MAXNODE]; // 每个字符串的结尾结点都有一个非0的val int last[MAXNODE]; // 输出链表的下一个结点 bool vis[510]; int sz; char viru[210],web[11000]; int flag; void init() { sz = 1; memset(ch[0], 0, sizeof(ch[0])); } // 插入字符串。v必须非0 void insert(char *s, int v) { int u = 0, n = strlen(s); for(int i = 0; i < n; i++) { int c = 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; } // 递归打印以结点j结尾的所有字符串 void print(int j) { if(j) { flag=1; vis[ val[j] ]=1; print(last[j]); } } // 在T中找模板 void find(char* T) { int n = strlen(T); int j = 0; // 当前结点编号,初始为根结点 for(int i = 0; i < n; i++) { // 文本串当前指针 int c = T[i]; while(j && !ch[j][c]) j = f[j]; // 顺着细边走,直到可以匹配 j = ch[j][c]; if(val[j]) print(j); else if(last[j]) print(last[j]);// 找到了! } } // 计算fail函数 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; } } // 按BFS顺序计算fail 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]]; } } } int main() { int n,m; while(scanf("%d",&n)!=EOF) { int i,j; init(); for(i=1;i<=n;i++) { scanf("%s",viru); insert(viru,i); } getFail(); scanf("%d",&m); int sum=0; for(i=1;i<=m;i++) { flag=0; memset(vis,0,sizeof(vis)); scanf("%s",web); find(web); if(flag) { sum++; printf("web %d:",i); for(j=1;j<=n;j++) if(vis[j]) printf(" %d",j); printf("\n"); } } printf("total: %d\n",sum); } return 0; }