HDU 4552 怪盗基德的挑战书

怪盗基德的挑战书

Time Limit: 1000ms

Memory Limit: 32768KB

This problem will be judged on HDU. Original ID: 4552
64-bit integer IO format: %I64d      Java class name: Main

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

Source

2013金山西山居创意游戏程序挑战赛——初赛（3）

解题：求每个后缀与原串的LCP

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 const int maxn = 100010;
 4 char s[maxn];
 5 int sa[maxn],rk[maxn],height[maxn],n;
 6 int c[maxn],tmp[2][maxn];
 7 void build_sa(int m) {
 8     int i,*x = tmp[0],*y = tmp[1];
 9     for(i = 0; i < m; ++i) c[i] = 0;
10     for(i = 0; i < n; ++i) c[x[i] = s[i]]++;
11     for(i = 1; i < m; ++i) c[i] += c[i-1];
12     for(i = n-1; i >= 0; --i) sa[--c[x[i]]] = i;
13
14     for(int k = 1; k <= n; k <<= 1) {
15         int p = 0;
16         for(i = 0; i < m; ++i) c[i] = 0;
17         for(i = n - k; i < n; ++i) y[p++] = i;
18         for(i = 0; i < n; ++i) if(sa[i] >= k) y[p++] = sa[i]-k;
19         for(i = 0; i < n; ++i) c[x[y[i]]]++;
20         for(i = 1; i < m; ++i) c[i] += c[i-1];
21         for(i = n - 1; i >= 0; --i) sa[--c[x[y[i]]]] = y[i];
22         swap(x,y);
23         x[sa[0]] = 0;
24         for(p = i = 1; i < n; ++i)
25             if(y[sa[i]] == y[sa[i-1]] && y[sa[i]+k] == y[sa[i-1]+k])
26                 x[sa[i]] = p-1;
27             else x[sa[i]] = p++;
28         if((m = p) >= n) break;
29     }
30 }
31 void getHeight() {
32     int i,j,k = 0;
33     for(i = 0; i < n; ++i) rk[sa[i]] = i;
34     for(i = 0; i < n; ++i) {
35         if(k) --k;
36         j = sa[rk[i]-1];
37         while(i + k < n && j + k < n && s[i+k] == s[j+k]) ++k;
38         height[rk[i]] = k;
39     }
40 }
41 int main() {
42     while(~scanf("%s",s)) {
43         n = strlen(s) + 1;
44         build_sa(128);
45         getHeight();
46         int pos = rk[0],tmp = n-1,ret = n-1;
47         while(pos + 1 < n && tmp) {
48             tmp = min(tmp,height[++pos]);
49             ret += tmp;
50         }
51         pos = rk[0],tmp = n-1;
52         while(pos > 1 && tmp) {
53             tmp = min(tmp,height[pos--]);
54             ret += tmp;
55         }
56         printf("%d\n",ret%256);
57     }
58     return 0;
59 }