poj3261 Milk Patterns（后缀数组）

 

【题目链接】

 

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

 

【题意】

 

　　至少出现k次的可重叠最长子串。

 

【思路】

       二分长度+划分height，然后判断是否存在一组的数目不小于k即可。

       需要注意的是：用末尾添0的方法解决n==1时的RE问题，但是在can中需要忽略height[0]且考虑height[1..n]。

【代码】

#include<cstdio>
#include<cstring>
#include<iostream>
#define FOR(a,b,c) for(int a=(b);a<=(c);a++)
using namespace std;

const int maxn = 20000+10;
const int maxm = 1000000+10;

int s[maxn];
int sa[maxn],c[maxm],t[maxn],t2[maxn];

void build_sa(int m,int n) {
    int i,*x=t,*y=t2;
    for(i=0;i<m;i++) c[i]=0;
    for(i=0;i<n;i++) c[x[i]=s[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(int k=1;k<=n;k<<=1) {
        int p=0;
        for(i=n-k;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        
        for(i=0;i<m;i++) c[i]=0;
        for(i=0;i<n;i++) c[x[y[i]]]++;
        for(i=0;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]]=y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
        if(p>=n) break;
        m=p;
    }
}
int rank[maxn],height[maxn];
void getHeight(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[j+k]==s[i+k]) k++;
        height[rank[i]]=k;
    }
}

int n,k;

bool can(int limit) {
    int cnt=1;                //忽略height[0] 截至height[n]
    for(int i=2;i<=n;i++) {    
        if(height[i]>=limit) {
            if(++cnt>=k) return true;
        }
        else cnt=1;
    }
    return false;
}

int main() {
    scanf("%d%d",&n,&k);
    int up=0;
    FOR(i,0,n-1) {
         scanf("%d",&s[i]);
         s[i]++;
         up=max(up,s[i]);
    }
    s[n]=0;
    build_sa(up+1,n);
    getHeight(n);
    int L=0,R=n;
    while(L<R) {
        int M=L+(R-L+1)/2;
        if(can(M)) L=M;
        else R=M-1;
    }
    printf("%d
",L);
    return 0;
}