[后缀数组+RMQ] hdu 4552 怪盗基德的挑战书

题意：求所有前缀在字符串里出现的次数和mod256.

思路：求所有后缀与最长后缀（字符串本身）的lcp和mod256

代码：

#include"cstdlib"
#include"cstdio"
#include"cstring"
#include"cmath"
#include"queue"
#include"algorithm"
#include"iostream"
#include"map"
using namespace std;
#define N 100012
int v[N],ra[N],sa[N],height[N],dp[N][30],Log[N];
int wa[N],wb[N],wv[N],wws[N];
char fuck[N];
int cmp(int *r,int a,int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int n,int m)
{
    int i,j,p,*x=wa,*y=wb;
    for(i=0; i<m; i++) wws[i]=0;
    for(i=0; i<n; i++) wws[x[i]=v[i]]++;
    for(i=1; i<m; i++) wws[i]+=wws[i-1];
    for(i=n-1; i>=0; i--) sa[--wws[x[i]]]=i;
    for(j=1,p=1; p<n; j*=2,m=p)
    {
        for(i=n-j,p=0; i<n; i++) y[p++]=i;
        for(i=0; i<n; i++) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(i=0; i<n; i++) wv[i]=x[y[i]];
        for(i=0; i<m; i++) wws[i]=0;
        for(i=0; i<n; i++) wws[wv[i]]++;
        for(i=1; i<m; i++) wws[i]+=wws[i-1];
        for(i=n-1; i>=0; i--) sa[--wws[wv[i]]]=y[i];
        for(swap(x,y),i=1,p=1,x[sa[0]]=0; i<n; i++) x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
    }
    return ;
}
void gethei(int n)
{
    int i,j,k=0;
    for(i=1; i<=n; i++) ra[sa[i]]=i;
    for(i=0; i<n; i++)
    {
        if(k) k--;
        j=sa[ra[i]-1];
        while(v[i+k]==v[j+k]) k++;
        height[ra[i]]=k;
    }
    return ;
}
void rmqinit(int n)
{
    int i,j;
    int m=Log[n];
    for(i=1; i<=n; i++) dp[i][0]=height[i];
    for(i=1; i<=m; i++)
    {
        for(j=1; j+(1<<i)-1<=n; j++)
            dp[j][i]=min(dp[j][i-1],dp[j+(1<<(i-1))][i-1]);
    }
    return ;
}
int lcp(int a,int b)
{
    a=ra[a];
    b=ra[b];
    if(a>b) swap(a,b);
    a++;
    int m=Log[b-a+1];
    return min(dp[a][m],dp[b-(1<<m)+1][m]);
}
int main()
{
    Log[0]=-1;
    for(int i=1; i<=N; i++) Log[i]=(i&(i-1))?Log[i-1]:Log[i-1]+1;
    while(scanf("%s",fuck)!=-1)
    {
        int i;
        int len=strlen(fuck);
        for(i=0; i<len; i++) v[i]=fuck[i]-'a'+2;
        v[len]=0;
        da(len+1,40);
        gethei(len);
        rmqinit(len);
        int ans=len;
        for(i=1; i<len; i++)
            ans+=lcp(i,0);
        printf("%d\n",ans%256);
    }
    return 0;
}