1. 程式人生 > >HDU_6194 後綴數組+RMQ

HDU_6194 後綴數組+RMQ

相關 但是 style const eof 公共前綴 處理 ++ pan

好絕望的。。想了五個多小時,最後還是沒A。。。賽後看了下後綴數組瞬間就有了思路。。。不過因為太菜,想了將近兩個小時才吧這個題幹掉。

首先,應當認為,後綴數組的定義是,某字符串S的所有後綴按照字典序有小到大的順序排列(使用下標表示後綴)。因為具體過程沒太看懂,但是參見劉汝佳藍書《算法競賽黑暗聖典》可以得到一個聰明的NLOGN的神器算法。不過這個不太重要。

之後還可以通過他在LCP問題中提到的RANK,height數組相關算法,處理出來height數組,之後其他的可以扔掉。

《黑暗聖典》中定義了height數組,height[k]的含義是,第rank[i]數組和rank[i]-1之間的最長公共前綴的長度。。。基於這個我們可以看到一些規則。

首先height數組的定義具有傳遞性,很容易想到的就是。。。出現且僅出現M次可以被簡單的定義為,傳遞且僅能夠被傳遞M次。。。

於是。。。我們有了如下算法.

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
const long long MAXN=400233;
const long long INF=1E8+7;
char s[MAXN];
//long long dp[MAXN];
long long sa[MAXN],t[MAXN],t2[MAXN],c[MAXN],n,len;
long long r1ank[MAXN],height[MAXN]; long long d[MAXN][30]; void RMQ_init() { for(int i=0;i<n;++i)d[i][0]=height[i]; for(int j=1;(1<<j)<=n;j++) for(int i=0;i+(1<<j)-1<n;i++) d[i][j]=min(d[i][j-1],d[i+(1<<(j-1))][j-1]); } long long RMQ(int a,int b) {
int k=0; while((1<<(k+1))<=b-a+1)k++; return min(d[a][k],d[b-(1<<k)+1][k]); } int m; void build_sa(int m) { long long i,*x=t,*y=t2; for( i=0;i<m;++i)c[i]=0; for( i=0;i<n;++i)c[x[i]=s[i]]++; for( i=1;i<m;++i)c[i]+=c[i-1]; for( i=n-1;i>=0;i--)sa[--c[x[i]]]=i; for( int k=1;k<= n ;k*=2) { int p=0; for(i=n-k;i<n;++i)y[p++]=i; for(i=0;i<n;++i)if(sa[i]>=k)y[p++]=sa[i]-k; for( i=0;i<m;++i)c[i]=0; for( i=0;i<n;++i)c[x[y[i]]]++; for( i=0;i<m;++i)c[i]+=c[i-1]; for( i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i]; swap(x,y); p=1;x[sa[0]]=0; for( i=1;i<n;++i) { x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++; }if(p>=n)break; m=p; } } void getHeight() { int i,j,k=0; for(int i=0;i<n;++i)r1ank[sa[i]]=i; for(int i=0;i<n;++i) { if(k)k--; int j=sa[r1ank[i]-1]; while(s[j+k]==s[i+k])k++; height[r1ank[i]]=k; } height[n]=0; } int main() { int t;cin>>t; while(t--) { // memset(height,0,sizeof(height)); scanf("%d %s",&m,s); n=strlen(s); len=n;n+=1;; build_sa(128); getHeight(); RMQ_init();long long ans=0; for(int i=0;i<n;++i) { if(m==1) { ans+=len-sa[i]-max(height[i],height[i+1]); continue; } int a=i+1;int b=i+m-1; long long limit=RMQ(a,b); if(n>b&&limit>max(height[i],height[b+1]))ans+=limit-max(height[i],height[b+1]); } cout<<ans<<"\n"; } }

HDU_6194 後綴數組+RMQ