hdoj 4552 怪盗基德的挑战书【求前缀在字符串中出现的次数之和】

怪盗基德的挑战书

Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)
Total Submission(s): 990    Accepted Submission(s):
478

Problem Description

　　“在树最美丽的那天，当时间老人再次把大钟平均分开时，我会降临在灯火之城的金字塔前，带走那最珍贵的笑容。”这是怪盗基德盗取巴黎卢浮宫的《蒙娜丽莎的微笑》这幅画时，挑战书上的内容。
　　但这次，怪盗基德的挑战书上出现了一串串小写字母“aaab
sdfeeddd...”。柯南以小学生的眼睛，超凡高中生的头脑，快速统计各种字母频率，字符串长度，并结合挑战书出现的时间等信息，试图分析怪盗基德的意图。最后，他将线索锁定在字符串的循环次数上。并且进一步推理发现，从字符串的第一位开始，到第i位，形成该字符串的子串(c1,
c2, c3 ... ci )。对于某一子串ci在该字符串中出现的次数记为ki，则全部子串的循环次数总和AIM = k1 + k2 + ... + ki +
... +
kn，柯南发现，AIM恰好对应一个ASCII码！所以，只要把挑战书上的字符串转变成数字，再找到对应的ASCII码，就可以破解这份挑战书了！
　　现在，你的任务就是把字符串转变成对应数字，因为ASCII码以及扩展ASCII码全部只有256个，所以，本题只要把结果对256取余即可。

 

Input

输入有多组测试数据；
每组测试数据只有一个字符串，由各种小写字母组成，中间无空格。
字符串的长度为L(0
< L <= 100000)。

 

Output

请计算并输出字符串的AIM值，每组数据输出一行。

 

Sample Input

aaa

abab

 

Sample Output

6

6

#include<stdio.h>
#include<string.h>
#define MAX 100100
#define mod 256
int next[MAX];
char str[MAX];
int dp[MAX];
int len,s;
void getfail()
{
	int i,j;
	next[0]=next[1]=0;
	for(i=1;i<len;i++)
    {
    	j=next[i];
    	while(j&&str[i]!=str[j])
    	    j=next[j];
    	next[i+1]=str[i]==str[j]?j+1:0;
	}
}
void kmp()
{
	int i,j;
	dp[0]=0;
	for(i=1;i<=len;i++)
	{
		dp[i]=(dp[next[i]]%mod+1)%mod;
		s=(s%mod+dp[i]%mod)%mod;
	}
}
int main()
{
	while(scanf("%s",str)!=EOF)
	{
		s=0;
		len=strlen(str);
		getfail();
		kmp();
		printf("%d\n",s);
	}
	return 0;
}