【KMP && 输出所有前缀=后缀的长度】POJ - 2752 Seek the Name, Seek the Fame

Step1 Problem:

给你字符串s, 输出所有前缀=后缀的长度(包括自身).

Step2 Ideas:

个人习惯从0开始,next[0] = 0;
next[i] : 代表下标从 0 到 i 这个子串,后缀 = 前缀 最长长度(不包括自身)。
下标 k 从 len-1 开始。
next[k] != 0, 长度为 next[k] 的后缀=前缀。
k 移动到下标 next[k-1] 位置,因为后缀 = 前缀,相当于前缀就是后缀,所以 移动到前缀的最后一个字符位置处。在重复上面操作

Step3 Code:

#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int N = 1000055;
char s[N];
int nex[N], len;
void get_nex()
{
    nex[0] = 0;
    for(int i = 1; i < len; i++)
    {
        int j = nex[i-1];
        while(j && s[i] != s[j])
            j = nex[j-1];
        if(s[i] == s[j]) nex[i] = j + 1;
        else nex[i] = 0;
    }
}
int main()
{
    while(~scanf("%s", s))
    {
        len = strlen(s);
        get_nex();
        vector<int> a;
        a.clear();
        int k = len;
        while(k != 0) {
        a.push_back(k);//此时的k,代表前缀 = 后缀的长度
        k = nex[k-1];}
        for(int i = a.size() - 1; i >= 0; i--)
        {
            printf("%d", a[i]);
            i?printf(" "):printf("\n");
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/bbbbswbq/article/details/80743467