「CF126B」 Password「KMP」

版权声明：本文为hzy原创文章，未经博主允许不可随意转载。 https://blog.csdn.net/Binary_Heap/article/details/81981224

题意

在字符串中找一个子串，满足既是前缀也是后缀，还在中间出现过.

题解

首先预处理出$KMP$的$next$数组，然后从$next_n$开始跳，不停让$i=next_i$。

考虑$next_n$的含义：前缀等于后缀。每跳一次相当于这个前缀等于后缀的长度减小了。

每次枚举的时候使用$KMP$匹配中间位置，查看是否在中间出现过，出现就输出答案.

#include <cstring>
#include <cstdio>

char * No_res = "Just a legend\n";

char s[1000100];
int n, f[1000100];

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

bool KMP(char * T, int len) {
    int j = 0;
    for(int i = 1; i < n-1; i ++) {
        for(; j && T[j] != s[i]; j = f[j]) ;
        if(T[j] == s[i]) ++ j;
        if(len == j) return true;
    }
    return false;
}

int main() {
    scanf("%s", s);
    n = strlen(s);
    get_fail();
    for(int k = f[n]; k; k = f[k])
        if(KMP(s, k)) return printf("%s\n", &s[n-k]), 0;
    printf(No_res);
    return 0;
}