模板—字符串—广义后缀自动机
模板—字符串—广义后缀自动机
Code(假广义后缀自动机):
#include <bits/stdc++.h> using namespace std; #define N 100010 map <int,int> son[N<<1]; char str[N]; int n; int tot=1,last=1,pre[N<<1],dis[N<<1],type[N<<1],many[N<<1],kind[N<<1]; namespace Sam { void pushup(int p,int x) {while(p&&type[p]!=x) many[p]++,type[p]=x,p=pre[p];} void insert(int x,int Tim) { int p=last,np=last=++tot; dis[np]=dis[p]+1,kind[np]=Tim; while(p&&!son[p][x]) son[p][x]=np,p=pre[p]; if(!p) {pre[np]=1,pushup(np,Tim);return;} int q=son[p][x]; if(dis[q]==dis[p]+1) {pre[np]=q,pushup(np,Tim);return;} int nq=++tot; dis[nq]=dis[p]+1,son[nq]=son[q],pre[nq]=pre[q]; type[nq]=type[q],many[nq]=many[q],pre[q]=pre[np]=nq; while(p&&son[p][x]==q) son[p][x]=nq,p=pre[p]; pushup(np,Tim); } } using namespace Sam; int main() { scanf("%d",&n); for(int i=1,len;i<=n;i++) { scanf("%s",str+1),len=strlen(str+1),last=1; for(int j=1;j<=len;j++) insert(str[j]-'a',i); } }
Code(真广义后缀自动机):
#include <bits/stdc++.h> using namespace std; #define N 100010 map <int,int> son[N<<1]; char str[N]; int n; int tot=1,last=1,pre[N<<1],dis[N<<1]; namespace Sam { void insert(int x,int Tim) { int p=last,np=last=++tot; if(son[p][x]) { int q=son[p][x]; if(dis[q]==dis[p]+1) last=q; else { int nq=++tot; last=nq; pre[nq]=pre[q],dis[nq]=dis[p]+1; pre[q]=nq,son[nq]=son[q]; while(p&&son[p][x]==q) son[p][x]=nq,p=pre[p]; } } dis[np]=dis[p]+1; while(p&&!son[p][x]) son[p][x]=np,p=pre[p]; if(!p) {pre[np]=1;return;} int q=son[p][x]; if(dis[q]==dis[p]+1) {pre[np]=q;return;} int nq=++tot; dis[nq]=dis[p]+1,son[nq]=son[q],pre[nq]=pre[q],pre[q]=pre[np]=nq; while(p&&son[p][x]==q) son[p][x]=nq,p=pre[p]; } } using namespace Sam; int main() { scanf("%d",&n); for(int i=1,len;i<=n;i++) { scanf("%s",str+1),len=strlen(str+1),last=1; for(int j=1;j<=len;j++) insert(str[j]-'a',i); } }