POJ3261 Milk Patterns

思路

后缀数组+二分答案的裸题
二分长度mid，然后如果至少有k个连续的height>=mid，就是true
否则就是false

代码

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
struct Node{
    int pos,r[2];
}midx[100000],x[100000];
int barrel[100000],sa[100000],ranks[100000],height[100000],n,k,s[100000];
int c_sort(int n,int lim){
    for(int i=0;i<2;i++){
        memset(barrel,0,sizeof(barrel));
        for(int j=1;j<=n;j++)
            barrel[x[j].r[i]]++;
        for(int j=1;j<=lim;j++)
            barrel[j]+=barrel[j-1];
        for(int j=n;j>=1;j--)
            midx[barrel[x[j].r[i]]--]=x[j];
        for(int j=1;j<=n;j++)
            x[j]=midx[j];
    }
    ranks[x[1].pos]=1;
    int cnt=1;
    for(int i=2;i<=n;i++){
        if(x[i].r[0]==x[i-1].r[0]&&x[i].r[1]==x[i-1].r[1])
            ranks[x[i].pos]=cnt;
        else
            ranks[x[i].pos]=++cnt;
    }
    return cnt;
}
void get_sa(int n){
    for(int i=1;i<=n;i++){
        x[i]=(Node){i,0,s[i]};
    }
    int cnt=c_sort(n,255);
    for(int i=1;cnt<n;i<<=1){
        for(int j=1;j<=n;j++)
            x[j]=(Node){j,(i+j<=n)?ranks[i+j]:0,ranks[j]};
        cnt=c_sort(n,cnt);
    }
    for(int i=1;i<=n;i++)
        sa[ranks[i]]=i;
    for(int i=1,j=0,k;i<=n;height[ranks[i++]]=j)
        for((j)?j--:0,k=sa[ranks[i]-1];s[i+j]==s[j+k];j++);
}
bool check(int x){
    int t=1,ans=1;
    for(int i=2;i<=n;i++){
        if(height[i]>=x){
            t++;
            if(t>=k)
                return true;
        }
        else{
            t=1;
        }
    }    // printf("x=%d ans=%d
",x,ans);
    return false;
}
int main(){
    scanf("%d %d",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i]);
    }
    get_sa(n);
    int ans=0,l=0,r=n;
    while(l<=r){
        int mid=(l+r)>>1;
        if(check(mid))
            ans=mid,l=mid+1;
        else
            r=mid-1;
    }
    printf("%d
",ans);
    return 0;
}