POJ 3261 后缀数组板子

题目链接：http://poj.org/problem?id=3261

 

题意：

       给你n个数字，要你找出出现超过k次子串的最长长度。

做法：

       后缀数组的板题吧。因为听说比较常用就先留个板子。东西还是比较好理解的。这个题也不是很难。有比较难的我晚点再理解。

       总之就是用一段连续的代码实现了对后缀串的排序。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
typedef long long ll;
using namespace std;
const int maxn=100010;
//以下为倍增算法求后缀数组
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];}
/**< 传入参数：str,sa,len+1,ASCII_MAX+1 */
void da(const 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]]++;//以字符的ascii码为下标
      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],k,len;
//求height数组
/**< str,sa,len */
void calheight(const int *s,int *sa,int n)
{
    int i,j,k=0;
    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(s[i+k]==s[j+k])k++;
        height[Rank[i]]=k;
    }
}
//height[i]表示，第i与第i-1小的后缀的公共前缀的长度
//sa[i]表示，第i小的后缀是从哪个字母开始的
//rank[i]表示，第i个字母开始的后缀为第几
//注意！输入从第0开始，但是height从1~len
int str[maxn];
int ck(int mi){
    int now=1;
    for(int i=2;i<=len;i++){
        if(height[i]>=mi){
            if(++now>=k) return 1;
        }
        else now=1;
    }
    return 0;
}
int main(){
    scanf("%d%d",&len,&k);
    int ma=-1;
    for(int i=0;i<len;i++){
        scanf("%d",str+i);
        str[i]++;
        ma=max(ma,str[i]);
    }
    str[len]=0;
    da(str,sa,len+1,ma+1);
    calheight(str,sa,len);
    int l=0,r=len+1,ans=0;
    while(l<=r){
        int mid=(l+r)>>1;
        if(ck(mid)) {
            ans=mid; l=mid+1;
        }
        else r=mid-1;
    }
    printf("%d\n",ans);
      return 0;
}