POJ 2752 既是前缀又是后缀

最新推荐文章于 2018-01-31 21:41:00 发布

原创最新推荐文章于 2018-01-31 21:41:00 发布 · 638 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#POJ 2752 既是前缀又是后缀

本文介绍了一种使用KMP算法快速找出字符串中既是前缀也是后缀的子串的方法，并提供了一个完整的C++代码实现。通过利用next数组特性，可以有效地找出所有符合条件的子串。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

分析：

给定一个字符串s，从小到大输出s中既是前缀又是后缀的子串的长度。

此题非常简单，借用KMP算法的next数组，设s的长度为n，则s串本身必定满足条件。其他满足条件的子串都有个特征，就是该子串的最后一个字符肯定与s的最后一个字符相同。这正是next数组发挥作用的时候。从n - 1位既最后一位开始回滚，若s[next[n-1]] == s[n-1]，则子串s[0,1,2,...,next[n-1]]是满足条件的子串。然后判断s[next[next[n-1]]] == s[n-1]是否成立，这样一直回滚，直到next[next[.....next[n-1]]] == -1为止。把答案从大到小存下来，再从小到大输出即可。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;
const int maxn = 400005;
int next[maxn];

void getnext(char *src, int len){

    int i = 0, j = -1;
    next[0] = j;
    while (i < len){
        while (j != -1 && src[j] != src[i])
            j = next[j];
        i++; j++;
        if (j >= len)
            next[i] = next[j-1];
        else
            next[i] = j;
    }
}

int main(){
    char s[maxn];
    int ans[maxn];
    while (scanf("%s", s) != EOF){
        memset(ans, 0, sizeof(ans));
        int len = strlen(s);
        getnext(s, len);
        int t = next[len-1]; //从最长开始寻找
        int cnt = 0;
        while (t != -1){
            if (s[t] == s[len-1]) ans[cnt++] = t + 1;
            t = next[t];
        }
        for (int i = cnt-1; i>=0; i--){
            printf("%d ", ans[i]);
        }
        printf("%d\n", len);
    }
    return 0;
}