hdoj 4552 怪盗基德的挑战书 【KMP】

怪盗基德的挑战书

Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)

Total Submission(s): 1105    Accepted Submission(s): 518

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
   
   
   
   

 

求子串在原串中出现的次数总和。

AC代码：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#define mem(a, b) memset(a, (b), sizeof(a))
#define Wi(a) while(a--)
#define Si(a) scanf("%d", &a)
#define Pi(a) printf("%d\n", (a))
#define INF 0x3f3f3f3f
#define mod 256
#include<algorithm>
#define N 100000+10
using namespace std;
char s[N];
int p[N];
int num[N];
void  getp()
{
	int len = strlen(s);
	int i = 0;
	int j = -1;
	p[0] = -1;
	while(i < len)
	{
		if(j==-1||s[i]==s[j])
			p[++i] = ++j;
		else  j = p[j];	
	}
}
int main()
{
	while(scanf("%s", s)!=EOF)
	{
		getp();
		int len = strlen(s);
		int sum = 0;
		num[0] = 0;
		for(int i = 1; i <= len; i++)
		{
			num[i] = (num[p[i]]%mod +1)%mod;
			sum = (sum%mod+num[i])%mod;
		}
		Pi(sum);
	}
	return 0;
}