后缀自动机(SAM)模板

最新推荐文章于 2024-08-03 17:41:27 发布

weixin_30455365

最新推荐文章于 2024-08-03 17:41:27 发布

阅读量70

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/TenderRun/p/5208074.html

 1 struct SAM{
 2     int ch[maxn][26],fa[maxn],len[maxn],cnt,last;
 3     void Init()
 4     {
 5         memset(ch,0,sizeof(ch));
 6         memset(fa,0,sizeof(fa));
 7         last=cnt=1;
 8     }
 9     void Add(int c)
10     {
11         int p=last,np=last=++cnt;
12         len[np]=len[p]+1;
13         while(!ch[p][c]&&p){
14             ch[p][c]=np;p=fa[p];
15         }
16         if(p==0)
17             fa[np]=1;
18         else{
19             int q=ch[p][c];
20             if(len[p]==len[q]-1){
21                 fa[np]=q;
22             }
23             else{
24                 int nq=++cnt;len[nq]=len[p]+1;
25                 memcpy(ch[nq],ch[q],sizeof(ch[q]));
26                 fa[nq]=fa[q];fa[q]=fa[np]=nq;
27                 while(ch[p][c]==q&&p){
28                     ch[p][c]=nq;
29                     p=fa[p];
30                 }
31             }
32         }
33     }
34 };

　　然后是自整理最全SAM模版，超级大杀器！

 1 #include <iostream>
 2 #include <cstring>
 3 #include <cstdio>
 4 using namespace std;
 5 const int N=200010;
 6 char s[N];
 7 int fa[N],pos[N],vis[N],sa[N],rank[N];
 8 int son[N][26],end[N],rht[N],lcp[N];
 9 int ch[N][26],len[N],id[N],tot;
10 int od[N],wv[N],lst,cnt;
11 
12 void Init(){
13     memset(ch,0,sizeof(ch));
14     memset(end,0,sizeof(end));
15     memset(son,0,sizeof(son));
16     memset(vis,0,sizeof(vis));
17     lst=cnt=1;tot=0;
18 }
19 
20 void Insert(int c){
21     int p=lst,np=lst=++cnt;end[lst]=1;
22     id[len[np]=len[p]+1]=np;rht[np]=1;
23     while(p&&!ch[p][c])ch[p][c]=np,p=fa[p];
24     if(!p)fa[np]=1;
25     else{
26         int q=ch[p][c],nq;
27         if(len[q]==len[p]+1)fa[np]=q;
28         else{
29             len[nq=++cnt]=len[p]+1;
30             fa[nq]=fa[q];fa[q]=fa[np]=nq;
31             memcpy(ch[nq],ch[q],sizeof(ch[q]));
32             while(ch[p][c]==q)ch[p][c]=nq,p=fa[p];
33         }
34     }
35 }
36 
37 void Get_Right(){
38     for(int i=1;i<=cnt;i++)wv[len[i]]++;
39     for(int i=1;i<=cnt;i++)wv[i]+=wv[i-1];
40     for(int i=1;i<=cnt;i++)od[wv[len[i]]--]=i;
41     for(int i=cnt;i>=1;i--)rht[fa[od[i]]]+=rht[od[i]];
42 }
43 
44 void Build_Tree(){
45     int l=strlen(s+1);
46     for(int i=l;i>=1;i--)Insert(s[i]-'a');
47     for(int i=l;i>=1;i--)
48         for(int x=id[i],p=l+1;x&&!pos[x];x=fa[x])
49             p-=len[x]-len[fa[x]],pos[x]=p;
50     for(int x=2;x<=cnt;x++)son[fa[x]][s[pos[x]]-'a']=x;    
51 }
52 
53 void DFS(int x,int l){
54     if(end[x])sa[rank[l-len[x]+1]=++tot]=l-len[x]+1;
55     for(int i=0;i<26;i++)if(son[x][i])DFS(son[x][i],l);
56 }
57 
58 void Build_SA(){
59     int l=strlen(s+1),k=0;DFS(1,l);
60     for(int i=1,j;i<=l;lcp[rank[i++]]=k)
61         for(k?k--:k,j=sa[rank[i]-1];s[i+k]==s[j+k];k++);
62 }
63 
64 int main(){
65     //freopen();
66     //freopen();
67     Init();
68     scanf("%s",s+1);
69     int l=strlen(s+1);
70     Build_Tree();Build_SA();
71     for(int i=1;i<=l;i++)printf("%d ",sa[i]);printf("\n");
72     for(int i=2;i<=l;i++)printf("%d ",lcp[i]);printf("\n");
73     return 0;
74 }