POJ - 3261 Milk Patterns（后缀数组+二分 求出现次数>=k的最长子串）

题目：找一个串中出现次数多于k的最长子串。

数据好像有些大，先离散一下。之后二分长度就可以了

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
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 ra[],int height[],int n,int m)
{
	n++;
	int 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++) ra[sa[i]]=i;
	for(int i=0;i<n;i++)
	{
		if(k) k--;
		int j=sa[ra[i]-1];
		while(str[i+k]==str[j+k])k++;
		height[ra[i]]=k;
	}
}

int m,n,k,t;
int ra[maxn],height[maxn],str[maxn],sa[maxn];
int a[maxn];

int check(int len)
{
    int ans=1;
    for(int i=2;i<=n;i++)
    {
        if(height[i]>=len)
            ans++;
        else ans=1;
        if(ans>=k)
            return 1;
    }
    return 0;
}


int main()
{
   //freopen("in.txt","r",stdin);
    while(~scanf("%d%d",&n,&k))
    {
        if(n==0)break;
        memset(str,0,sizeof str);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&str[i]);
            a[i]=str[i];
        }

        sort(a,a+n);
        int o=unique(a,a+n)-a;
        for(int i=0;i<n;i++)
            str[i]=lower_bound(a,a+o,str[i])-a+1;
        /*for(int i=0;i<n;i++)
        printf("%d    ",str[i]);
        puts("");*/
        da(str,sa,ra,height,n,10000);

        int l=1,r=n,ans=0;
        while(l<=r)
        {
            int mid=(l+r)>>1;
            if(check(mid))
            {
                ans=mid;
                l=mid+1;
            }
            else r=mid-1;
        }
        printf("%d\n",ans);
    }
	return 0;
}