POJ2752 既是前缀也是后缀 KMP变形

最新推荐文章于 2021-12-26 22:18:10 发布

WT_cnyali

最新推荐文章于 2021-12-26 22:18:10 发布

阅读量699

点赞数 1

分类专栏：算法 Sol 字符串文章标签： kmp poj

本文链接：https://blog.csdn.net/WT_cnyali/article/details/69366940

版权

算法同时被 3 个专栏收录

43 篇文章 0 订阅

订阅专栏

Sol

30 篇文章 0 订阅

订阅专栏

字符串

10 篇文章 0 订阅

订阅专栏

　　给出一些字符串，求出字符串中所有既是前缀也是后缀的子串。

　　联想到KMP算法的next数组。这道题就可做了。
　　不多说，这整个字符串显然是满足条件的。
　　而且，对于任意一个满足条件的前缀，它最后一个字符肯定与整个字符串的最后一个字符相同。假设现在有位置k使 $[1,k]$ 满足条件，那么显然 $[1,next_k]$ 仍然满足条件，并且，由于next[]是最长的公共前后缀，所以不存在某满足条件的前缀长度大于 $next_k$ 小于 $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 ;
}