BZOJ1717: [Usaco2006 Dec]Milk Patterns 产奶的模式

题目链接

离散化后后缀数组。若将模式串看为最长公共前缀，即是求原串中K个后缀的最长公共前缀。易知若要前缀最长，则K个串的首位置的rank值必定连续，即这k个串的首位置为sa[i]~sa[i+k-1] (i+k-1 < n)。所以即是求连续的K-1个height值中最小值的最大值。可以用单调队列维护。

离散化从0开始的求height过程中要加判断。。超过原串长的部分不可算。。拍出来还调了半天。。

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#define N 20005
#define M 10000005
#define INF 1e9
using namespace std;
typedef unsigned int ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,K,ans;
int sa[N],t[N],t2[N],c[N],rank[N],hei[N];
int num[N],mp[1000005];
char s[N];

void Build_Sa()
{
    int *x=t,*y=t2;
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=s[i]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(int k=1;k<=n;k<<=1)
    {
        int p=0;
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        p=1;swap(x,y);x[sa[0]]=0;
        for(int i=1;i<n;i++)
            x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
        if(p>=n) break;m=p;
    }
}

void Get_Height()
{
    for(int i=0;i<n;i++) rank[sa[i]]=i;
    int k=0;
    for(int i=0;i<n;i++)
    {
        if(!rank[i]) continue;
        if(k) k--;
        int j=sa[rank[i]-1];
        while(s[i+k]==s[j+k]&&i+k<n&&j+k<n) k++;
        hei[rank[i]]=k;
    }
}

int q[N];
void Humdrum_Queue()
{
    int head=1,tail=0;
    for(int i=0;i<K-1;i++) 
    {
        while(head<=tail&&hei[q[tail]]>=hei[i]) tail--;
        q[++tail]=i;
    }
    for(int i=K-1;i<n;i++)
    {
        while(head<=tail&&hei[q[tail]]>=hei[i]) 
            tail--;
        q[++tail]=i;    
        while(q[head]<i-K+1) head++;
        ans=max(ans,hei[q[head]]);
    }
}

int main()
{
    n=read();K=read()-1;
    for(int i=0;i<n;i++) num[i]=s[i]=read();
    sort(num,num+n);
    for(int i=0;i<n;i++) if(i==0||num[i]!=num[i-1]) mp[num[i]]=m++;
    for(int i=0;i<n;i++) s[i]=mp[s[i]];
    Build_Sa();Get_Height();
    Humdrum_Queue();
    printf("%d\n",ans);
    return 0;
}