POJ 3261 Milk Patterns 后缀数组+二分

同样在论文中出现过的问题。至少出现k次的最长重复子串，可以重叠。

这个题要离散化一下，数据比较大...

#include"cstdlib"
#include"cstdio"
#include"cstring"
#include"cmath"
#include"queue"
#include"algorithm"
#include"iostream"
#include"map"
using namespace std;
#define N 200012
int sa[N],ra[N],height[N];
int t1[N],t2[N],c[N];
int a[N];
int v[N];
int cmp(int *r,int a,int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void getsa(int s[],int n,int m)
{
    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]=s[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];
        for(swap(x,y),p=1,i=1,x[sa[0]]=0; i<n; i++) x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
        if(p>=n)break;
        m=p;
    }
}
void getheight(int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) ra[sa[i]]=i;
    for(i=0;i<n;i++)
    {
        if(k) k--;
        j=sa[ra[i]-1];
        while(v[i+k]==v[j+k]) k++;
        height[ra[i]]=k;
    }
}
bool check(int n,int k,int x)
{
    int sum=1;
    for(int i=1;i<=n;i++)
    {
        if(height[i]>=x)
        {
            sum++;
            if(sum>=k) return true;
        }
        else sum=1;
    }
    return false;
}
int main()
{
    int n,k;
    while(scanf("%d%d",&n,&k)!=-1)
    {
        int m=0,c=2;
        map<int,int>mark;
        while(n--)
        {
            int x;
            scanf("%d",&x);
            if(!mark[x]) mark[x]=c++;

            v[m++]=mark[x];
        }
        v[m]=0;
        getsa(v,m+1,c);
        getheight(m);
        int l=0,r=m;
        int ans=0;
        while(l<=r)
        {
            int mid=(l+r)/2;
            if(check(m,k,mid))
            {
                ans=mid;
                l=mid+1;
            }
            else r=mid-1;
        }
        printf("%d\n",ans);
    }
    return 0;
}