[hdu-6629]string matching 扩展kmp 2019多校5

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6629

题目大意：给一个字符串s [0,...len-1]   求子串 si [ i,...len-1 ] 与 s的最长公共前缀，  即求所有子串后缀的最长公共前缀的长度；

给了一个暴力算法（每一个后缀子串都与原串从第一位开始比），求这个暴力算法要比较多少次。

题解：用扩展kmp求每个后缀子串si与原串s的最长公共前缀，然后如果就是后缀si，则比较次数+nex（si），否则+nex（si）+1   ，nex（si）为最长公共前缀长度

 

一个将扩展kmp清晰的博客链接：https://subetter.com/algorithm/extended-kmp-algorithm.html

#include<bits/stdc++.h>
using namespace std;

int const maxn=1000100;
//扩展kmp:求t每一个后缀字符串匹配s最长前缀长度
/*
extend[i]:S[i]...S[n-1]与 T 的最长相同前缀的长度
next[i]： T[i]...T[m - 1]与 T 的最长相同前缀长度；
*/
int lent,nex[maxn];
char t[maxn];
long long ans;
void getnext(){
    lent=strlen(t);
    int a=0,p=0;//p是最长的匹配到的位置，a是它开始匹的位置
    nex[0]=lent;
    for(int i=1;i<lent;i++){
        if(i>=p||i+nex[i-a]>=p){
            if(i>=p)p=i;
            while(p<lent&&t[p]==t[p-i])p++;
            nex[i]=p-i;a=i;
        }
        else nex[i]=nex[i-a];
        if(i+nex[i]==lent)ans+=nex[i];
        else ans+=nex[i]+1;
    }
}
int main(){
    int Y;
    scanf("%d",&Y);
    while(Y--){
        scanf("%s",t);
        ans=0;
        getnext();
        cout<<ans<<endl;
    }
    return 0;
}