SPOJ DISUBSTR

Given a string, we need to find the total number of its distinct substrings.

Input

T- number of test cases. T<=20;
Each test case consists of one string, whose length is <= 1000

Output

For each test case output one number saying the number of distinct substrings.

Example

Sample Input:
2
CCCCC
ABABA

Sample Output:
5
9

Explanation for the testcase with string ABABA: 
len=1 : A,B
len=2 : AB,BA
len=3 : ABA,BAB
len=4 : ABAB,BABA
len=5 : ABABA
Thus, total number of distinct substrings is 9.

每个子串一定是某个后缀的前缀， 那么原问题等价于求所有后缀之间的不相

同的前缀的个数。如果所有的后缀按照 suffix(sa[1]), suffix(sa[2]),suffix(sa[3]), ...... ,suffix(sa[n])的顺序计算，不难发现，对于每一次新加进来的后缀 suffix(sa[k]),它将产生 n-sa[k]+1 个新的前缀。但是其中有height[k]个是和前面的字符串的前缀是相同的。 所以 suffix(sa[k])将“贡献”出 n-sa[k]+1- height[k]个不同的子串。累加后便是原问题的答案。这个做法的时间复杂度为 O(n)。 

#include <iostream>#include <string.h>#include <stdio.h>#include <algorithm>#include <queue>#include <math.h>#include <vector>using namespace std;const int maxn = 1111;int cnt[maxn],sa[maxn],height[maxn],res[maxn],rk[maxn],w[maxn];void getSa(int len,int up){    int *k=rk,*r=res,*id=height;    for(int i=0;i<up;i++) cnt[i]=0;    for(int i=0;i<len;i++) cnt[k[i]=w[i]]++;    for(int i=0;i<up;i++) cnt[i+1]+=cnt[i];    for(int i=len-1;i>=0;i--){        sa[--cnt[k[i]]]=i;    }    int d=1,p=0;    while(p<len){        for(int i=len-d;i<len;i++) id[p++]=i;        for(int i=0;i<len;i++) if(sa[i]-d>=0) id[p++]=sa[i]-d;        for(int i=0;i<len;i++) r[i]=k[id[i]];        for(int i=0;i<up;i++) cnt[i]=0;        for(int i=0;i<len;i++) cnt[r[i]]++;        for(int i=0;i<up;i++) cnt[i+1]+=cnt[i];        for(int i=len-1;i>=0;i--){            sa[--cnt[r[i]]]=id[i];        }        p=0;        swap(k,r);        k[sa[0]]=p++;        for(int i=0;i<len-1;i++){            if(sa[i]+d<len&&sa[i+1]+d<len&&r[sa[i]]==r[sa[i+1]]&&r[sa[i+1]+d]==r[sa[i]+d]){                k[sa[i+1]]=p-1;            }            else                k[sa[i+1]]=p++;        }        if(p>=len) return;        d=d*2,up=p,p=0;    }}void getHeight(int len){    for(int i=0;i<len;i++) rk[sa[i]]=i;    height[1]=0;    for(int i=0,p=0;i<len-1;i++){        int j=sa[rk[i]-1];        while(i+p<len&&j+p<len&&w[i+p]==w[j+p])            p++;        height[rk[i]]=p;        p=max(0,p-1);    }}void getSuffix(char s[]){    int len=(int)strlen(s);    for(int i=0;i<len;i++){        w[i]=s[i];    }    w[len++]=0;    getSa(len,400);    getHeight(len);}void doit(char s[]){    int len=strlen(s);    int ans=0;    for(int i=1;i<=len;i++){        ans+=len-sa[i]-height[i];    }    printf("%d\n",ans);}int main(){    int T;    scanf("%d",&T);    for(int i=1;i<=T;i++){        char s[maxn];        scanf("%s",s);        getSuffix(s);        doit(s);    }}