您的位置:首页 > 其它

【模板】后缀数组

2013-02-20 23:58 423 查看
串处理问题神器,RMQ预处理后能O(1)回答俩后缀的LCP。

另外有一种常见处理方法:二分LCP长度H,将sa分块,则每一块内的后缀同且仅同当前块内的另一后缀LCP>=H。

还有一种常见思考方法:考虑贡献值。

Code:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int SN=100010;

int n,s[SN],sa[SN],c[SN],rank[SN],hei[SN],h[SN],lg[SN],rmq[SN][30],ta0[SN],ta1[SN];

bool Cmp(int i, int j) { return (i<=n && j<=n && s[i]==s[j]); }
void SufArr()
{
int i,j,k,p,m=SN, *x=ta0,*y=ta1;
for(i=1; i<=m; i++) c[i]=0;
for(i=1; i<=n; i++) c[x[i]=s[i]]++;
for(i=1; i<=m; i++) c[i]+=c[i-1];
for(i=n; i>=1; i--) sa[c[x[i]]--]=i;
for(k=1,p=0; ; k<<=1,p=0)
{
for(i=n-k+1; i<=n; i++) y[++p]=i;
for(i=1; i<=n; i++) if(sa[i]-k>0) y[++p]=sa[i]-k;
for(i=1; i<=m; i++) c[i]=0;
for(i=1; i<=n; i++) c[x[y[i]]]++;
for(i=1; i<=m; i++) c[i]+=c[i-1];
for(i=n; i>=1; i--) sa[c[x[y[i]]]--]=y[i];
swap(x,y);
for(i=2,x[sa[1]]=p=1; i<=n; i++)
x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k])? p: ++p;
if(p==n) break;  else m=p;
}
for(i=1; i<=n; i++) rank[sa[i]]=i;

for(i=1,j=0; i<=n; i++) lg[i]=( (1<<j) >= (i+1)/2 )? j: ++j;
for(i=1,h[0]=1; i<=n; i++)
if(rank[i]==1) h[i]=hei[rank[i]]=0;
else { for(k=Max(h[i-1],1); Cmp(i+k-1, sa[rank[i]-1]+k-1); k++); h[i]=hei[rank[i]]=k-1; }
}
void RMQ()
{
int i,k,up=Lg(SN);
for(i=1; i<=n; i++) rmq[i][0]=hei[i];
for(k=1; k<=up; k++)
for(i=1; i+(1<<k)-1<=n; i++)
rmq[i][k]=Min(rmq[i][k-1], rmq[i+(1<<(k-1))][k-1]);
}
int Ask(int l, int r)
{
int len,tmp;
l=rank[l], r=rank[r]; if(l>r) tmp=l,l=r,r=tmp; l++;
len=r+1-l, tmp=lg[len]-1, tmp=(tmp)?tmp:1;
return Min(rmq[l][tmp], rmq[r-(1<<tmp)+1][tmp]);
}

int main()
{
freopen("oj.in", "r", stdin);
freopen("oj.out", "w", stdout);
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: