bzoj 1717 [USACO06DEC] Milk Patterns (後綴數組+二分)

題目大意：求可重疊的相同子串數量至少是K的子串最長長度  

洛谷傳送門

依然是後綴數組+二分，先用後綴數組處理出height

每次二分出一個長度x，然後去驗證，在排序的後綴串集合裏，有沒有連續數量多於K個串的長度>=x，

但據說有一種高端做法是把二分換成單調隊列，能減少常數，可惜我並沒有看懂......

原題好像是哈希的騷操作，但網上的題解好像都是後綴數組......

比上一道男人八題簡單多了，我原來的錯代碼竟然卡過去了70分......

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 20010
#define inf 0x3f3f3f3f
#define maxn 1000010
using namespace std;

int n,len,tot,K;
int a[N],tr[N],sa[N],rk[N],hs[N],h[N];
struct node{int id,val,w;}d[N];
int cmp1(node s1,node s2){return s1.val<s2.val;}
int cmp2(node s1,node s2){return s1.id<s2.id;}
int gint()
{
    int rett=0,fh=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')fh=-1;c=getchar();}
    while(c>='0'&&c<='9'){rett=(rett<<3)+(rett<<1)+c-'0';c=getchar();}
    return rett*fh;
}
bool check(int k,int x,int y){
    if(x+k>len||y+k>len) return 0;
    else return (rk[x]==rk[y]&&rk[x+k]==rk[y+k])?1:0;
}
void get_sa()
{
    int cnt=0,i;
    for(i=1;i<=len;i++) hs[a[i]]++;
    for(i=1;i<=tot;i++) if(hs[i]) tr[i]=++cnt;
    for(i=1;i<=tot;i++) hs[i]+=hs[i-1];
    for(i=1;i<=len;i++) rk[i]=tr[a[i]],sa[hs[a[i]]--]=i;
    for(int k=1;cnt<len;k<<=1)
    {
        for(i=1;i<=cnt;i++) hs[i]=0;
        for(i=1;i<=len;i++) hs[rk[i]]++;
        for(i=1;i<=cnt;i++) hs[i]+=hs[i-1];
        for(i=len;i>=1;i--) if(sa[i]>k) tr[sa[i]-k]=hs[rk[sa[i]-k]]--;
        for(i=1;i<=k;i++) tr[len-i+1]=hs[rk[len-i+1]]--;
        for(i=1;i<=len;i++) sa[tr[i]]=i;
        for(i=1,cnt=0;i<=len;i++) tr[sa[i]]=check(k,sa[i],sa[i-1])?cnt:++cnt;
        for(i=1;i<=len;i++) rk[i]=tr[i];
    }
}
void get_height()
{ 
    for(int i=1;i<=n;i++){
        if(rk[i]==1) continue;
        for(int j=max(1,h[rk[i-1]]-1);;j++)
            if(a[i+j-1]==a[sa[rk[i]-1]+j-1]) h[rk[i]]=j;
            else break;
    }
}
void descrete()
{
    sort(d+1,d+n+1,cmp1);
    d[0].val=-1;
    for(int i=1;i<=n;i++)
        if(d[i].val==d[i-1].val)
            d[i].w=d[i-1].w;
        else d[i].w=++tot;
    sort(d+1,d+n+1,cmp2);
    for(int i=1;i<=n;i++)
        a[i]=d[i].w;
}
int check(int ans)
{
    int i,cnt=0;
    for(i=1;i<=n;)
    {
        if(h[i]<ans) {i++;continue;}
        for(cnt=0;h[i]>=ans&&i<=n;i++) 
            cnt++;
        if(cnt>=K-1) return 1;
    }return 0;
}

int main()
{
    scanf("%d%d",&n,&K);len=n;
    for(int i=1;i<=n;i++) 
        d[i].val=gint(),d[i].id=i;
    descrete();
    get_sa();
    get_height();
    int l=0,r=n,ans=0;
    while(l<=r){
        int mid=(l+r)>>1;
        if(check(mid)) ans=mid,l=mid+1;
        else r=mid-1;
    }printf("%d\n",ans);
    return 0;
}