poj--2752Seek the Name, Seek the Fame KMPnext数组的应用

来源:互联网 发布:淘宝怎么打新品标 编辑:程序博客网 时间:2024/04/30 04:39

题目可以装换成求以最后一个字符结尾的与某一前缀相同的所有后缀的长度;所以需要利用KMP函数中next数组的性质。
我们可以先求出前n-1个字每的next值,然后再求出第n个字母所有可能的(而不是最长的)next值。
当然我们可以先求出所有字母的next值,然后再倒着扫一遍,即next[len],next[next[len]]……..。这样也可以将所有的长度求出来。
注意每一个串的本身是满足一定要求的。

代码如下:

#include<iostream>#include<cstdio>#include<cstring>using namespace std;char str[500000];int next[500000];int ans[500000];int cnt;void Getnext(){    int j=0;    int n=strlen(str+1);    next[1]=0;    for(int i=2;i<=n;i++)    {        if(i==n)        {            while(j>0)            {                if(str[j+1]==str[i])                    ans[cnt++]=j+1;                j=next[j];            }            if(str[j+1]==str[i])                ans[cnt++]=j+1;        }        while(j>0&&str[j+1]!=str[i])            j=next[j];        if(str[j+1]==str[i])            j++;        next[i]=j;    }}int main(){    while(scanf("%s",str+1)!=EOF)    {         cnt=0;         ans[cnt++]=strlen(str+1);         Getnext();         for(int i=cnt-1;i>=0;i--)            printf("%d ",ans[i]);         printf("\n");    }  return 0;}
0 0
原创粉丝点击