51Nod-1638-字符串重组

ACM模版

描述

题解

仔细分析这个问题两个串儿的结构，首先将第一个串通过 $i$ 和 $j$ 拆解成三部分，$i + 1 \sim j - 1$ 作为第二个串儿的第一部分，$j \sim n - 1$ 翻转拼接其后，$0 \sim i$ 翻转拼接其后，所以其实这个问题也就是为字符串匹配问题。首先将原串进行翻转然后 $kmp$，之所以要这样做是因为第二个串儿后边两部分是翻转后的匹配，并且要求 $i$ 尽量大的情况下 $j$ 尽量小，所以也就是第二个串儿第一部分尽量小，所以我们翻转后考虑后边两部分。

此时，我们考虑翻转后，原串儿依然被划分为三部分，不过，和题目中描述不同的是，第一个串儿的第一部分不用翻转放在第二个串的第三部分，第一个串儿的第三部分不用翻转放在第二个串的第二部分，而第一个串的第二部分则是翻转后放在第二个串的第一部分。此时也就是说考虑两串的后缀的公共前缀，这样的话，我们就应该使用 $exKmp$。

此代码改自 $rank1$ 大佬 LittleFairy_MY 的代码，第一次写扩展 $KMP$，不是特别的会，理解上可能有些许偏差，请大佬们见谅。

代码

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int MAXN = 1e6 + 10;

int len_a, len_b;
char a[MAXN], b[MAXN], c[MAXN];
int p[MAXN], _nt[MAXN], z[MAXN], extend[MAXN];

void kmp()
{
    memcpy(c, a, sizeof(a));
    reverse(c, c + len_a);

    p[0] = 0;
    for (int i = 1, j = 0; i < len_a; ++i)
    {
        while (j && c[j] != c[i])
        {
            j = p[j - 1];
        }
        if (c[j] == c[i])
        {
            ++j;
        }
        p[i] = j;
    }

    for (int i = 0, j = 0; i < len_a; ++i)
    {
        while (j && b[i] != c[j])
        {
            j = p[j - 1];
        }
        if (b[i] == c[j])
        {
            ++j;
        }
        _nt[i] = j;
    }
}

void exkmp()
{
    int k = 1, p, i, j;
    for (j = 0; j + 1 < len_a && b[j] == b[j + 1]; ++j) ;
    z[1] = p = j;

    for (i = 2; i < len_a; ++i)
    {
        if (z[i - k] < p - i + 1)
        {
            z[i] = z[i - k];
        }
        else
        {
            for (j = max(0, p - i + 1); i + j < len_a && b[i + j] == b[j]; ++j) ;
            z[i] = j;
            k = i;
            p = i + j - 1;
        }
    }

    for (j = 0; j < len_a && a[j] == b[j]; ++j) ;
    extend[0] = j;
    k = 0;
    p = j - 1;
    for (i = 1; i < len_a; ++i)
    {
        if (z[i - k] < p - i + 1)
        {
            extend[i] = z[i - k];
        }
        else
        {
            for (j = max(0, p - i + 1); i + j < len_a && a[i + j] == b[j]; ++j) ;
            extend[i] = j;
            k = i;
            p = i + j - 1;
        }
    }
}

int main()
{
    fgets(a, MAXN, stdin);
    fgets(b, MAXN, stdin);
    len_a = (int)strlen(a) - 1;
    len_b = (int)strlen(b) - 1;
    a[len_a] = '\0';
    b[len_b] = '\0';

    if (len_a != strlen(b))
    {
        puts("-1 -1");
        return 0;

    }

    kmp();
    exkmp();

    int _i = -1, _j = -1;
    for (int i = 0; i + 1 < len_a && a[i] == b[len_a - 1 - i]; ++i)
    {
        if (extend[i + 1] >= len_a - _nt[len_a - 2 - i] - i - 1)
        {
            _i = i;
            _j = len_a - _nt[len_a - 2 - i];
        }
    }

    printf("%d %d\n", _i, _j);

    return 0;
}