POJ 2752+KMP+利用next陣列性質求出所有相同的字首和字尾

acm_lkl發表於2020-04-04

題目連結:點選進入
這個題目要求所有相同的字首和字尾的長度。我們可以利用KMP演算法中next陣列的性質,在next[len]這個點不斷的失配下去,這樣就可以將所有相同的前字尾的長度求出來。還要注意這個中整個串的長度也可以看成是一個合法的解。

程式碼如下:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

const int maxn=400000+100;
char str[maxn];
int  next1[maxn];
int  ans[maxn],cnt;

void kmp_next1(char x[],int m,int next1[])
{
    int i,j;
    j=next1[0]=-1;
    i=0;
    while(i<m-1)
    {
        while(-1!=j && x[i]!=x[j]) j=next1[j];
        next1[++i]=++j;
    }
    ///在next[m]這個點要一直"失配"下去
    ///這樣就可以將所有與字尾相同的字首求出來
    while(-1!=j)
    {
       if(x[i]==x[j])
         ans[cnt++]=j+1;
       j=next1[j];
    }
}

int main()
{
    freopen("in.txt","r",stdin);
    while(scanf("%s",str)!=EOF)
    {
        int len=strlen(str);
        cnt=0;
        kmp_next1(str,len,next1);
        for(int i=cnt-1;i>=0;i--)
           printf("%d ",ans[i]);
        printf("%d\n",len);
    }
  return 0;
}

相關文章