POJ2752 既是前綴也是後綴 KMP變形

  給出一些字符串,求出字符串中所有既是前綴也是後綴的子串。

  聯想到KMP算法的next數組。這道題就可做了。
  不多說,這整個字符串顯然是滿足條件的。
  而且,對於任意一個滿足條件的前綴,它最後一個字符肯定與整個字符串的最後一個字符相同。假設現在有位置k使[1,k] 滿足條件,那麼顯然[1,nextk] 仍然滿足條件,並且,由於next[]是最長的公共前後綴,所以不存在某滿足條件的前綴長度大於nextk 小於k 。從k=n開始,一直k = next[k]迭代下去,遇到s[k]==s[n]就記錄答案,到k=0 時爲止。這樣我們就把答案從大到小記錄下來了。逆序輸出即可。
  

#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真是短小而精悍

發佈了55 篇原創文章 · 獲贊 39 · 訪問量 4萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章