题意:给出字符串,求所有既是前缀又是后缀的串的大小。
思路:KMP的next[]应用。。
例如题目给出的串 ababcababababcabab
对应的next值 -1,0,0,1,2,0,1,2,3,4,3,4,3,4,5,6,7,8,9(next[18])
next[18]=9,代表着位置18以前的既是前缀,又是后缀的串最大是9(不包括本身),然后找
next[9]=4,也就是位置9以前,有4个是前缀,并且和位置9向前数4个的字符串相同,因为位置9以前的字符串(9个)和位置18以前的9个字符串(也就是末尾的9个)相同,所以next[9]=4也代表了有4个字符串既是前缀也是后缀。同理,next[4]=2. next[2]=0,不再找了。。如果首字母等于尾字母,那么也算一个前后缀,数值等于1.
开一个数组tmp,把2,4,9,18(本身字符串大小,可以看作是最大的前后缀)存起来,然后循环输出即可。。
#include<iostream>
using namespace std;
int next[400005];
char s[400005];
int len;
void get_next()
{
int k=-1,j=0;
next[0]=-1;
len=strlen(s);
while(j<len)
{
if(k==-1||s[j]==s[k])
{
next[j+1]=k+1;
j++;k++;
}
else
k=next[k];
}
}
int main()
{
int tmp[400005];
while(scanf("%s",s)!=EOF)
{
int i=1;
get_next();
for(int g=0;g<=len;g++)
printf("%d ",next[g]);
int end=len;
tmp[0]=len;
while(next[len]!=0)//递归的找
{
tmp[i++]=next[len];
len=next[len];
}
if(s[0]==s[end])
printf("1 ");
for(int j=i-1;j>0;j--)
printf("%d ",tmp[j]);
printf("%d\n",tmp[0]);
}
}