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

 

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100010;
char s[maxn];
int sa[maxn],rk[maxn],height[maxn],n;
int c[maxn],tmp[2][maxn];
void build_sa(int m) {
    int i,*x = tmp[0],*y = tmp[1];
    for(i = 0; i < m; ++i) c[i] = 0;
    for(i = 0; i < n; ++i) c[x[i] = s[i]]++;
    for(i = 1; i < m; ++i) c[i] += c[i-1];
    for(i = n-1; i >= 0; --i) sa[--c[x[i]]] = i;

    for(int k = 1; k <= n; k <<= 1) {
        int p = 0;
        for(i = 0; i < m; ++i) c[i] = 0;
        for(i = n - k; i < n; ++i) y[p++] = i;
        for(i = 0; i < n; ++i) if(sa[i] >= k) y[p++] = sa[i]-k;
        for(i = 0; i < n; ++i) c[x[y[i]]]++;
        for(i = 1; i < m; ++i) c[i] += c[i-1];
        for(i = n - 1; i >= 0; --i) sa[--c[x[y[i]]]] = y[i];
        swap(x,y);
        x[sa[0]] = 0;
        for(p = i = 1; i < n; ++i)
            if(y[sa[i]] == y[sa[i-1]] && y[sa[i]+k] == y[sa[i-1]+k])
                x[sa[i]] = p-1;
            else x[sa[i]] = p++;
        if((m = p) >= n) break;
    }
}
void getHeight() {
    int i,j,k = 0;
    for(i = 0; i < n; ++i) rk[sa[i]] = i;
    for(i = 0; i < n; ++i) {
        if(k) --k;
        j = sa[rk[i]-1];
        while(i + k < n && j + k < n && s[i+k] == s[j+k]) ++k;
        height[rk[i]] = k;
    }
}
int main() {
    while(~scanf("%s",s)) {
        n = strlen(s) + 1;
        build_sa(128);
        getHeight();
        int pos = rk[0],tmp = n-1,ret = n-1;
        while(pos + 1 < n && tmp) {
            tmp = min(tmp,height[++pos]);
            ret += tmp;
        }
        pos = rk[0],tmp = n-1;
        while(pos > 1 && tmp) {
            tmp = min(tmp,height[pos--]);
            ret += tmp;
        }
        printf("%d\n",ret%256);
    }
    return 0;
}