题目: 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;
}