CodeForces - 126B Password KMP+dp思想

CodeForces - 126B Password

题意：

给你一个字符串，找出一个最长子串 t ，t 既是前缀也是后缀，并且 t 在字符串中间也出现过

思路：

利用KMP求出 nxt 数组。然后预处理出整个字符串有多少个与该前缀相同的子串。用到DP思想，cnt[nxt[i]]+=cnt[i]，倒着更新一遍。那么cnt大等于3的前缀，均为可行解

 利用next[next[i]]是次长公共前缀解决

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>
#include<bitset>
#include<stack>
#include<vector>
#define mod 10007
using namespace std;
typedef long long ll;
const int maxn = 1e6+10;
char s[maxn];
int nxt[maxn];
int cnt[maxn];  //字符串有多少个与该前缀相同的子串
int len;
void get_nxt( )
{
    int i,j;
    i = 0;
    j = nxt[0] = -1;
    while(i<=len)
    {
        if(j==-1 || s[i]==s[j])
        {
            j++,i++;
            nxt[i] = j;
        }
        else
            j = nxt[j];
    }
}
int main()
{
    scanf("%s",s);
    len = strlen(s);
    get_nxt();
    for(int i=1;i<=len;i++)
        cnt[i] = 1;
    for(int i=len; i ;i--)
        cnt[nxt[i]] += cnt[i];
    int ans = nxt[len];
    while(ans)
    {
        if(cnt[ans]>=3)
        {
            for(int i=0;i<ans;i++)
                printf("%c",s[i]);
            printf("\n");
            return 0;
        }
        ans = nxt[ans];
    }
    printf("Just a legend\n");
    return 0;
}