POJ - 3261：Milk Patterns （后缀数组之求至少重复k次的子串长度）

题目大意：

给你一个串，以及整数k，求这个串中至少重复k次的子串的长度。

解题思路：

二分二分二分！！！

后缀数组，一言不合就二分。

直接二分长度，然后判断当前长度是否合法即可。

Ac代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <algorithm>
#include <set>
#include <functional>
#define next ne
#define rank ra
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
const int INF=1e9+7;
char s[maxn];
int n,k,a[maxn];
vector<int> v;

int getid(int x) { return lower_bound(v.begin(),v.end(),x)-v.begin()+1; }

int sa[maxn],rank[maxn],height[maxn];
int t1[maxn],t2[maxn],r[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(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=str[i]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(int j=1;j<=n;j<<=1)
    {
        p=0;
        for(int i=n-j;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1,x[sa[0]]=0;
        for(int 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(int i=0;i<=n;i++) rank[sa[i]]=i;
    for(int 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;
    }
}
bool check(int lt)
{
    int cnt=1;
    for(int i=2;i<=n;i++)
    {
        if(height[i]>=lt)   //满足要求 cnt++ 注意cnt初值为1
        {
            cnt++;
            if(cnt>=k) return 1;
        }
        else cnt=1;
    }
    if(cnt>=k) return 1;
    return 0;
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]),v.push_back(a[i]);    //离散一下
    sort(v.begin(),v.end()),v.erase(unique(v.begin(),v.end()),v.end());
    for(int i=0;i<n;i++) r[i]=getid(a[i]);
    r[n]=0;
    da(r,sa,rank,height,n,v.size()+5);
    int L=0,R=n,res=0;
    while(L<=R) //二分长度
    {
        int mid=(L+R)>>1;
        if(check(mid)) res=mid,L=mid+1;
        else R=mid-1;
    }
    printf("%d\n",res);
    //system("pause");
}