后缀数组小模板 POJ 2774

本文介绍了一个模板题的解决方案,该题目涉及字符串处理,通过将两个字符串连接并构建后缀数组来解决。文章提供了完整的C++代码实现,并使用了height数组来辅助计算最长公共前缀。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

模板题,把俩串连起来求后缀数组,通过height数组解 当模板用咯


#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#define Min(a, b) ((a) < (b)) ? (a) : (b)
#define Max(a, b) ((a) > (b)) ? (a) : (b)
using namespace std;

const int maxn = 200000 + 7;

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

int wa[maxn], wb[maxn], WS[maxn], wv[maxn];
int Rank[maxn], height[maxn];

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++;
    }
}

void calheight(int *r, int *sa, int n) {
    int i, j, k = 0;
    for (i = 1; 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++);
}

int sa[maxn];
int r[maxn];

char s[maxn], s1[maxn];

int main() {
    scanf("%s", s);
    scanf("%s", s1);
    int len1 = strlen(s), len2 = strlen(s1);
    for (int i = 0; i < len2; ++i) s[len1 + i + 1] = s1[i];
    s[len1] = 128;
    int len = len1 + len2 + 1;
    int n = len + 1;
    for (int i = 0; i < len; ++i) r[i] = s[i];
    r[len1] = 128;
    r[len] = 128;
    DA(r, sa, n + 1, 265);
    calheight(r, sa, n);
    int ans = 0;
    for (int i = 1; i < n; ++i) {
        if (ans < height[i]) {
            int x = Min(sa[i], sa[i - 1]);
            int y = Max(sa[i], sa[i - 1]);
            if (x < len1 && y > len1) ans = height[i];
        }
    }
    printf("%d\n", ans);
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值