HDU 4552 怪盗基德的挑战书 （后缀数组，4级）

B - 怪盗基德的挑战书
Crawling in process...Crawling failedTime
Limit:1000MS Memory Limit:32768KB
64bit IO Format:%I64d & %I64u
SubmitStatus

Appoint description:
System Crawler (2013-05-30)

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
求所有后缀和原串的最长公共前缀和。后缀数组求出h,得到原串的rank 左右扫一遍，后RMQ直接搞。
#include<cstring>
#include<cstdio>
#include<iostream>
#define FOR(i,a,b) for(int i=a;i<=b;++i)
#define clr(f,z) memset(f,z,sizeof(f))
#define ll(x) (1<<x)
using namespace std;
const int msize=1e5+9;
const int sig=257;
class SUFFIX_ARRAY
{
public:
int ran[msize],t1[msize],c[msize],sa[msize];
int n;
int idx(char x)
{
if(x=='\0')return 0;
return x-'a'+1;
}
bool cmp(int*r,int i,int k)
{
return r[ sa[i] ]==r[ sa[i-1] ]&&r[ sa[i]+k ]==r[ sa[i-1]+k ];
}
void build_SA(char*s,int m)
{ int*wx=t1,*wy=ran;
n=strlen(s)+1;
FOR(i,0,m-1)c[i]=0;
FOR(i,0,n-1)c[ wx[i]=idx(s[i]) ]++;
FOR(i,1,m-1)c[i]+=c[i-1];
for(int i=n-1;i>=0;--i)sa[ --c[ wx[i] ] ]=i;
for(int k=1;k<=n;k<<=1)
{
int p=0;
FOR(i,n-k,n-1)wy[p++]=i;//二关键字排序
FOR(i,0,n-1)if(sa[i]>=k)wy[p++]=sa[i]-k;
FOR(i,0,m-1)c[i]=0;
FOR(i,0,n-1)++c[ wx[ wy[i] ] ];
FOR(i,1,m-1)c[i]+=c[i-1];
for(int i=n-1;i>=0;--i)sa[ --c[ wx[ wy[i] ] ] ]=wy[i];
swap(wx,wy);
wx[ sa[0] ]=0;
p=1;
FOR(i,1,n-1)wx[ sa[i] ]=cmp(wy,i,k)?p-1:p++;
if(p>=n)break;
m=p;
}
--n;
}
int h[msize];
void get_H(char*s)
{ int k=0;
FOR(i,0,n)ran[ sa[i] ]=i;
FOR(i,0,n-1)
{
if(k)--k;
int j=sa[ ran[i]-1 ];
while( s[i+k]==s[j+k] )++k;
h[ ran[i] ]=k;
}
}
void debug()
{ printf("sa=");
FOR(i,0,n)printf("%d ",sa[i]);puts("");
printf("rank=");
FOR(i,0,n)printf("%d ",ran[i]);puts("");
printf("h=");
FOR(i,0,n)printf("%d ",h[i]);puts("");
}
int rmq[msize][32],bit[msize];
void initRMQ()
{
bit[0]=-1;
FOR(i,1,n)bit[i]=(i&(i-1))==0?bit[i-1]+1:bit[i-1];
FOR(i,1,n)rmq[i][0]=h[i];
FOR(i,1,bit
)
for(int j=1;j+ll(i)-1<=n;++j)
rmq[j][i]=min(rmq[j][i-1],rmq[j+ll(i-1)][i-1]);
}
int LCP(int l,int r)
{
l=ran[l];r=ran[r];
if(l>r)swap(l,r);
++l;//之前一个已经求过了。
int t=bit[r-l+1];
r-=ll(t)-1;
return min(rmq[l][t],rmq[r][t]);
}
};
SUFFIX_ARRAY ty;
char s[msize];
int main()
{
while(~scanf("%s",s))
{
ty.build_SA(s,257);
ty.get_H(s);
ty.initRMQ();
int hh=1e8;
int ans=0;
for(int i=1;s[i];++i)
{
ans+=ty.LCP(0,i);
ans%=256;
}
ans+=strlen(s);
ans%=256;
printf("%d\n",ans);
}
}