hdu 4552 怪盗基德的挑战书

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4552

思路：

其实可以用KMP+DP解决

我这里给出后缀数组的解题方法

思路：

初始化ans=n(n为字符串的长度)

然后你会发现答案就是ans加上每个后缀与最长后缀的最长公共前缀的长度

代码：

#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
using namespace std;
#define maxn 100010
int wa[maxn],wv[maxn],wb[maxn],wq[maxn];
int height[maxn],rank[maxn],sa[maxn];
int r[maxn];
char s[maxn];
int 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,*t;
    for(i=0;i<m;i++) wq[i]=0;
    for(i=0;i<n;i++) wq[x[i]=r[i]]++;
    for(i=1;i<m;i++) wq[i]+=wq[i-1];
    for(i=n-1;i>=0;i--) sa[--wq[x[i]]]=i;

    for(j=1,p=1;p<n;j*=2,m=p)
    {
       for(p=0,i=n-j;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++) wq[i]=0;
       for(i=0;i<n;i++) wq[wv[i]]++;
       for(i=1;i<m;i++) wq[i]+=wq[i-1];
       for(i=n-1;i>=0;i--) sa[--wq[wv[i]]]=y[i];
       for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
               x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;

    }
    return ;
}
void callheight(int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) rank[sa[i]]=i;
    for(i=0;i<n;i++)
    {
        if(k)k--;
        j=sa[rank[i]-1];
        while(r[i+k]==r[j+k]) k++;
        height[rank[i]]=k;
    }
}
int main()
{
 int Case=0;
 while(scanf("%s",s)!=EOF)
 {
     n=strlen(s);
     for(int i=0;i<n;i++)
             r[i]=s[i];
             r[n]=0;
     da(n+1,130);
     callheight(n);
     int ans=n;
     int t=rank[0];
     int temp=n;
     while(t<n)//加上排名在t后面的后缀与最长后缀的公共前缀长度
     {
         temp=min(temp,height[t+1]);
         ans+=temp;
         t++;
     }
     t=rank[0];
     temp=n;
     while(t>1)//加上排名在t前面的后缀与最长后缀的公共前缀的长度
     {

         temp=min(temp,height[t]);
         ans+=temp;
         t--;
     }
     cout<<ans%256<<endl;
 }
 return 0;
}