POJ3261（后缀数组+二分）

题目

总是被奇怪的错误给缠住，无谓地浪费宝贵的时间，什么时候才能一遍AC啊啊啊
求可重叠的k次最长重复子串

题解

二分答案，将后缀字符串按height数组分组，满足条件且组内元素个数>=k个，即为可行解

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
#define N 40000
int t1[N],t2[N],sa[N],h[N],rk[N],c[10*N],a[N];
char s[N];
int m,n,k;
void calcsa(int n,int m){
    int *x=t1,*y=t2,p=0,f=0;
    for(int i=1;i<=m;i++)c[i]=0;
    for(int i=1;i<=n;i++)c[x[i]=a[i]]++;
    for(int i=1;i<=m;i++)c[i]+=c[i-1];
    for(int i=n;i;i--)sa[c[x[i]]--]=i;
    for(int i=1;i<=n&&p<=n;i<<=1){
        p=0;
        for(int j=n-i+1;j<=n;j++)y[++p]=j;
        for(int j=1;j<=n;j++)if(sa[j]>i)y[++p]=sa[j]-i;
        for(int j=1;j<=m;j++)c[j]=0;
        for(int j=1;j<=n;j++)c[x[y[j]]]++;
        for(int i=1;i<=m;i++)c[i]+=c[i-1];
        for(int j=n;j;j--)sa[c[x[y[j]]]--]=y[j];
        swap(x,y);x[sa[1]]=1;p=2;
        for(int j=2;j<=n;j++)
        x[sa[j]]=y[sa[j]]==y[sa[j-1]]&&y[sa[j]+i]==y[sa[j-1]+i]?p-1:p++;
        m=p;
    }
    for(int i=1;i<=n;i++)rk[sa[i]]=i;
    for(int i=1;i<=n;i++){
        int j=sa[rk[i]-1];
        if(f)f--;while(a[i+f]==a[j+f])f++;
        h[rk[i]]=f;
    }
}
int panduan(int mid)
{
	int ans1=0;
	for(int i=1;i<=n;i++)
	{
		//cout<<i<<' '<<ans1<<endl;
		if(h[i]<mid)
		{
			ans1=0;
		}
		else
		{
			ans1++;
		}
		if(ans1>=k-1)
		{
			return 1;
		}
	}
	return 0;
}
int main()
{
	cin>>n>>k;
	int x1,maxn=0;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&x1);
		a[i]=x1+1;
		maxn=max(maxn,a[i]);
	}
	calcsa(n,maxn+1);
	/*for(int i=1;i<=n;i++)
	{
		cout<<h[i]<<' ';
	}
	cout<<endl;
	for(int i=1;i<=n;i++)
	{
		for(int j=sa[i];j<=n;j++)
		{
			cout<<a[j]-1;
		}
		cout<<endl;
	}
	cout<<endl;*/
	int lf=0,rg=n;
	while(lf<rg)
	{
		int mid=(lf+rg+1)>>1;
		if(panduan(mid))
		{
			lf=mid;
		}
		else
		{
			rg=mid-1;
		}
	}
	cout<<lf<<endl;
}