要解决这道题,首先要理解KMP中的next数组,现在觉得这个next数组真的很重要啊!!!
要找到一个字符串的所有的既是前缀又是后缀的子串的长度,那么就要知道,next指针的的含义,指向最长既是前缀又是后缀的前缀位置的后一位,那么一个对于一个位置的next的指针的位置的next指针指向的位置,于当前位置来说是什么意义呢?其实就是一个当前位置为止,既是前缀又是后缀的前缀的后一位,只不过不再是最长的了。
那么明白这个原理,就可以求一个字符串所有的既是前缀也是后缀的子串了以及它们各自长度。
代码如下:
#include <cstdio>
#include <cstring>
const int N = 400010;
int next[N], sl, ans[N], cnt;
char str[N];
void getNext()
{
int j, k;
j = 0, k = -1; next[0] = -1;
while ( j < sl ) {
if ( k == -1 || str[j] == str[k] ) next[++j] = ++k;
else k = next[k];
}
}
int main()
{
while ( scanf("%s", str) != EOF ) {
sl = strlen(str), cnt = 0;
getNext();
//for ( int i = 0; i <= sl; ++i ) printf("%d ", next[i]);
int tmp = sl;
while ( tmp ) {
ans[cnt++] = tmp;
tmp = next[tmp];
//printf("%d ", tmp);
}
printf("%d", ans[--cnt]);
for ( int i = cnt-1; i >= 0; --i ) printf(" %d", ans[i]);
printf("\n");
}
}