SPOJ 694 Distinct Substrings(尾碼數組 所有不相同子串個數)

題目連結：http://www.spoj.com/problems/DISUBSTR/

SPOJ 上還有一題，除了資料範圍不一樣其他的就都行：http://www.spoj.com/problems/SUBST1/

把下面程式的maxn改成60000就OK了！

第一次做spoj上的題目

這個題目還是利用height數組，表示尾碼數組理解還是靠height數組的理解

這個題目其實思路還是不太容易想到的，但是一下想到了就會變的暴簡單

開始排名第一的串先上，那麼這個串一定會產生 n-sa[1]個首碼，也就是n-sa[1]個不同的子串

完成之後從第二個開始找

那麼如果不考慮重複的話第二個會產生n-sa[2]個子串，但是和前面一定是重複了height[2]個，因為前面height個是重複的

接著就是排名第i個串一定會產生n-sa[i]+height[i]個和前面所有串都不相同的子串

然後把所有的結果全部加起來就是這個題目的答案

#include <iostream>#include <stdio.h>#include <algorithm>#include <math.h>#include <string.h>using namespace std;#define maxn 1100#define ws ws1int wa[maxn],wb[maxn],wv[maxn],ws[maxn];int cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}void da(char *r,int *sa,int n,int m){int i,j,p,*x=wa,*y=wb,*t;for(i=0;i<m;i++) ws[i]=0;for(i=0;i<n;i++) ws[x[i]=r[i]]++;for(i=1;i<m;i++) ws[i]+=ws[i-1];for(i=n-1;i>=0;i--) sa[--ws[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++) ws[i]=0;for(i=0;i<n;i++) ws[wv[i]]++;for(i=1;i<m;i++) ws[i]+=ws[i-1];for(i=n-1;i>=0;i--) sa[--ws[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;}int rank[maxn],height[maxn];void calheight(char *r,int *sa,int n){int i,j,k=0;for(i=1;i<=n;i++) rank[sa[i]]=i;//在計算height的時候順便就把rank計算出來了，反正也要用for(i=0;i<n;height[rank[i++]]=k)for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);return;}char rec[maxn];int sa[maxn];int n;int find_ans(){    int i;    int ans=n-sa[1];    for(i=2;i<=n;i++)//從1開始，因為最小的是空串    {        ans+=n-sa[i]-height[i];    }    return ans;}int main(){    int i,j,k;    int left,right;    int mid;    int t;    scanf("%d",&t);    while(t--)    {    scanf("%s",rec);        if(strcmp("#",rec)==0)        return 0;        n=strlen(rec);     da(rec,sa,n+1,200);//這裡是n+1 因為看這個函數裡面是 < n 的     calheight(rec,sa,n);//注意這裡面是 n 了因為看函數裡面是 <=n 的，所以這裡要注意    if(n==1)    {        printf("1\n");        continue;    }    printf("%d\n",find_ans());    }    return 0;}