KMP + 求相等前后缀--- POJ Seek the Name, Seek the Fame

Seek the Name, Seek the Fame 

Problem's Link:  http://poj.org/problem?id=2752

---

 

Mean: 

给你一个字符串，求这个字符串中有多少前缀是和后缀相等的， 按递增顺序输出这些前缀的长度。

analyse:

KMP之next数组的运用。

next[k]表示的是s[0....next[k]] 和s[k-next[k] .... k] 这段是相等的，即以k结尾的最长相等前后缀长度。

next[k-1]表示的是与s[0....k]具有最长后缀，且与s[0...k]具有最长前缀的位置。也就是在k位置失配后需要移到next[k-1]的位置。

这里特别说明一下，失配后移到next[k]还是next[k-1]，取决于next数组的开始编号。如果从0开始编号，那就移到next[k-1];从1开始编号，就移到next[k]。

Time complexity: O(N)

 

Source code: 

/*
* this code is made by crazyacking
* Verdict: Accepted
* Submission Date: 2015-07-28-08.55
* Time: 0MS
* Memory: 137KB
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
#define  LL long long
#define  ULL unsigned long long
using namespace std;
const int MAXN = 400010;
int n;
char s [ MAXN ];
int Next [ MAXN ];
int ans [ MAXN ];
void getNext()
{
      Next [ 0 ] = 0;
      for( int i = 1 , k = 0; i <n; ++ i)
      {
            while(s [ i ] !=s [ k ] && k) k = Next [ k - 1 ];
            if(s [ i ] ==s [ k ]) ++ k;
            Next [ i ] = k;
      }
}
int main()
{
      ios_base :: sync_with_stdio( false);
      cin . tie( 0);
      while( ~ scanf( "%s" ,s))
      {
           n = strlen(s);
            getNext();
            int k =n - 1;
            int cnt = 0;
            while( Next [ k ])
            {
                  ans [ cnt ++ ] = Next [ k ];
                  k = Next [ k - 1 ];
            }
            ans [ cnt ++ ] =n;
            sort( ans , ans + cnt);
            for( int i = 0; i < cnt; ++ i)
            {
                  printf( i == cnt - 1 ? "%d \n " : "%d " , ans [ i ]);
            }
      }
      return 0;
}
/*

*/