HDU 6629 string matching —— 扩展KMP

This way

题意：

有一个函数，它每次会暴力的去跑字符串s的所有后缀于字符串s的最长公共前缀，给你字符串s，问你这个函数会跑几次

题解：

学到了，模板。。
这是逼我们baidu

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int MAX=1000100; //字符串长度最大值
int Next[MAX],extend[MAX];

//预处理计算Next数组
void getNext(char *str)
{
    int i=0,j,po,len=strlen(str);
    Next[0]=len; //初始化Next[0]
    while(str[i]==str[i+1] && i+1<len) i++; Next[1]=i; //计算Next[1]
    po=1; //初始化po的位置
    for(i=2;i<len;i++)
    {
        if(Next[i-po]+i < Next[po]+po) //第一种情况，可以直接得到Next[i]的值
            Next[i]=Next[i-po];
        else //第二种情况，要继续匹配才能得到Next[i]的值
        {
            j = Next[po]+po-i;
            if(j<0) j=0; //如果i>po+Next[po],则要从头开始匹配
            while(i+j<len && str[j]==str[j+i]) j++; Next[i]=j;
            po=i; //更新po的位置
        }
    }
}

//计算extend数组
void EXKMP(char *s1,char *s2)
{
    int i=0,j,po,len=strlen(s1),l2=strlen(s2);
    getNext(s2); //计算子串的Next数组
    while(s1[i]==s2[i] && i<l2 && i<len) i++; extend[0]=i;
    po=0; //初始化po的位置
    for(i=1;i<len;i++)
    {
        if(Next[i-po]+i < extend[po]+po) //第一种情况，直接可以得到extend[i]的值
            extend[i]=Next[i-po];
        else //第二种情况，要继续匹配才能得到extend[i]的值
        {
            j = extend[po]+po-i;
            if(j<0) j=0; //如果i>extend[po]+po则要从头开始匹配
            while(i+j<len && j<l2 && s1[j+i]==s2[j]) j++; extend[i]=j;
            po=i; //更新po的位置
        }
    }
}
char s[MAX];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s",s);
        EXKMP(s,s);
        int len=strlen(s);
        ll ans=0;
        for(int i=1;i<len;i++)
            ans+=(ll)(extend[i]==len-i?extend[i]:extend[i]+1);
        printf("%lld\n",ans);
    }
    return 0;
}