luogu p2582(後綴數組)

傳送門c++

題意:

給你一個字符串\(str\),問出現次數爲\(k\)的最長的子串的長度。數組

分析:

首先咱們先將字符串\(str\)的全部後綴進行排序,並求出他們兩兩的\(height\)數組。spa

根據\(height\)數組的含義,\(height[i]=lcp(i,i-1)\),咱們知道,假若存在一個子串出現了k次,那麼一定存在一個連續的區間\([l,r],(r-l+1 \ge k-1)\),使得\(lcp(l,r) !=0\)。那麼咱們他們\(lcp\)中的最小值就是答案。所以咱們發現,咱們如今要求的是\(height\)數組中,長度至少爲\(k-1\)的最小值,並要使得最小值最大化。而這個顯然是一個經典的劃窗問題,咱們能夠經過單調隊列在\(\mathcal{O}(n)\)的時間複雜度中求出答案。debug

故總體的複雜度爲\(\mathcal{O}(nlogn)\)code

代碼:

#include <bits/stdc++.h>
#define maxn 20010
using namespace std;
int rk[maxn],sa[maxn],height[maxn],tmp[maxn],cnt[maxn],n,k;
int str[maxn],tot=0;
unordered_map<int,int>mp;
unordered_map<int,bool>vis;
void SA(int n,int m){
    int i,j,k;
    n++;
    for(i=0;i<n+5;i++) rk[i]=sa[i]=height[i]=tmp[i]=0;
    for(i=0;i<m;i++) cnt[i]=0;
    for(i=0;i<n;i++) cnt[rk[i]=mp[str[i]]]++;
    for(i=1;i<m;i++) cnt[i]+=cnt[i-1];
    for(i=0;i<n;i++) sa[--cnt[rk[i]]]=i;
    for(k=1;k<=n;k<<=1){
        for(i=0;i<n;i++){
            j=sa[i]-k;
            if(j<0) j+=n;
            tmp[cnt[rk[j]]++]=j;
        }
        sa[tmp[cnt[0]=0]]=j=0;
        for(i=1;i<n;i++){
            if(rk[tmp[i]]!=rk[tmp[i-1]]||rk[tmp[i]+k]!=rk[tmp[i-1]+k])
                cnt[++j]=i;
            sa[tmp[i]]=j;
        }
        memcpy(rk,sa,n*sizeof(int));
        memcpy(sa,tmp,n*sizeof(int));
        if(j>=n-1) break;
    }
    //get height[]
    i=0,k=0,height[0]=0;
    for(j=rk[0];i<n-1;i++,k++){
        while(~k&&mp[str[i]]!=mp[str[sa[j-1]+k]]){
            height[j]=k--;
            j=rk[sa[j]+1];
        }
    }
}
void debug(){
    //ababa
    // sa[1]=4,sa[2]=2,sa[3]=0,sa[4]=3,sa[5]=1
    // rk[0]=3,rk[1]=5,rk[2]=2,rk[3]=4,rk[4]=1
    for(int i=1;i<=n;i++){
        printf("sa[%d]=%d\n",i,sa[i]);
    }
    for(int i=0;i<n;i++){
        printf("rank[%d]=%d\n",i,rk[i]);
    }
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++){
        scanf("%d",&str[i]);
        if(!mp[str[i]]) mp[str[i]]=++tot;
    }
    SA(n,tot+1);
    deque<int>que;
    int res=0;
    for(int i=1;i<=n;i++){
        while(!que.empty()&&i-que.front()>=k-1) que.pop_front();
        while(!que.empty()&&height[que.back()]>=height[i]) que.pop_back();
        que.push_back(i);
        if(!que.empty()&&i>=k-1) res=max(res,height[que.front()]);
    }
    printf("%d\n",res);
    return 0;
}
相關文章
相關標籤/搜索