洛谷P2852 [USACO06DEC]牛奶模式Milk Patterns (Suffix Array)后缀数组

前面后缀数组的板子相信大家都看得出来
出现k次就相当于我们选择k个后缀,求LCP
对于后缀i,j(rank[i]<rank[j])
其LCP是 height(i+1~j)的最小值
所以答案一定在rank连续的k个后缀中
维护连续rank上k个后缀的LCP的最大值即可
这个用单调队列就可以了= =
理论上要离散化一下,虽然数据比较水,不离散也可以

/*
@Date    : 2019-07-18 21:34:03
@Author  : Adscn ([email protected])
@Link    : https://www.cnblogs.com/LLCSBlog
*/
#include<bits/stdc++.h>
using namespace std;
#define IL inline
#define RG register
#define gi getint()
#define gc getchar()
#define File(a) freopen(a".in","r",stdin);freopen(a".out","w",stdout)
IL int getint()
{
    RG int xi=0;
    RG char ch=gc;
    bool f=0;
    while(ch<'0'|ch>'9')ch=='-'?f=1:f,ch=gc;
    while(ch>='0'&ch<='9')xi=(xi<<1)+(xi<<3)+ch-48,ch=gc;
    return f?-xi:xi;
}
template<typename T>
IL void pi(T k,char ch=0)
{
    if(k<0)k=-k,putchar('-');
    if(k>=10)pi(k/10,0);
    putchar(k%10+'0');
    if(ch)putchar(ch);
}
const int N=2e4+7;
int a[N],b[N];
int n,k;
namespace SA{
    int rkx[N],rky[N],cnt[N],*key1=rkx,*key2=rky,sa[N],rank[N];
    int height[N];
    int m;
    inline void Qsort()
    {
        memset(cnt,0,sizeof cnt);
        for(int i=1;i<=n;++i)++cnt[key1[i]];
        for(int i=1;i<=m;++i)cnt[i]+=cnt[i-1];
        for(int i=n;i;--i)sa[cnt[key1[key2[i]]]--]=key2[i];
    }
/*
    Attention:
    sa -> address
    rank -> rank
    key1[i] ->key1[i]'s rank
    key2[i] ->key2(rank)[i]'s address
*/
    void getsa()
    {
        m=n;
        for(int i=1;i<=n;++i)key1[i]=a[i],key2[i]=i;
        Qsort();
        for(int k=1,q=0;q<n;k<<=1)
        {
            q=0;
            for(int i=n-k+1;i<=n;++i)key2[++q]=i;
            for(int i=1;i<=n;++i)if(sa[i]>k)key2[++q]=sa[i]-k;
            Qsort();
            swap(key1,key2);
            q=key1[sa[1]]=1;
            for(int i=2;i<=n;++i)
                key1[sa[i]]=(key2[sa[i-1]]==key2[sa[i]]&&
                    key2[sa[i-1]+k]==key2[sa[i]+k])?q:++q;
            m=q;
        }
        for(int i=1;i<=n;++i)rank[sa[i]]=i;
    }
    inline void getheight()
    {
        int k=0;
        for(int i=1;i<=n;++i)
        {
            if(k)--k;
            int s=sa[rank[i]-1];
            while(s+k<=n&&i+k<=n&&a[s+k]==a[i+k])++k;
            height[rank[i]]=k;
        }
    }
}
using namespace SA;
int main(void)
{
    #ifndef ONLINE_JUDGE
    File("file");
    #endif
    n=gi,k=gi;
    for(int i=1;i<=n;++i)b[i]=a[i]=gi;
    sort(b+1,b+n+1);
    int m=unique(b+1,b+n+1)-b-1;
    for(int i=1;i<=n;++i)a[i]=lower_bound(b+1,b+m+1,a[i])-b;
    getsa();
    getheight();
    static int Q[1000100];
    int ans=0;
    int head=0,tail=-1;
    --k;
    for(int i=1;i<=n;++i)
    {
        while(head<=tail&&Q[head]+k<=i)++head;
        while(head<=tail&&height[Q[tail]]>height[i])--tail;
        Q[++tail]=i;
        if(i>=k)ans=max(ans,height[Q[head]]);
    }
    printf("%d",ans);   
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/LLCSBlog/p/11211493.html