POJ 3261 Milk Patterns (离散化+后缀数组 k次最长重复子串（可重叠）)

来源：互联网发布：月子餐30天食谱知乎编辑：程序博客网时间：2024/05/21 05:37

题意：给定一个字符串，求至少出现k 次的最长重复子串，这k 个子串可以重叠。

思路：仍是二分出长度再以长度把height数组分类，再判是否存在k次

因为只有20000个数，但是数的范围是1000000 所以先离散化后基数排序效率将大大提高。

//Accepted860K47MS#include<cstdio>#include<iostream>#include<cstring>#include<algorithm>using namespace std;int n,k;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 rank[],int height[],int n,int m){    n++;    int i,j,p,*x=t1,*y=t2;    for(i=0;i<m;i++) c[i]=0;    for(i=0;i<n;i++) c[x[i]=str[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(j=1;j<=n;j<<=1)    {        p=0;        for(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<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);        p=1;        x[sa[0]]=0;        for(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(i=0;i<=n;i++) rank[sa[i]]=i;    for(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;    }}int rank[MAXN],height[MAXN];int r[MAXN],sa[MAXN];struct node{    int tab,val;}a[MAXN];bool cmp2(const node &a,const node &b){    return a.val < b.val;}bool ok(int x){    int cnt=1;    for(int i=2;i<=n;i++)    {        if(height[i]>=x) cnt++;        else cnt=1;        if(cnt>=k) return true;    }    return false;}int main(){    while(~scanf("%d%d",&n,&k))    {        for(int i=0;i<n;i++) a[i].tab=i;        for(int i=0;i<n;i++) scanf("%d",&a[i].val);        sort(a,a+n,cmp2);        int l=0;        r[a[0].tab]=++l;        for(int i=1;i<n;i++)//离散化            if(a[i].val!=a[i-1].val) r[a[i].tab]=++l;            else r[a[i].tab]=l;        r[n]=0;        da(r,sa,rank,height,n,MAXN);        int lb=0,ub=n+1;        while(ub-lb>1)        {            int mid=(lb+ub)>>1;            if(ok(mid)) lb=mid;            else ub = mid;        }        printf("%d\n",lb);    }    return 0;}

0 0