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=;i<m;i++) wq[i]=;
     for(i=;i<n;i++) wq[x[i]=r[i]]++;
     for(i=;i<m;i++) wq[i]+=wq[i-];
     for(i=n-;i>=;i--) sa[--wq[x[i]]]=i;

     for(j=,p=;p<n;j*=,m=p)
     {
        for(p=,i=n-j;i<n;i++) y[p++]=i;
        for(i=;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(i=;i<n;i++) wv[i]=x[y[i]];
        for(i=;i<m;i++) wq[i]=;
        for(i=;i<n;i++) wq[wv[i]]++;
        for(i=;i<m;i++) wq[i]+=wq[i-];
        for(i=n-;i>=;i--) sa[--wq[wv[i]]]=y[i];
        for(t=x,x=y,y=t,p=,x[sa[]]=,i=;i<n;i++)
                x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;

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

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