1458:Seek the Name, Seek the Fame
时间限制: 1000 ms 内存限制: 65536 KB
提交数: 890 通过数: 489
【题目描述】
原题来自:POJ 2752
给定若干字符串(这些字符串总长 ≤4×105 ),在每个字符串中求出所有既是前缀又是后缀的子串长度。
例如:ababcababababcabab,既是前缀又是后缀的:ab,abab,ababcabab,ababcababababcabab。
【输入】
输入若干行,每行一个字符串。
【输出】
对于每个字符串,输出一行,包含若干个递增的整数,表示所有既是前缀又是后缀的子串长度。
【输入样例】
ababcababababcabab
aaaaa
【输出样例】
2 4 9 18
1 2 3 4 5
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define N 400010
#define ull unsigned long long
ull sum[N],power[N];
#define p 41
char s[N];
int l;
bool check(int i)
{
if(sum[i-1]*power[l-i+1]==sum[l]-sum[l-i+1]) return 1;
return 0;
}
int main()
{
power[0]=1;
for(int i=1;i<=N;i++)
{
power[i]=power[i-1]*p;
}
while(~scanf("%s",s+1))
{
l=strlen(s+1);
memset(sum,0,sizeof(sum));
for(int i=1;i<=l;i++)
{
sum[i]=sum[i-1]*p+(ull)(s[i]-'A'+1);
}
for(int i=l;i>=1;i--)
{
if(check(i)) printf("%d ",l-i+1);
}
putchar('\n');
}
return 0;
}