[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=200005;
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=0;i<m;i++) wc[i]=0;
    for(i=0;i<n;i++) wc[x[i]=r[i]]++;
    for(i=1;i<m;i++) wc[i]+=wc[i-1];
    for(i=n-1;i>=0;i--) sa[--wc[x[i]]]=i;//基数排序
    for(j=1,p=1;p<n;j*=2,m=p){
        for(p=0,i=n-j;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;//y是第二关键词的sa 
        for(i=0;i<n;i++) wv[i]=x[y[i]];
        for(i=0;i<m;i++) wc[i]=0;
        for(i=0;i<n;i++) wc[wv[i]]++;
        for(i=1;i<m;i++) wc[i]+=wc[i-1];
        for(i=n-1;i>=0;i--) sa[--wc[wv[i]]]=y[i];
        for(swap(x,y),p=1,x[sa[0]]=0,i=1;i<n;i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1: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=0;
    for(i=1;i<=n;i++) rank1[sa[i]]=i;
    for(i=0;i<n;height[rank1[i++]]=k)
        for(k?k--:0,j=sa[rank1[i]-1];r[i+k]==r[j+k];k++);
    return;
}
//sa是名次所在位置，rank是位置所在名次
int r[N<<1];
int main() {
    ios::sync_with_stdio(0);
    int n=0;
    string s;
    cin>>s;
    for(int i=0;i<s.size();i++) r[n++]=s[i]-'a'+1; r[n]=0;
    make_sa(r,sa,n+1,27);
    make_height(r,sa,n);
    for(int i=1;i<=n;i++) cout<<sa[i]+1<<" ";cout<<"
";//这里的1是逻辑上的首位 
    for(int i=1;i<=n;i++) cout<<height[i]<<" ";cout<<"
";
}

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

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

const int N = 200005;
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=0;i<m;i++) wc[i]=0;
    for(i=0;i<n;i++) wc[x[i]=r[i]]++;
    for(i=1;i<m;i++) wc[i]+=wc[i-1];
    for(i=n-1;i>=0;i--) sa[--wc[x[i]]]=i;
    for(j=1,p=1;p<n;j*=2,m=p){
        for(p=0,i=n-j;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(i=0;i<n;i++) wv[i]=x[y[i]];
        for(i=0;i<m;i++) wc[i]=0;
        for(i=0;i<n;i++) wc[wv[i]]++;
        for(i=1;i<m;i++) wc[i]+=wc[i-1];
        for(i=n-1;i>=0;i--) sa[--wc[wv[i]]]=y[i];
        for(swap(x,y),p=1,x[sa[0]]=0,i=1;i<n;i++)
            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1: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=0;
    for(i=1;i<=n;i++) rank1[sa[i]]=i;
    for(i=0;i<n;height[rank1[i++]]=k)
        for(k?k--:0,j=sa[rank1[i]-1];r[i+k]==r[j+k];k++);
    return;
}
//sa是名次所在位置，rank是位置所在名次
int r[N];
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    string s;
    cin>>s;
    for(int i=0;i<s.size();i++) r[i]=(int)s[i];
    int t=(int)s.size();
    r[t]=0;
    make_sa(r, sa,t+1,128);//注意+1，因为补0的原因
    make_height(r, sa, t);
    for(int i=1;i<=t;i++) cout<<sa[i]+1<<" ";
    cout<<"
";
    for(int i=1;i<=t;i++) cout<<height[i]<<" ";
    cout<<"
";
    return 0;
}

后缀数组练习

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

#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include<iostream>
#include<cstdlib>
using namespace std;
const int N=200005;
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=0;i<m;i++) wc[i]=0;
    for(i=0;i<n;i++) wc[x[i]=r[i]]++;
    for(i=1;i<m;i++) wc[i]+=wc[i-1];
    for(i=n-1;i>=0;i--) sa[--wc[x[i]]]=i;
    for(j=1,p=1;p<n;j*=2,m=p){
        for(p=0,i=n-j;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(i=0;i<n;i++) wv[i]=x[y[i]];
        for(i=0;i<m;i++) wc[i]=0;
        for(i=0;i<n;i++) wc[wv[i]]++;
        for(i=1;i<m;i++) wc[i]+=wc[i-1];
        for(i=n-1;i>=0;i--) sa[--wc[wv[i]]]=y[i];
        for(swap(x,y),p=1,x[sa[0]]=0,i=1;i<n;i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
    return;
}
int rank1[N],height[N],sa[N];
void make_height(int *r,int *sa,int n){
    int i,j,k=0;
    for(i=1;i<=n;i++) rank1[sa[i]]=i;
    for(i=0;i<n;height[rank1[i++]]=k)
        for(k?k--:0,j=sa[rank1[i]-1];r[i+k]==r[j+k];k++);
    return;
}

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