51Nod - 1304 ：字符串的相似度 （裸的扩展KMP）

我们定义2个字符串的相似度等于两个串的相同前缀的长度。例如 "abc" 同 "abd" 的相似度为2，"aaa" 同 "aaab" 的相似度为3。

给出一个字符串S，计算S同他所有后缀的相似度之和。例如：S = "ababaa"，所有后缀为：

 

ababaa 6

babaa 0

abaa 3

baa 0

aa 1

a 1

 

S同所有后缀的相似度的和 = 6 + 0 + 3 + 0 + 1 + 1 = 11

Input输入一个字符串S(1 <= L <= 1000000)，L为字符串S的长度，且S由a-z的小写字母组成。Output输出S同所有后缀的相似度的和。Sample Input

ababaa

Sample Output

11

：  KMP可以求S串以i结尾的后缀与T串的前缀最长公共串长度。

： 扩展KMP可以求S串以i为起点的的后缀与T串的前缀的最长公共字串。

此题就是求所有i的扩展KMP长度；

以前学过，已经忘得差不多了。这里再抄一发。

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int N = ;
int next[N],extand[N];
char S[N]; long long ans;
void getnext(){
     int i,length=strlen(S+);
     next[]=length;
     for(i=;i+<length&&S[i+]==S[i+];i++);
     next[]=i;
     int a=;
     for(int k=;k<=length;k++){
          int p=a+next[a]-, L=next[k-a+];
          if(L>=p-k+){
              int j=(p-k+)>?(p-k+):;
              while(k+j<=length&&S[k+j]==S[j+]) j++;
              next[k]=j, a=k;
          }
          else next[k]=L;
    }
    for(int i=;i<=length;i++) ans+=next[i];
    printf("%lld\n",ans);
}
int main(){
    scanf("%s",S+);
    getnext();
    return ;
}