您的位置:首页 > 运维架构

后缀数组(dc3 +RMQ线段树维护模板)

2014-05-20 09:31 423 查看
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
#define maxn 200100
using namespace std;
int r[maxn];
int Rank[maxn],sa[maxn],height[maxn];
int wa[maxn],wb[maxn],wv[maxn],ws[maxn];
char a[maxn],b[maxn];
int cmp(int *r,int a,int b,int le)
{
return r[a]==r[b]&&r[a+le]==r[b+le];
}
void da(int *r,int *sa,int n,int m)
{
int i,j,p,*x=wa,*y=wb,*t;
for ( i = 0; i < m; i++) ws[i]=0;
for ( i = 0; i < n; i++) ws[ x[i] = r[i] ]++;
for ( i = 1; i < m; i++) ws[i]+=ws[i-1];
for ( i = n-1; i >= 0; i--) sa[--ws[x[i]]]=i;
for ( j = 1,p=1; p <n ; j*=2,m=p)
{
for ( p = 0,i=n-j; i < n; i++) y[p++]=i;
for ( i = 0; i < n; i++) if(sa[i]>=j) y[p++]=sa[i]-j;
for ( i = 0; i < n; i++) wv[i]=x[y[i]];
for ( i = 0; i < m; i++) ws[i]=0;
for ( i = 0; i < n; i++) ws[wv[i]]++;
for ( i = 1; i < m; i++) ws[i]+=ws[i-1];
for ( i = n-1; i >= 0 ; i--) sa[--ws[wv[i]]]=y[i];
for (t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
x[sa[i]] = cmp(y,sa[i-1],sa[i],j)?p-1:p++;
}
return ;
}
void calheight( int *r,int *sa,int n)
{
int i,j,k=0;
for ( i = 0; i <=n ; i++) Rank[sa[i]]=i;
for(i=0;i<n;height[Rank[i++]]=k)
for(k?k--:0,j=sa[Rank[i]-1];r[i+k]==r[j+k];k++);
return ;
}
struct pi
{
int min;
}pp[4*maxn];
void build(int le,int ri,int tot)
{
if(le==ri)
{
pp[tot].min=height[le];
return ;
}
int mid;
mid=(le+ri)/2;
build(le,mid,2*tot);
build(mid+1,ri,2*tot+1);
pp[tot].min=min(pp[2*tot].min,pp[2*tot+1].min);
return ;
}
int query(int le,int ri,int tot,int ll,int rr)
{
int p,q;
p=100000000;
q=100000000;
if(le>ri)
return 0;
if(le<=ll&&ri>=rr)
{
return pp[tot].min;
}
int mid;
mid=(ll+rr)/2;
if(le<=mid)
{
p=query(le,ri,2*tot,ll,mid);
}
if(ri>mid)
{
q=query(le,ri,2*tot+1,mid+1,rr);
}
if(p==100000000&&q==100000000)
return 0;
return min(p,q);
}
int main()
{
int i,n,k,f,le,ri,mid,p;
while(scanf("%d%d",&n,&k)!=EOF)
{
for(i=0;i<n;i++)
{
scanf("%d",&f);
r[i]=f+1;
}
r
=0;//为了使rank从1开始,防止height[rank[i]-1]越界。
da(r,sa,n+1,20002);
calheight(r,sa,n);
for(i=2;i<=n;i++)
height[i];      //height[i]数组的定义sa[i-1]和sa[i]的最长公共前缀,而rank从1到n,所以height数组从2到n。
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: