給出一些字符串,求出字符串中所有既是前綴也是後綴的子串。
聯想到KMP算法的next數組。這道題就可做了。
不多說,這整個字符串顯然是滿足條件的。
而且,對於任意一個滿足條件的前綴,它最後一個字符肯定與整個字符串的最後一個字符相同。假設現在有位置k使k=n
開始,一直k = next[k]
迭代下去,遇到s[k]==s[n]
就記錄答案,到
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std ;
const int maxn = 4e5+10 ;
char tar[maxn] ;
int n, m, nxt[maxn], ans[maxn] ;
int main() {
int i, j, k ;
while ( scanf ( "%s", tar+1 ) != EOF ) {
n = strlen(tar+1) ;
memset ( nxt, 0, sizeof(nxt) ) ;
for ( i = 2, j = 0 ; i <= n ; i ++ ) {
for ( ; j && tar[j+1] != tar[i] ; j = nxt[j] ) ;
if ( tar[j+1] == tar[i] ) ++ j ;
nxt[i] = j ;
}
int cnt = 0 ;
ans[++cnt] = n ;
j = nxt[n] ;
for ( ; j ; j = nxt[j] )
if ( tar[j] == tar[n] ) ans[++cnt] = j ;
for ( i = cnt ; i ; i -- )
printf ( "%d ", ans[i] ) ;
putchar('\n') ;
}
return 0 ;
}
再感嘆一下:KMP真是短小而精悍