【spoj705】 Distinct Substrings

【题目描述】

给定一个字符串，计算其不同的子串个数。

【输入格式】

一行一个仅包含大写字母的字符串，长度<=50000

【输出格式】

一行一个正整数，即不同的子串个数。

【样例输入】

ABABA

【样例输出】

9

【思路】
　　一看就知道是后缀数组题啦～但是我不会写QAQ。。只好现学现用啦～
　　在字符串最后补上一个'$',不因为别的只因为它比‘A’还要小。。不然你补ascII码是0的也可以。。
　　申请rank数组和sa数组，rank[i]=j代表后缀i排第j位，sa[i]=j代表排名第i的是后缀j。也就是说rank和sa是相反的运算。
　　首先将sa数组按照单字母的顺序排个序，更新rank数组，不过记得字母相同的排名也要相同，也就是如果str[sa[i]]==str[sa[i-1]]的话rank[sa[i]]=rank[sa[i-1]];else rank[sa[i]]=rank[sa[i-1]]+1;
　　然后将k从0开始枚举，每次继续对sa进行排序，但是是以rank[sa[i]]为第一关键字，rank[sa[i]+2k]为第二关键字排序。
　　在Trank中（其实就是tmpRank）更新好新的rank值，一样记得如果rank[sa[i]]==rank[sa[i-1]]&&rank[sa[i]+2k]==rank[sa[i-1]+2k]的话排名不要上升。
　　重复这一步，直到2k>=n或者所有后缀的排名都不同。
　　然后正常情况下k增加logN次，每次如果用计数排序只要O(N),一共O(NlogN)。
　　但是不会写计数排序啊QAQ。。所以用快排好了。。多加一个log，一般不会被卡的吧。。
　　计算出来sa和rank之后还要计算height数组，height[i]代表sa[i]和sa[i-1]的最长公共前缀，如果按照1——n的顺序计算的话是O(N2)的，显然不够优秀，于是我们按照一种奇怪的顺序计算。
　　先算height[rank[1]]，然后是height[rank[2]]……
　　这样的话就会有一个性质：height[rank[i]]>=height[rank[i-1]]-1
　　我也不知道为什么但是就是这样的。。
　　然后就没有然后了。。
　　我们知道任何一个子串都是某一个后缀的一个前缀
　　对于后缀i来说，有length-i个前缀，其中有height[i]个和前一个后缀相同
　　所以答案就是Σlength-i-height[i]

 #include <iostream>
 #include <cstring>
 #include <string>
 #include <cstdio>
 #include <cstdlib>
 #include <cmath>
 #include <algorithm>
 #include <queue>
 #include <stack>
 #include <map>
 #include <set>
 #include <list>
 #include <vector>
 #include <ctime>
 #include <functional>
 #define pritnf printf
 #define scafn scanf
 #define sacnf scanf
 #define For(i,j,k) for(int i=(j);i<=(k);(i)++)
 #define Clear(a) memset(a,0,sizeof(a))
 using namespace std;
 typedef unsigned int Uint;
 const int INF=0x3fffffff;
 const double eps=1e-;
 ///==============struct declaration==============

 ///==============var declaration=================
 const int MAXN=;
 int n,k;
 int sa[MAXN],rank[MAXN],h[MAXN],trank[MAXN],height[MAXN];
 char str[MAXN];
 ///==============function declaration============
 bool cmp(int a,int b){return rank[a]==rank[b]?rank[a+(<<k)]<rank[b+(<<k)]:rank[a]<rank[b];}
 bool cmp1(int a,int b){return str[a]<str[b];}
 ///==============main code=======================
 int main()
 {
     scanf("%s",str+);
     n=strlen(str+);str[n+]='$';n++;
     for(int i=;i<=n;i++)
         sa[i]=i;
     sort(sa+,sa++n,cmp1);
     rank[sa[]]=;
     for(int i=;i<=n;i++)
         if (str[sa[i]]!=str[sa[i-]])
             rank[sa[i]]=rank[sa[i-]]+;
         else
             rank[sa[i]]=rank[sa[i-]];
     for(k=;(<<k)<=n;k++){
         sort(sa+,sa++n,cmp);
         trank[sa[]]=;
         for(int i=;i<=n;i++){
             if (rank[sa[i]]!=rank[sa[i-]]||rank[sa[i]+(<<k)]!=rank[sa[i-]+(<<k)])
                 trank[sa[i]]=trank[sa[i-]]+;
             else
                 trank[sa[i]]=trank[sa[i-]];
         }
         for(int i=;i<=n;i++)
             rank[i]=trank[i];
         if (rank[sa[n]]==n) break;
     }
     for(int i=;i<=n;i++)
         rank[sa[i]]=i;
     ///height[i]表示sa[rank[i]]和sa[rank[i-1]]的最长前缀
     height[rank[]]=;
     for(int i=;i<=n;i++){
         height[rank[i]]=max(height[rank[i-]]-,);
         int p=i,q=sa[rank[i]-];
         while (str[p+height[rank[i]]]==str[q+height[rank[i]]])
             height[rank[i]]++;
     }
     long long ans=;
     for(int i=;i<=n;i++)
         ans+=n-sa[i]-height[i];
       printf("%lld\n",ans);
    return ;
 }
 ///================fuction code====================

Spoj 705

　　　尽量多做后缀数组的题目吧，熟能生巧，现在只是刚刚学会了后缀数组，还要多多加油才是。