poj3261最长重复k次子串

来源：互联网发布：dos命令和linux命令编辑：程序博客网时间：2024/05/21 19:36

http://poj.org/problem?id=3261

求最长重复k次的子串长度。

二分答案，变成判定性问题，按照连续的height值不小于k分组，如果存在一组满足k各height大于k就存在

#include<iostream>#include<cstdio>#include<cstring>>#include<string>#define maxn 1000010using namespace std;int wa[maxn],wb[maxn],wv[maxn],Ws[maxn];int cmp(int *r,int a,int b,int l){    return r[a]==r[b]&&r[a+l]==r[b+l];}void da(int *r,int *sa,int n,int m){    int i,j,p,*x=wa,*y=wb,*t;    for(i=0;i<m;i++) Ws[i]=0;    for(i=0;i<n;i++) Ws[x[i]=r[i]]++;    for(i=1;i<m;i++) Ws[i]+=Ws[i-1];    for(i=n-1;i>=0;i--) sa[--Ws[x[i]]]=i;    for(j=1,p=1;p<n;j*=2,m=p)    {        for(p=0,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<n;i++) wv[i]=x[y[i]];        for(i=0;i<m;i++) Ws[i]=0;        for(i=0;i<n;i++) Ws[wv[i]]++;        for(i=1;i<m;i++) Ws[i]+=Ws[i-1];        for(i=n-1;i>=0;i--) sa[--Ws[wv[i]]]=y[i];        for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)            x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;    }    return;}int sa[maxn],Rank[maxn],height[maxn];void calheight(int *r,int *sa,int n){    int i,j,k=0;    for(i=1;i<=n;i++) Rank[sa[i]]=i;    for(i=0;i<n;height[Rank[i++]]=k)        for(k?k--:0,j=sa[Rank[i]-1];r[i+k]==r[j+k];k++);    return;}int n,k;int b[maxn];int check(int m){    int flag=1;    for(int i=1;i<=n;i++)    {        if(height[i]>=m)        {            flag++;            if(flag>=k)            return 1;        }        else        flag=1;//一开始这里没写错了，只要一组中一个不满足，这组就废了    }    return 0;}int main(){    while(scanf("%d%d",&n,&k)!=-1)    {        for(int i=0;i<n;i++)        scanf("%d",&b[i]);        b[n]=0;//cout<<"etrh"<<endl;        da(b,sa,n+1,1000010);        calheight(b,sa,n);        int ans,low=0,high=n,mid;        while(low<=high)        {            mid=(low+high)/2;            if(check(mid))            {                ans=mid;                low=mid+1;            }            else                high=mid-1;        }        printf("%d\n",ans);    }    return 0;}

0 0