C - Milk Patterns POJ - 3261

https://cn.vjudge.net/contest/311974#problem/C 

总结：后缀数组；lcp（最长公共前缀）

sa[i]表示：排名i的串是在那个下标开始（就是是啥串）；

 height[i]表示：排名i和i-1的lcp;即等于LCP(i,i-1)

rk[i]表示：下标i开始的串排名是多少；

1. LCP(i,j)=LCP(j,i);
2. LCP(i,i)=len(sa[i])=n-sa[i]+1;
3. LCP(i,k)=min(LCP(i,j),LCP(j,k)) 对于任意1<=i<=j<=k<=n
4. LCP(i,k)=min(LCP(j,j-1)) 对于任意1<i<=j<=k<=n

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <queue>
#include <map>
#include <set>
#include <stack>
using namespace std;

#define sfi(x) scanf("%d",&x)
#define sfc(x) scanf("%c",x)
#define sfl(x) scanf("%lld",&x)
#define sfs(x) scanf("%s",x)

#define rint register int
#define pb push_back
#define fl() printf("flag!\n")
#define INF 0x3f3f3f3f
#define ll long long
#define mem(x,y) memset(x,y,sizeof(x))
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)

const int maxn=2e5+9;
const int mod=1e9+7;

int a[maxn];
int b[maxn];
int t1[maxn],t2[maxn],rk[maxn],sa[maxn],c[maxn],height[maxn];
int n,m;

void get_sa(int *s,int n,int m)
{
    int *x=t1,*y=t2;
    int i,k;
    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(k=1;k<=n;k<<=1)
    {
        int p=0;
        for(i=n-k;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        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);
        x[sa[0]]=0;
        p=1;
        for(i=1;i<n;i++)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]) ? p-1:p++;
        if(p>=n) break;
        m=p;
    }
}


void get_height(int *s,int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) rk[sa[i]]=i;
    for(i=0;i<n;i++)
    {
        if(k) --k;
        j=sa[rk[i]-1];
        while(s[i+k]==s[j+k]) k++;
        height[rk[i]]=k;
    }
}

bool ch(int mid,int n,int k)
{
    int num=1;
    int Max=0;
    for(int i=1;i<=n;i++)
    {
        if(height[i]>=mid)
        {
            num++;
            Max=max(Max,num);
        }
        else num=1;
    }
    return Max>=k;
}

int main()
{
    //FAST_IO;
    //freopen("input.txt","r",stdin);

    while(~sfi(n))
    {
        int k;
        sfi(k);
        for(int i=0;i<n;i++) sfi(a[i]),b[i]=a[i];
        sort(b,b+n);

        m=unique(b,b+n)-b;

        for(int i=0;i<n;i++)
        {
            a[i]=lower_bound(b,b+m,a[i])-b+1;
            //cout<<a[i]<<endl;
        }

        a[n]=0;
        get_sa(a,n+1,m+1);
        get_height(a,n);
        int l=0,r=20000;
        int ans=0;
        while(l<=r)
        {
            int mid=(l+r)>>1;
            if(ch(mid,n,k)) ans=mid,l=mid+1;
            else r=mid-1;
        }
        printf("%d\n",ans);
    }

    return 0;

}