HDU4080后缀数组

出现K次或K次以上最长子串，并且输出最右边的子串首字母坐标。

和之前总结的一个题一样，二分长度，找到统计height数组大于等于K的个数，满足返回true；这里因为还要找最右边子串首字母坐标，因而需要同时找sa数组（记录下标的数组）里的最大值。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdlib>
const int maxn = 40005;
using namespace std;
int sa[maxn];
int Rank[maxn], height[maxn];
int t1[maxn],t2[maxn],c[maxn];
//求sa数组需要的中间变量，不需要赋值
//带排序的字符串放在s数组中，从s[0] - s[n-1],
//长度为n，且最大值小于m
//除了s[n-1]之外的所有s[i] 都大于0，r[n-1] = 0
//函数结束后，结果放在sa数组中
bool cmp(int *r,int a,int b,int l){
    return r[a] == r[b] && r[a+l] == r[b+l];
}
void da(int str[],int sa[],int Rank[],int height[],int n,int m){
    n++;
    int i,j,p,*x = t1,*y = t2;
    for(i = 0;i< m ;i ++) c[i] = 0;
    for(i=0;i<n;i++) c[x[i] = str[i]   ]++;
    for(i=1;i<m;i++) c[i] += c[i-1];
    for(i=n-1;i>=0;i--) sa[--c[x[i] ]] = i;
    for(j=1;j<=n;j<<=1){
        p = 0;
        //直接利用sa数组排序第二关键字
        for(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保存的就是按照第二关键字排序的结果
        //基数排序第一关键字
        for(i=0;i<m;i++) c[i] = 0;
        for(i=0;i<n;i++) c[x[y[i]] ]++;
        for(i=1;i<m;i++) c[i] += c[i-1];
        for(i=n-1;i>=0;i--) sa[--c[x[y[i] ]] ] = y[i];
        //根据sa和x数组计算新的x数组
        swap(x,y);
        p = 1;x[ sa[0]] = 0;
        for(i=1;i<n;i++)
            x[sa[i] ] = cmp(y,sa[i-1],sa[i],j) ? p-1:p++;
        if(p >= n) break;
        m = p; //下次基数排序的最大值
    }
    int k = 0;
    n--;
    for(i=0;i<=n;i++) Rank[sa[i] ] = i;
    for(i=0;i<n;i++){
        if(k) k--;
        j = sa[Rank[i] - 1 ];
        while(str[i+k] == str[j+k] ) k++;
        height[Rank[i] ] = k;
    }
}

int RMQ[maxn];
int mm[maxn];
int best[20][maxn];
void initRMQ(int n){
    mm[0] = -1;
    for(int i=1;i<=n;i++)
        mm[i] = ((i & (i-1))==0 ) ? mm[i-1] +1 : mm[i-1];
    for(int i=1;i<=n;i++) best[0][i] = i;
    for(int i=1;i<=mm[n];i++)
        for(int j=1;j+(1<<i)-1 <=n;j++){
            int a  = best[i-1][j];
            int b = best[i-1][j+(1<<(i-1)) ];
            if(RMQ[a] < RMQ[b]) best[i][j] = a;
            else best[i][j] = b;
        }
}
int askRMQ(int a,int b){
    int t;
    t = mm[b-a+1];
    b-=(1<<t)-1;
    a = best[t][a] ; b = best[t][b];
    return RMQ[a] < RMQ[b] ? a:b;
}
int lcp(int a,int b){
    a = Rank[a]; b = Rank[b];
    if(a > b) swap(a,b);
    return height[askRMQ(a+1,b)];
}

char s[maxn];
int arr[maxn];
    int n,m;
int maxx;
bool fun(int k){
    int cnt = 1;
    int tmp = 0;
    int flag = 0;
    for(int i=2;i<=n;i++){
        if(height[i] >= k){
            cnt ++;
            tmp = max(tmp,max(sa[i-1],sa[i])); //下标最大值
        }else{
            cnt = 1; tmp = 0;
        }
        if(cnt >= m){
            if(!flag) maxx = tmp;
            else maxx = max(maxx,tmp);
            flag = 1; //cnt 大于 m之后不直接返回，继续统计求maxx
        }

    }
    return flag;
}
int main(){

    while(scanf("%d",&m) != EOF && m){
        scanf("%s",s);
        n = strlen(s);
        int tmpmax  =0 ;
        for(int i=0;i<n;i++){
            arr[i] = s[i];
            tmpmax = max(tmpmax,arr[i]);
        }
        arr[n] = 0;
        da(arr,sa,Rank,height,n,tmpmax +1);
        int l = 1,r = n;
        int ans = 0;
        while(l <= r){
            int mid = (l + r)>>1;
            if(fun(mid)){
                ans = mid;
                l = mid + 1;
            }else
                r = mid - 1;
        }
        if(ans != 0){
            printf("%d %d\n",ans,maxx);
        }else
            puts("none");
    }
}