题意:求字符串S的每个后缀与S的最长相同前缀。
EXKMP
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <cstdio>
#include <vector>
#include <climits>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define PII pair<int,int>
#define MP make_pair
const LL mod = 1e9+7;
const int MX = 1e6+5;
char s[MX],ss[MX];
int net[MX],ex[MX]; //ex数组即为extend数组
void GETNEXT(char *str){//预处理计算next数组
int i=0,j,po,len=strlen(str);
net[0]=len;//初始化net[0]
while(str[i]==str[i+1]&&i+1<len)//计算net[1]
i++;
net[1]=i;
po=1;//初始化po的位置
for(i=2; i<len; i++){
if(net[i-po]+i<net[po]+po)//第一种情况,可以直接得到net[i]的值
net[i]=net[i-po];
else{//第二种情况,要继续匹配才能得到net[i]的值
j=net[po]+po-i;
if(j<0)j=0;//如果i>po+net[po],则要从头开始匹配
while(i+j<len&&str[j]==str[j+i])//计算net[i]
j++;
net[i]=j;
po=i;//更新po的位置
}
}
}
//计算extend数组
void EXKMP(char *s1,char *s2){ //求出s中每一个后缀和子串t的最长公共前缀
int i=0,j,po,len=strlen(s1),l2=strlen(s2);
GETNEXT(s2);//计算子串的net数组
while(s1[i]==s2[i]&&i<l2&&i<len)//计算ex[0]
i++;
ex[0]=i;
po=0;//初始化po的位置
for(i=1; i<len; i++){
if(net[i-po]+i<ex[po]+po)//第一种情况,直接可以得到ex[i]的值
ex[i]=net[i-po];
else{//第二种情况,要继续匹配才能得到ex[i]的值
j=ex[po]+po-i;
if(j<0)j=0;//如果i>ex[po]+po则要从头开始匹配
while(i+j<len&&j<l2&&s1[j+i]==s2[j])//计算ex[i]
j++;
ex[i]=j;
po=i;//更新po的位置
}
}
}
int main(){
int T;cin>>T;while(T--){
scanf("%s",s);int le=strlen(s);
strcpy(ss,s);
for(int i=0;i<le+2;i++) net[i]=ex[i]=0;
EXKMP(ss,s);
LL ans=0;
for(int i=1;i<le;i++){
ans=ans+(LL)ex[i]-(i+ex[i]==le)+1;
}
cout<<ans<<'\n';
}
return 0;
}