后缀数组!!!

acwing五十分钟看四遍

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int n,m;
char s[N];
int x[N],sa[N],c[N],y[N],rk[N],height[N];
void get_sa()
{
    for(int i=1;i<=n;i++) c[x[i]=s[i]]++;
   
    for(int i=2;i<=m;i++) c[i]+=c[i-1];
    for(int i=n;i>0;i--) sa[c[x[i]]--]=i;//从后往前赋值,并不是指按大小顺序从后往前,而是按照原本s的倒序
    for(int k=1;k<=n;k<<=1)
    {
        int num=0;
        for(int i=n-k+1;i<=n;i++) y[++num]=i;//y记录的是只按第二关键字排序的顺序
        for(int i=1;i<=n;i++)
        if(sa[i]>k) y[++num]=sa[i]-k;
        //再重新进行一遍按第一关键字赋值c
        for(int i=1;i<=m;i++) c[i]=0;
        for(int i=1;i<=n;i++) c[x[i]]++;
        for(int i=2;i<=m;i++) c[i]+=c[i-1];
        //再按第一第二关键字排序
        for(int i=n;i>0;i--) sa[c[x[y[i]]]--]=y[i],y[i]=0;//此处还应把y[i]赋为0
        swap(x,y);//这下y记录的就是按照第一关键字离散化后的值,sa记录按双关键字排序
        x[sa[1]]=1,num=1;
        for(int i=2;i<=n;i++) x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?   num:++num;
        //此刻x[i]记录的是按照第一第二两个关键字排序后以第i位开头的后缀的排序,直接就可用于下一轮倍增
        if(num==n) break;
        m=num;//更新离散后的rank范围,一开始按ASCII码离散,但是一轮过后是按有多少个后缀离散的
    }
}
void get_height()//h[i]表示第i个后缀与sa[rk[i]-1]的最长公共前缀,有性质h[i]>=h[i-1]用来优化
{
    for(int i=1;i<=n;i++) rk[sa[i]]=i;
    for(int i=1,k=0;i<=n;i++)//从小到大枚举并且计算每个后缀
    {
        if(rk[i]==1) continue;
        if(k) k--;
        int j=sa[rk[i]-1];
        while(j+k<=n&&i+k<=n&&s[j+k]==s[i+k]) k++;
        height[rk[i]]=k;
    }
}
int main()
{
    cin>>s+1;
    n=strlen(s+1);
    m=122;
    get_sa();
    get_height();
    for(int i=1;i<=n;i++)
    cout<<sa[i]<<' ';
    cout<<"\n";
    for(int i=1;i<=n;i++)
    cout<<height[i]<<' ';
   // cout<<"\n";
    //cout<<s+1<<endl;
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值