HUST 1328 String

11:

KMP next 的强大

题意求前缀在S中出现的次数之和

next[j] 表示 S[0....NEXT[J]]==S[J-NEXT[J].....J];

于是我们得到。。后加入一个字符所得到新的前缀会多ADD[next[J]]个

#include<stdio.h>

#include<string.h>

#include<algorithm>

#include<iostream>

#include<math.h>

using namespace std;

#define mod 1000000007

typedef long long ll;

#define N 123456

char s[N];

int next[N];

int n;

ll a[N];

void kmp()

{

n=strlen(s);

int i=0,j=-1;

next[0]=-1;

while (i<n)

{

if (j==-1||s[i]==s[j])

{

i++;

j++;

next[i]=j;

}else j=next[j];

}

}

int main()

{

while (scanf("%s",s)!=EOF)

{

kmp();

ll ans=0;

memset(a,0,sizeof(a));

for (int i=1;i<=n;i++)

{

a[i]=a[next[i]];

a[i]++;

ans+=a[i];

}

printf("%lld\n",ans);

}

return 0;

}