做题CF119D. String Transformation——KMP

Posted 2021-01-05 cly-none

题意：有两个字符串$a,b$，下标从$0$开始。求数对$(i,j)$满足$a[i+1:j] + r(a[j:n]) + r(a[0:i+1]) = b$，其中$r(s)$表示字符串$s$的反串。若有多组解，输出其中$i$最大，然后$j$尽可能小的一组。

$|a|,|b| leq 10^6$

首先考虑枚举$i$。那么，我们就要让$a[i+1:j] + r(a[j:n]) = b[0:n-i-1]$。因此，前面的$a[i+1:j]$必须是$b$的一个前缀。这个限制比较简单，求出最长公共前缀后，就可以转化为$j leq r$的形式。

接下来，我们得满足$r(a[j:n])$是$b[0:n-i-1]$的一个后缀。并且，我们只要求出满足这个条件的最小的$j$就可以了。也就是求出$a$最长的后缀，它在翻转后也是$b[0:n-i-1]$的后缀。这个问题比较复杂，要进行化简。先解决翻转。记$a_r$为$a$的反串，那么，问题就成为求最长的$a_r$的前缀，它也是$b[0:n-i-1]$的后缀。开始于一个固定位置的前缀，结束于多个不同位置的后缀。仔细一想的话，这正是KMP中nex数组的定义。因此，我们把$a_r$和$b$连起来（分隔符还是要加的），求一遍nex数组就可以了。

顺便一提，最长公共前缀必须用EXKMP来求，不然会T。

时间复杂度$O(n)$。

#include <bits/stdc++.h>
using namespace std;
const int N = 1000010;
char a[N],b[N],s[N << 1];
int nex[N << 1],n,pre[N << 1],pw[N],ans1,ans2;
int main() {
  cin.getline(a+1,N);
  cin.getline(b+1,N);
  if (strlen(a+1) != strlen(b+1)) {
    puts("-1 -1");
    return 0;
  }
  n = strlen(a+1);
  for (int i = 1 ; i <= n ; ++ i)
    s[i] = a[n - i + 1];
  s[n+1] = 0;
  for (int i = 1 ; i <= n ; ++ i)
    s[i + n + 1] = b[i];
  nex[0] = -1;
  for (int i = 1, j = -1 ; i <= (n << 1) + 1 ; nex[i++] = ++ j)
    while (j >= 0 && s[i] != s[j+1]) j = nex[j];
  for (int i = 1 ; i <= n ; ++ i)
    s[i] = b[i];
  s[n+1] = 0;
  for (int i = 1 ; i <= n ; ++ i)
    s[i + n + 1] = a[i];
  pre[1] = 2 * n + 1;
  while (s[2 + pre[2]] == s[1 + pre[2]] && 2 + pre[2] <= 2 * n + 1)
    ++ pre[2];
  int p = 2;
  for (int i = 3 ; i <= 2 * n + 1 ; ++ i) {
    if (pre[i - p + 1] < pre[p] + p - i)
      pre[i] = pre[i - p + 1];
    else {
      int j = max(0,p + pre[p] - i);
      while (s[i + j] == s[1 + j] && i + j <= 2 * n + 1)
                ++ j;
      pre[i] = j;
      p = i;
    }
  }
  ans1 = ans2 = -1;
  for (int i = 1 ; i <= n ; ++ i) {
    int r = pre[i + n + 1] + i;
    int l = nex[2 * n + 1 - (i-1)];
    if (n - l + 1 <= r) ans1 = i - 2, ans2 = n - l;
    if (a[i] != b[n-i+1]) break;
  }
  if (ans1 == -1) ans2 = -1;
  printf("%d %d
",ans1,ans2);
  return 0;
}

小结：这道题难度其实不大，但推导过程较长，因此需要时刻保持思路清晰，心态平稳。