HDU 6629. string matching (扩展KMP)

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

题意：
给定一个字符串T；
取子串T[i,len-1]与T[0,len-1]暴力匹配，i∈[1,len-1]；
问需要做多少次字符比较。

分析：
即求扩展KMP的nxt数组；
nxt[i]表示T[i,len-1]与T[0,len-1]的最长公共前缀。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long llong;
const int tmax=1e6+5;
int nxt[tmax],ex[tmax],len;
void getNext(char *s)
{
    int i=0,j,p0;
    nxt[0]=len;
    while(s[i]==s[i+1]&&i+1<len) i++;
    nxt[1]=i;
    p0=1;
    for(i=2;i<len;i++)
    {
        if(nxt[i-p0]+i<nxt[p0]+p0)
            nxt[i]=nxt[i-p0];
        else
        {
            j=nxt[p0]+p0-i;
            j=max(j,0);
            while(i+j<len&&s[j]==s[i+j]) j++;
            nxt[i]=j;
            p0=i;
        }
    }
    return;
}
void work()
{
    llong ans=0;
    for(int i=1;i<len;i++)
        if(i+nxt[i]>=len)
            ans+=nxt[i];
        else ans+=nxt[i]+1;
    cout<<ans<<endl;
    return;
}
int main()
{
    char s[tmax];
    int kase;
    scanf("%d",&kase);
    while(kase--)
    {
        scanf("%s",s);
        len=strlen(s);
        getNext(s);
        work();
    }
    return 0;
}