[ACM模板]AC自动机

#include<stdio.h>
#include<vector>
#include<queue>
#include<string.h>
using namespace std;
#define ACMXNODE 110   //AC自动机节点数(最大为所有字符串字符个数和)
int charIdx(char ch){ return ch-'a'; }
#define CHARSIZE 128   //不同字符个数(128可以处理大部分情况，空间紧张可以减少这个值,但要保证charIdx每个字符都对应一个下标)
class ACAUTO{
private:
  int sz;
  int ch[ACMXNODE][CHARSIZE];
  int val[ACMXNODE];       //节点标记值，0表示非叶子节点。叶子节点记录一个自定义值（如是第几个字符串）。（相同字符串会互相覆盖）
  int f[ACMXNODE];         //fail数组
  int last[ACMXNODE];      last[u]：上一个u结尾的字符串。如有两个字符串(abcd,bc)，节点u表示abc,则节点last[u]指向bc。
  int print(int i,int j);
public:
  ACAUTO();
  //初始化+建图
  int clearMap();          //初始化
  int ins(char *s,int v);  //插入一个字符串，v代表字符串结尾节点标记值。(0：非字符串结尾，其他：自定义)
  int getFail();           //建立失配指针。bfs查找
  int kmpFind(char *T);    //找出trie树中所有为T子串的字符串
};
ACAUTO::ACAUTO(){clearMap();}
int ACAUTO::clearMap(){
  sz=1;
  memset(ch[0],0,sizeof(ch[0]));
  memset(val,0,sizeof(val));
  return 0;
}
int ACAUTO::ins(char* s,int v=1){
  int u=0;
  int n=strlen(s);
  for (int i=0;i<n;i++){
    int c=charIdx(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;
  return 0;
}
int ACAUTO::getFail(){
  queue<int> q;
  //先处理节点0的fail值
  f[0]=0;
  for (int c=0;c<CHARSIZE;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<CHARSIZE;c++){
       int u=ch[r][c];
       if (!u){ch[r][c]=ch[f[r]][c];continue;}
       q.push(u);
       f[u]=ch[f[r]][c];
       last[u]=val[f[u]]?f[u]:last[f[u]];
    }
  }
}

int ACAUTO::kmpFind(char *T){
  int n=strlen(T);
  int j=0;
  for (int i=0;i<n;i++){
    int c=charIdx(T[i]);
    j=ch[j][c];
    if (val[j])print(i,j);
      else
      if (last[j]) print(i,last[j]);
  }
}
int ACAUTO::print(int i,int j){
  if (j){
     printf("模板串T第%d个字符结尾子串和AC自动机编号为%d的一样,last[%d]=%d\n",i+1,val[j],j,last[j]);
     print(i,last[j]);
  }
  return 0;
}
ACAUTO ac;
int main(){
  int n;
  scanf("%d",&n);
  char s[100];
  ac.clearMap();
  for (int i=1;i<=n;i++){
    scanf("%s",s);
    ac.ins(s,i);
  }
  ac.getFail();
  while (1){
    scanf("%s",s);
    ac.kmpFind(s);
  }
  return 0;
}