2018年全国多校算法寒假训练营练习比赛（第五场） C 字符串的问题 KMP算法（判断是否是前缀和后缀）

题目描述 
有一个字符串 让你找到这个字符串 S 里面的子串T 这个子串 T 必须满足即使这个串的前缀 也是这个
串的后缀 并且 在字符串中也出现过一次的(提示 要求满足前后缀的同时也要在字符串中出现一次 只是前后缀可不行 输出最长满足要求字符串)
输入描述:
给出一个字符串 长度 1 到 1e6  全部是小写字母
输出描述:
如果找的到就输出这个子串T 如果不行就输出 Just a legend
示例1
输入
fixprefixsuffix
输出
fix
示例2
abcdabc
输出

Just a legend

思路：先判断字符串子串的是否为前缀和后缀，然后计算该子串在字符串中出现的次数，出现次数大于3，同时是最长的符合条件的子串即可输出。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N = 1000005;
int Next[N];
int ans[N];
int cnt;
int len;

char S[N], T[N];
char answer[N];
int slen, tlen;
void getNext()
{
    Next[0] = -1;
    int i = 0, j = -1;
    while (i < len)
    {
        if (j == -1 || S[i] == S[j])
        {
            ++i;
            ++j;
            Next[i] = j;
        }
        else j = Next[j];
    }
}
int KMP_Count()
{
    int ans = 0;
    int i, j = 0;
    if(slen == 1 && tlen == 1)
    {
        if(S[0] == T[0])
            return 1;
        else
            return 0;
    }
    getNext();
    for(i = 0; i < slen; i++)
    {
        while(j > 0 && S[i] != T[j])
            j = Next[j];
        if(S[i] == T[j])
            j++;
        if(j == tlen)
        {
            ans++;
            j = Next[j];
        }
    }
    return ans;
}
int main()
{

    scanf("%s",S);
    len = strlen(S);
    slen = strlen(S);
    getNext();
    cnt = 0;
    int t = Next[len - 1];
    while (t != -1)
    {
        if (S[t] == S[len - 1]) ans[cnt++] = t + 1;
        t = Next[t];
    }
    int flag=0;
    for(int j=cnt-1; j>=0; j--) //从小到大
    {

        for(int i=0; i<ans[j]; i++) //得到子串
            T[i]=S[i];
        tlen = strlen(T);
        int num=KMP_Count();  //计算出现次数
       // printf("%d  %d   %d\n",ans[j],num,tlen);
        if(num>=3)
        {
            strcpy(answer,T);
            flag=1;
        }

    }
    if(flag)
        printf("%s\n",answer);
    else
        printf("Just a legend\n");

    return 0;
}