POJ 3261 Milk Patterns (离散化+后缀数组 k次最长重复子串（可重叠）)

题意：给定一个字符串，求至少出现k 次的最长重复子串，这k 个子串可以重叠。

思路：仍是二分出长度再以长度把height数组分类，再判是否存在k次

因为只有20000个数，但是数的范围是1000000 所以先离散化后基数排序效率将大大提高。

//Accepted	860K	47MS
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

int n,k;
const int MAXN = 20010;
int t1[MAXN],t2[MAXN],c[MAXN];
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;
        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;
        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];
        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 rank[MAXN],height[MAXN];

int r[MAXN],sa[MAXN];
struct node
{
    int tab,val;
}a[MAXN];
bool cmp2(const node &a,const node &b)
{
    return a.val < b.val;
}
bool ok(int x)
{
    int cnt=1;
    for(int i=2;i<=n;i++)
    {
        if(height[i]>=x) cnt++;
        else cnt=1;
        if(cnt>=k) return true;
    }
    return false;
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=0;i<n;i++) a[i].tab=i;
        for(int i=0;i<n;i++) scanf("%d",&a[i].val);
        sort(a,a+n,cmp2);
        int l=0;
        r[a[0].tab]=++l;
        for(int i=1;i<n;i++)//离散化
            if(a[i].val!=a[i-1].val) r[a[i].tab]=++l;
            else r[a[i].tab]=l;
        r[n]=0;
        da(r,sa,rank,height,n,MAXN);
        int lb=0,ub=n+1;
        while(ub-lb>1)
        {
            int mid=(lb+ub)>>1;
            if(ok(mid)) lb=mid;
            else ub = mid;
        }
        printf("%d\n",lb);
    }
    return 0;
}