[tyvj1860]后缀数组

题目链接：http://www.tyvj.cn/p/1860

解题关键：模板题。贴一个代码详解

http://www.cnblogs.com/staginner/archive/2012/02/02/2335600.html

注意：一个字符串中的所有子串都必然是它的后缀的前缀

da算法，必须保证字符串中每个字符的映射>=1

注意点：sa从0~n-1，rank1数组和height数组从1-n

height[1]恒为0

 #include<cstdlib>
 #include<cstring>
 #include<cstdio>
 #include<algorithm>
 #include<iostream>
 #include<cmath>
 using namespace std;
 const int N=;
 int wa[N],wb[N],wv[N],wc[N];
 bool cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}//比较两个字符串是否完全相同
 void make_sa(int *r,int *sa,int n,int m){//最大值小于m
     int i,j,p,*x=wa,*y=wb;//x,y是指针
     for(i=;i<m;i++) wc[i]=;
     for(i=;i<n;i++) wc[x[i]=r[i]]++;
     for(i=;i<m;i++) wc[i]+=wc[i-];
     for(i=n-;i>=;i--) sa[--wc[x[i]]]=i;//基数排序
     for(j=,p=;p<n;j*=,m=p){
         for(p=,i=n-j;i<n;i++) y[p++]=i;
         for(i=;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;//y是第二关键词的sa
         for(i=;i<n;i++) wv[i]=x[y[i]];
         for(i=;i<m;i++) wc[i]=;
         for(i=;i<n;i++) wc[wv[i]]++;
         for(i=;i<m;i++) wc[i]+=wc[i-];
         for(i=n-;i>=;i--) sa[--wc[wv[i]]]=y[i];
         for(swap(x,y),p=,x[sa[]]=,i=;i<n;i++) x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;//x是rank数组，1 1 2 2 2 3 这种
     }
     return;
 }
 int rank1[N],height[N],sa[N];//注意sa和rank的对应关系，sa的数值从0开始，rank的数值从1开始
 void make_height(int *r,int *sa,int n){
     int i,j,k=;
     for(i=;i<=n;i++) rank1[sa[i]]=i;
     for(i=;i<n;height[rank1[i++]]=k)
         for(k?k--:,j=sa[rank1[i]-];r[i+k]==r[j+k];k++);
     return;
 }
 //sa是名次所在位置，rank是位置所在名次
 int r[N<<];
 int main() {
     ios::sync_with_stdio();
     int n=;
     string s;
     cin>>s;
     for(int i=;i<s.size();i++) r[n++]=s[i]-'a'+; r[n]=;
     make_sa(r,sa,n+,);
     make_height(r,sa,n);
     for(int i=;i<=n;i++) cout<<sa[i]+<<" ";cout<<"\n";//这里的1是逻辑上的首位
     for(int i=;i<=n;i++) cout<<height[i]<<" ";cout<<"\n";
 }

下面这种调用方式更无脑一些

 #include <cstdlib>
 #include <cstring>
 #include <cstdio>
 #include <algorithm>
 #include<iostream>
 #include<cstdlib>
 using namespace std;

 const int N = ;
 int wa[N],wb[N],wv[N],wc[N];

 bool cmp(int *r,int a,int b,int l){
     return r[a]==r[b]&&r[a+l]==r[b+l];
 }

 void make_sa(int *r,int *sa,int n,int m){//m是至多有多少个字符，up的意思
     int i,j,p,*x=wa,*y=wb;
     for(i=;i<m;i++) wc[i]=;
     for(i=;i<n;i++) wc[x[i]=r[i]]++;
     for(i=;i<m;i++) wc[i]+=wc[i-];
     for(i=n-;i>=;i--) sa[--wc[x[i]]]=i;
     for(j=,p=;p<n;j*=,m=p){
         for(p=,i=n-j;i<n;i++) y[p++]=i;
         for(i=;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
         for(i=;i<n;i++) wv[i]=x[y[i]];
         for(i=;i<m;i++) wc[i]=;
         for(i=;i<n;i++) wc[wv[i]]++;
         for(i=;i<m;i++) wc[i]+=wc[i-];
         for(i=n-;i>=;i--) sa[--wc[wv[i]]]=y[i];
         for(swap(x,y),p=,x[sa[]]=,i=;i<n;i++)
             x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
     }
     return;
 }

 int rank1[N],height[N],sa[N];//输出都是从1开始存的，rank从0开始，最后需要+1，读入从下标0
 void make_height(int *r,int *sa,int n){
     int i,j,k=;
     for(i=;i<=n;i++) rank1[sa[i]]=i;
     for(i=;i<n;height[rank1[i++]]=k)
         for(k?k--:,j=sa[rank1[i]-];r[i+k]==r[j+k];k++);
     return;
 }
 //sa是名次所在位置，rank是位置所在名次
 int r[N];
 int main() {
     ios::sync_with_stdio();
     cin.tie();
     cout.tie();
     string s;
     cin>>s;
     for(int i=;i<s.size();i++) r[i]=(int)s[i];
     int t=(int)s.size();
     r[t]=;
     make_sa(r, sa,t+,);//注意+1，因为补0的原因
     make_height(r, sa, t);
     for(int i=;i<=t;i++) cout<<sa[i]+<<" ";
     cout<<"\n";
     for(int i=;i<=t;i++) cout<<height[i]<<" ";
     cout<<"\n";
     return ;
 }

后缀数组练习

poj2774 最长公共子串问题，将两个字符串接在一起解决。height数组就代表lcp（最长公共前缀）

 #include <cstdlib>
 #include <cstring>
 #include <cstdio>
 #include <algorithm>
 #include<iostream>
 #include<cstdlib>
 using namespace std;
 const int N=;
 int wa[N],wb[N],wv[N],wc[N],n,m;
 bool cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}
 void make_sa(int *r,int *sa,int n,int m){
     int i,j,p,*x=wa,*y=wb;
     for(i=;i<m;i++) wc[i]=;
     for(i=;i<n;i++) wc[x[i]=r[i]]++;
     for(i=;i<m;i++) wc[i]+=wc[i-];
     for(i=n-;i>=;i--) sa[--wc[x[i]]]=i;
     for(j=,p=;p<n;j*=,m=p){
         for(p=,i=n-j;i<n;i++) y[p++]=i;
         for(i=;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
         for(i=;i<n;i++) wv[i]=x[y[i]];
         for(i=;i<m;i++) wc[i]=;
         for(i=;i<n;i++) wc[wv[i]]++;
         for(i=;i<m;i++) wc[i]+=wc[i-];
         for(i=n-;i>=;i--) sa[--wc[wv[i]]]=y[i];
         for(swap(x,y),p=,x[sa[]]=,i=;i<n;i++) x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
     }
     return;
 }
 int rank1[N],height[N],sa[N];
 void make_height(int *r,int *sa,int n){
     int i,j,k=;
     for(i=;i<=n;i++) rank1[sa[i]]=i;
     for(i=;i<n;height[rank1[i++]]=k)
         for(k?k--:,j=sa[rank1[i]-];r[i+k]==r[j+k];k++);
     return;
 }

 int r[N<<];
 int main() {
     ios::sync_with_stdio();
     string s,t;
     cin>>s>>t;
     int n=;
     for(int i=;i<s.size();i++) r[n++]=s[i]-'a'+; r[n++]=;int sl=n;
     for(int i=;i<t.size();i++) r[n++]=t[i]-'a'+; r[n]=;
     make_sa(r, sa, n+, );
     make_height(r, sa, n);
     int ans=;
     for(int i=;i<n;i++){
         if((sa[i]<sl)!=(sa[i+]<sl)){
             ans=max(ans,height[i+]);
         }
     }
     cout<<ans<<"\n";
     return ;
 }