后缀数组+lcp+RMQ板子

3 篇文章 0 订阅
2 篇文章 0 订阅
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector>
using namespace std;
#define ll long long
#define pb push_back
#define mk make_pair
#define INF 0x3f3f3f3f
const int maxn = 1e5 + 5;

int t1[maxn],t2[maxn],c[maxn];

bool cmp(int *r, int a,int b, int l) {
    return r[a] == r[b] &&r[a+l] == r[b+l];
}

void da(int str[],int sa[],int Rank[],int lcp[],int n,int m) {//get sa数组
    ++n;
    int i,j,p,*x = t1,*y = t2;
    for(i=0;i<m;i++) c[i] = 0;
    for(i=0;i<n;i++) c[x[i] = str[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(j=1;j<=n;j<<=1) {
        p=0;
        for(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<m;i++) c[i]=0;
        for(i=0;i<n;i++) c[x[y[i]]]++;
        for(i=1;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] ] = cmp(y,sa[i-1],sa[i],j) ? p-1:p++;
        }
        if(p>=n)break;
        m=p;
    }
    int k = 0;
    n--;
    for(i=0;i<=n;i++) Rank[sa[i]]=i;//lcp
    for(i=0;i<n;i++) {
        if(k)--k;
        j = sa[Rank[i]-1];
        while(str[i+k] == str[j+k]) ++k;
        lcp[Rank[i]] = k;
    }
}

int lcp[maxn],a[maxn],sa[maxn],Rank[maxn];

char s[maxn];

int d[maxn][40];
int len;

void rmq_init(int *A, int n) {//RMQ
    for(int i=0;i<n;i++) d[i][0]=A[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]);
        }
    }
}

int ASK(int l,int r) {
    int k=0;
    while((1<<(k+1)) <= r-l+1) ++k;
    return min(d[l][k],d[r-(1<<k)+1][k] ) ;
}

int ask(int l,int r) {
    if(l==r) return len-sa[r];
    return ASK(l+1,r);
}

int main(){  
    scanf("%s",s);
    len = strlen(s);
    for(int i=0;i<len;i++) {
        a[i] = s[i] - 'a' +1;
    }
    a[len] = 0;
    da(a,sa,Rank,lcp,len,30);
    rmq_init(lcp,len+1);
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值