小Y的字符串（拓展KMP）

最新推荐文章于 2024-08-25 08:02:26 发布

Wang_128

最新推荐文章于 2024-08-25 08:02:26 发布

阅读量566

点赞数

CC 4.0 BY-SA版权

分类专栏： ----kmp ----模板

本文链接：https://blog.youkuaiyun.com/a664607530/article/details/80989760

----模板同时被 2 个专栏收录

63 篇文章

订阅专栏

----kmp

16 篇文章

订阅专栏

本文详细介绍了如何使用拓展KMP算法解决字符串匹配问题，并通过具体实例演示了算法的实现过程。文章提供了完整的C++代码，解释了如何计算两个字符串间的最长公共前缀，特别适用于处理字符串a的所有后缀与字符串b之间的匹配。

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

题目链接：https://www.nowcoder.com/acm/contest/127/D

解题思路：拓展KMP，只要求出字符串b对字符串a每一个后缀的最长公共前缀即可

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <map>
#include <cmath>
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <bitset>
#include <functional>

using namespace std;

#define LL long long
const int INF = 0x3f3f3f3f;

int nt[2000009], extend[2000009];
char s1[2000009], s2[2000009];


void GetNext(char T[], int nt[]) {
      int t_len = strlen(T);
      nt[0] = t_len;
      int a, p;
      for (int i = 1, j = -1; i < t_len; i++, j--) {
        if (j < 0 || i + nt[i - a] >= p) {
          if (j < 0)
            p = i, j = 0;
          while (p < t_len && T[p] == T[j])
            p++, j++;
          nt[i] = j;
          a = i;
        }
        else nt[i] = nt[i - a];
      }
}

void GetExtend(char S[], char T[], int extend[], int nt[]) {
      GetNext(T, nt);
      int a, p;
      int s_len = strlen(S);
      int t_len = strlen(T);
      for (int i = 0, j = -1; i < s_len; i++, j--) {
        if (j < 0 || i + nt[i - a] >= p) {
          if (j < 0) p = i, j = 0;
          while (p < s_len && j < t_len && S[p] == T[j])
            p++, j++;
          extend[i] = j;
          a = i;
        }
        else extend[i] = nt[i - a];
      }
}

int main() {
  while (~scanf("%s %s", s1, s2)) {
    GetExtend(s1, s2, extend, nt);
    LL ans = 0;
    int len1 = strlen(s1), len2 = strlen(s2);
    for (int i = 0; s1[i]; i++) {
      if (extend[i] == len2) {
        ans = ans + 1LL * extend[i] - 1LL;
        if (s1[i + extend[i]] < s2[extend[i]]) {
          ans = ans + 1LL * len1 - 1LL * i -  1LL * extend[i];
        }
      } else {
        if (s1[i + extend[i]] < s2[extend[i]]) {
          ans = ans + 1LL * len1 - 1LL * i -  1LL * extend[i];
        }
        printf("%lld\n", ans);
      }
    }
    printf("%lld\n", ans);
  }
  return 0;
}