[kmp] Jzoj P5178 So many prefix?

Description

 

Input

Output

 

Sample Input

样例一：
abababc

样例二：
isdashagayisdashagaydashisnotagaydashisnotagay

Sample Output

样例一：
6

样例二：
30

 

Data Constraint

 

题解

• 首先显然的是从第一位开始，每个偶数位对应的前缀一定是一个答案，那我们记f[i]=1(i%2==0)

   

• 然后我们考虑KMP的过程，后面串匹配前缀，所以如果他能匹配到一个偶数，也就是说nxt[j]=i(i%2==0)

• 那么我们就可以发现这个字符串对应的肯定是一个长度为偶数的前缀，那么直接累加就好了
  
  

代码

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
int next[200010],f[200010],len,j,ans;
char ch[200010];
int main()
{
    scanf("%s",ch+1),len=strlen(ch+1);
    for (int i=2;i<=len;i++)
    {
        while (j>0&&ch[j+1]!=ch[i]) j=next[j];
        if (ch[j+1]==ch[i]) j++;
        next[i]=j;
    }
    for (int i=2;i<=len;i++)
    {
        if (i%2==0) f[i]++;
        f[i]+=f[next[i]],ans+=f[i];
    }
    printf("%d",ans);
}