HDU - 4552 怪盗基德的挑战书

点击打开链接

“在树最美丽的那天，当时间老人再次把大钟平均分开时，我会降临在灯火之城的金字塔前，带走那最珍贵的笑容。”这是怪盗基德盗取巴黎卢浮宫的《蒙娜丽莎的微笑》这幅画时，挑战书上的内容。 
　　但这次，怪盗基德的挑战书上出现了一串串小写字母“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<stdlib.h>
#include<string.h>
int next[200100];
char s[200100];
int dp[200100];

int getnext(char* s,int n)
{
	int i=0,j=0;
	next[0] = -1;
	j = next[i];
	while(i<n)
	{
		if(j == -1 || s[i] == s[j])
		{
			i++;
			j++;
			next[i]=j;			
		}
		else
		{
		   j=next[j];
		}
	}
}
int main()
{
	while(scanf("%s",s)!=EOF)
	{	
	   int n=strlen(s);	    	     	   
	   getnext(s,n); 	    
		int sum=0;
		memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
		{
            dp[i]=(dp[next[i]]+1);
         //   printf("i:%d %d %d %d\n",i,next[i],dp[next[i]],dp[i]);
            sum=(sum+dp[i])%256;
        }
        printf("%d\n",sum);
	} 	
	return 0;
}