POJ-2752-Seek the Name, Seek the Fame(KMPnext数组应用)

Seek the Name, Seek the Fame

Time Limit: 2000MS		Memory Limit: 65536K
Total Submissions: 21810		Accepted: 11411

Description

The little cat is so famous, that many couples tramp over hill and dale to Byteland, and asked the little cat to give names to their newly-born babies. They seek the name, and at the same time seek the fame. In order to escape from such boring job, the innovative little cat works out an easy but fantastic algorithm: 

Step1. Connect the father's name and the mother's name, to a new string S. 
Step2. Find a proper prefix-suffix string of S (which is not only the prefix, but also the suffix of S). 

Example: Father='ala', Mother='la', we have S = 'ala'+'la' = 'alala'. Potential prefix-suffix strings of S are {'a', 'ala', 'alala'}. Given the string S, could you help the little cat to write a program to calculate the length of possible prefix-suffix strings of S? (He might thank you by giving your baby a name:) 

Input

The input contains a number of test cases. Each test case occupies a single line that contains the string S described above. 

Restrictions: Only lowercase letters may appear in the input. 1 <= Length of S <= 400000. 

Output

For each test case, output a single line with integer numbers in increasing order, denoting the possible length of the new baby's name.

Sample Input

ababcababababcabab
aaaaa

Sample Output

2 4 9 18
1 2 3 4 5

题意：找既是前缀也是后缀的长度

思路：其实原理很简单，首先设next[i] = t，则next[i]代表 s[0]~s[t-1] == s[i-t]~s[i-1],所以

如左图，假设黑色线来代表字符串str，其长度是len，红色线的长度代表next[len]，根据next数组定义易得前缀的next[len]长度的子串和后缀next[len]长度的子串完全相同（也就是两条线所对应的位置）。我们再求出next[len]位置处的next值，也就是图中蓝线对应的长度。同样可以得到两个蓝线对应的子串肯定完全相同，又由于第二段蓝线属于左侧红线的后缀，所以又能得到它肯定也是整个字符串的后缀。
所以对于这道题，求出len处的next值，并递归的向下求出所有的next值，得到的就是答案。

一定注意理解next数组的含义。

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<cstring>
#include<string>
#include<vector>
#include<cmath> 
#include<map>
#define mem(a,b) memset(a,b,sizeof(a))
#define mod 1000000007
using namespace std;
typedef long long ll;
const int maxn = 1e6+5;
const double esp = 1e-7;
const int ff = 0x3f3f3f3f;

char s[maxn];
int ne[maxn];
int ans[maxn];

void get_next(char *s,int len)
{
	ne[0] = -1;
	int i = 0,j = -1;
	while(i< len)
	{
		if(j == -1||s[i] == s[j])
		{
			i++;j++;
			ne[i] = j;
		}
		else
			j = ne[j];
	}
}

int main()
{
	while(~scanf(" %s",s))
	{
		mem(ne,0);
		int len = strlen(s),cnt = 1;
		
		get_next(s,len);
		
		ans[1] = len;
		int j = ne[len];
		while(j> 0)//不断取next 
		{
			ans[++cnt] = j;
			j = ne[j];
		}
		
		for(int i = cnt;i> 1;i--)
			printf("%d ",ans[i]);
		printf("%d\n",ans[1]);
		
	}
	return 0;
}