CF126B KMP算法运用

罗侯

于 2017-10-19 09:49:43 发布

阅读量628

点赞数

CC 4.0 BY-SA版权

分类专栏： kmp算法文章标签： kmp

本文链接：https://blog.youkuaiyun.com/computer_user/article/details/78280695

kmp算法专栏收录该内容

16 篇文章

订阅专栏

本文介绍了如何利用KMP算法解决Codeforces比赛中的CF126B题目，重点在于分析字符串中既是前缀又是后缀且在中间部分出现的最长子串的方法。通过理解KMP算法的next数组，可以判断模式字符串在中间部分的出现情况，借助set容器进行有效查找。

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

CF126B题目链接

分析：给定一个字符串，查找既是前缀又是后缀且在中间部分出现的最长子串。

模式字符串记为P(下标从0开始)，next[q] = k 表示P[q]之前的子串中，存在长度为k的相同前缀和后缀，即P[0]~P[k-1]与P[q-k]~P[q-1]依次相同。

所以只要判断Next[i]是否在中间出现过即可，使用set容器记录。

代码如下：

#include <cstdio>
#include <set>
#include <cstring>
using namespace std;

const int maxn = 1e6+10;
int Next[maxn];
int n;
char s[maxn];
bool flag;
set<int>v;

void make_next(){
   Next[0] = Next[1] = 0;
   for (int i=1; i<n; i++) {
      int j = Next[i];
      while (j && s[i]!=s[j]) j = Next[j];
      Next[i+1] = s[j]==s[i]?j+1:0;
   }
}

int main(){
    while (scanf("%s",s)!=EOF){
        n = strlen(s);
        make_next();
        v.clear();
        for (int i=0; i<n; i++) v.insert(Next[i]);
        flag = 0;
        int x = n;
        while (Next[x]){
            if (v.count(Next[x])) {
                 flag = 1;
                 for (int i=0; i<Next[x]; i++) printf("%c",s[i]);
                 printf("\n");
                 break;
            }
            x = Next[x];
        }
        if (!flag) printf("Just a legend\n");
    }
    return 0;
}