hihocoder 1415 后缀数组三·重复旋律3 (后缀数组 + 二分 经典问题)

时间限制:5000ms

单点时限:1000ms

内存限制:256MB

描述

小Hi平时的一大兴趣爱好就是演奏钢琴。我们知道一个音乐旋律被表示为长度为 N 的数构成的数列。小Hi在练习过很多曲子以后发现很多作品中的旋律有共同的部分。

旋律是一段连续的数列，如果同一段旋律在作品A和作品B中同时出现过，这段旋律就是A和B共同的部分，比如在abab 在 bababab 和 cabacababc 中都出现过。小Hi想知道两部作品的共同旋律最长是多少？

解题方法提示

输入

共两行。一行一个仅包含小写字母的字符串。字符串长度不超过 100000。

输出

一行一个整数，表示答案。

样例输入

abcdefg
abacabca

样例输出

3

题目链接：http://hihocoder.com/problemset/problem/1415

题目分析：求两个字符串的最长公共子串，首先把两个字符串拼起来，然后就和hihocoder 1407类似了，不过多一个条件，公共前缀的两部分需要分别来自两个字符串，时间复杂度O(nlogn)

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int const MAX = 200005;
char s1[MAX], s2[MAX], s[MAX];
int n, m, midPos, sa[MAX];
int rk[MAX], tp[MAX], tax[MAX], height[MAX];

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

void radix_sort() {
    for (int i = 0; i <= m; i++) {
        tax[i] = 0;
    }
    for (int i = 1; i <= n; i++) {
        tax[rk[tp[i]]]++;
    }
    for (int i = 1; i <= m; i++) {
        tax[i] += tax[i - 1];
    }
    for (int i = n; i >= 1; i--) {
        sa[tax[rk[tp[i]]]--] = tp[i];
    }
}

void get_sa() {
    for (int i = 1; i <= n; i++) {
        rk[i] = s[i] - 'a' + 1;
        tp[i] = i;
        m = max(m, s[i] - 'a' + 1);
    }
    radix_sort();
    for (int j = 1, p = 0; p < n; j <<= 1, m = p) {
        p = 0;
        for (int i = n - j + 1; i <= n; i++) {
            tp[++p] = i;
        }
        for (int i = 1; i <= n; i++) {
            if (sa[i] > j) {
                tp[++p] = sa[i] - j;
            }
        }
        radix_sort();
        swap(rk, tp);
        rk[sa[1]] = p = 1;
        for (int i = 2; i <= n; i++) {
            rk[sa[i]] = cmp(tp, sa[i], sa[i - 1], j) ? p : ++p;
        }
    }
}

void get_height() {
    for (int i = 1, j = 0; i <= n; i++) {
        if (j) {
            j--;
        }
        int prevPos = sa[rk[i] - 1];
        while (i + j <= n && prevPos + j <= n && s[i + j] == s[prevPos + j]) {
            j++;
        }
        height[rk[i]] = j;
    }
}

bool judge(int x) {
    int maPos = 0, miPos = n;
    for (int i = 1; i <= n; i++) {
        if (height[i] < x) {
            maPos = sa[i];
            miPos = sa[i];
        } else {
            maPos = max(maPos, sa[i]);
            miPos = min(miPos, sa[i]);
            if (miPos + x - 1 <= midPos && maPos > midPos) {
                return true;
            }
        }
    }
    return false;
}

int main() {
    scanf("%s%s", s1 + 1, s2 + 1);
    midPos = strlen(s1 + 1);
    strcat(s + 1, s1 + 1);
    strcat(s + 1, s2 + 1);
    n = strlen(s + 1);
    get_sa();
    get_height();
    // for (int i = 1; i <= n; i++) {
    //     printf("height[%d] = %d sa[%d] = %d\n", i, height[i], i, sa[i]);
    // }
    int l = 0, r = n, mid = 0, ans = 0;
    while (l <= r) {
        mid = (l + r) >> 1;
        if (judge(mid)) {
            ans = mid;
            l = mid + 1;
        } else {
            r = mid - 1;
        }
    }
    printf("%d\n", ans);
}