PAT T1005 Programming Pattern

建立后缀数组，遍历height数组找到连续大于len的最长子序列~

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e7+;
char s[maxn];
int n;
int rk[maxn];
int sa[maxn];
int height[maxn];
int tmp[maxn];
int c[maxn];
int t1[maxn];
int t2[maxn];
void build_sa (char s[],int n,int m) {
    int i,j,p;
    int *x=t1;
    int *y=t2;
    for (i=;i<m;i++) c[i]=;
    for (i=;i<n;i++) c[x[i]=s[i]]++;
    for (i=;i<m;i++) c[i]+=c[i-];
    for (i=n-;i>=;i--) sa[--c[x[i]]]=i;
    for (j=;j<=n;j<<=) {
        p=;
        for (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<m;i++) c[i]=;
        for (i=;i<n;i++) c[x[y[i]]]++;
        for (i=;i<m;i++) c[i]+=c[i-];
        for (i=n-;i>=;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=;
        x[sa[]]=;
        for (i=;i<n;i++)
        x[sa[i]]=y[sa[i-]]==y[sa[i]]&&y[sa[i-]+j]==y[sa[i]+j]?p-:p++;
        if (p>=n) break;
        m=p;
    }
}
void getHeight (char s[],int n) {
    int i,j,k=;
    for (i=;i<=n;i++) rk[sa[i]]=i;
    for (i=;i<n;i++) {
        if (k) k--;
        j=sa[rk[i]-];
        while (s[i+k]==s[j+k]) k++;
        height[rk[i]]=k;
    }
}
int main () {
    int len;
    scanf ("%d",&len);
    getchar ();
    cin.getline (s,maxn);
    n=strlen(s);
    build_sa (s,n+,);
    getHeight (s,n);
    int cnt=;
    int res=;
    int ans=;
    for (int i=;i<=n;i++) {
        if (height[i]>=len) ans++;
        else ans=;
        if (sa[i]+len<=n&&cnt<ans) {
            cnt=ans;
            res=sa[i];
        }
    }
    for (int i=;i<len;i++)
    printf ("%c",s[res+i]);
    printf (" %d",cnt);
    return ;
}