ac自动机模板

template<int maxnode,int sigma_size>

struct Aho_Corasick{

int ch[maxnode][sigma_size];

int val[maxnode],last[maxnode],f[maxnode];

int sz;

//int cnt[maxnode];

//bool mark[maxnode];

Aho_Corasick() {

sz=1;

memset(ch[0],0,sizeof(ch[0]));

}

int idx(char c) {

return c-'a';

}

void init() {

sz=1;

memset(ch[0],0,sizeof(ch[0]));

// memset(cnt,0,sizeof(cnt));

//memset(mark,0,sizeof(mark));

}

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;

cnt[u]++;

}

int find(char *T) {

int ans=0;

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]) print(j,ans);

else print(last[j],ans);

//printf("FF %d %d\n",i,ans);

}

return ans;

}

void print(int j,int &ans) {

if(j && !mark[j]) {

ans+=cnt[j];

mark[j]=1;

print(last[j],ans);

}

}

void getFail() {

queue<int>q;

f[0]=0;

last[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) { ch[r][c]=ch[f[r]][c]; 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]];

}

}

}

};