19 hdu多校 string matching //exkmp(模板）

题意：求字符串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;
}