UOJ #35. BZOJ 1031 后缀排序 后缀数组模板

真-模板题系列。板子直接用了sort，O(n*logn*logn)的，将就着T吧。

UOJ 35 需要用到高度数组，BZOJ 1031把字符串重复两次求个后缀数组就可以了。

UOJ 35：

#include <bits/stdc++.h>
using namespace std;
const int maxn=100005;
string s;
int n,K,Rank[maxn],tmp[maxn],sa[maxn],lcp[maxn];
inline bool compare_sa(int i,int j) {
    if (Rank[i]!=Rank[j])
        return Rank[i]<Rank[j];
    else {
        int ri=(i+K<=n)?Rank[i+K]:-1;
        int rj=(j+K<=n)?Rank[j+K]:-1;
        return ri<rj;
    }
}
inline void construct_sa(string S,int *sa) {
    n=S.length();
    for (int i=0;i<=n;++i) {
        sa[i]=i;
        Rank[i]=i<n?S[i]:-1;
    }
    for (K=1;K<=n;K*=2) {
        sort(sa,sa+n+1,compare_sa);
        tmp[sa[0]]=0;
        for (int i=1;i<=n;++i)
            tmp[sa[i]]=tmp[sa[i-1]]+(compare_sa(sa[i-1],sa[i])?1:0);
        for (int i=0;i<=n;++i)
            Rank[i]=tmp[i];
    }
}
inline void construct_lcp(string S,int *sa,int *lcp) {
    n=S.length();
    int h=0;
    lcp[0]=0;
    for (int i=0;i<n;++i) {
        int j=sa[Rank[i]-1];
        h=max(h-1,0);
        for (;j+h<n&&i+h<n;++h)
            if (S[j+h]!=S[i+h])
                break;
        lcp[Rank[i]-1]=h;
    }
}
inline void print() {
    for (int i=1;i<=n;++i)
        printf("%d%c",sa[i]+1,i==n?'\n':' ');
    for (int i=1;i<n;++i)
        printf("%d%c",lcp[i],i==n?'\n':' ');
}
int main()
{
    cin>>s;
    construct_sa(s,sa);
    construct_lcp(s,sa,lcp);
    print();
    return 0;
}

BZOJ 1031：

#include <bits/stdc++.h>
using namespace std;
const int maxn=200005;
string s;
int n,K,Rank[maxn],tmp[maxn],sa[maxn],lcp[maxn];
inline bool compare_sa(int i,int j) {
    if (Rank[i]!=Rank[j])
        return Rank[i]<Rank[j];
    else {
        int ri=(i+K<=n)?Rank[i+K]:-1;
        int rj=(j+K<=n)?Rank[j+K]:-1;
        return ri<rj;
    }
}
inline void construct_sa(string S,int *sa) {
    n=S.length();
    for (int i=0;i<=n;++i) {
        sa[i]=i;
        Rank[i]=i<n?S[i]:-1;
    }
    for (K=1;K<=n;K*=2) {
        sort(sa,sa+n+1,compare_sa);
        tmp[sa[0]]=0;
        for (int i=1;i<=n;++i)
            tmp[sa[i]]=tmp[sa[i-1]]+(compare_sa(sa[i-1],sa[i])?1:0);
        for (int i=0;i<=n;++i)
            Rank[i]=tmp[i];
    }
}
inline void construct_lcp(string S,int *sa,int *lcp) {
    n=S.length();
    int h=0;
    lcp[0]=0;
    for (int i=0;i<n;++i) {
        int j=sa[Rank[i]-1];
        h=max(h-1,0);
        for (;j+h<n&&i+h<n;++h)
            if (S[j+h]!=S[i+h])
                break;
        lcp[Rank[i]-1]=h;
    }
}
inline void print() {
    for (int i=1;i<=n;++i)
        if (sa[i]<(n>>1))
            putchar(s[sa[i]+(n>>1)-1]);
}
int main()
{
    cin>>s;
    s+=s;
    construct_sa(s,sa);
    print();
    return 0;
}