spoj 694 Distinct Substrings

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.

 

题目大意： 给你一个字符串，问互补相同的字串有几个。 

我们知道每一个字串都是后缀的前缀，所以我们只要先求出后缀数组，然后对于排名相邻的两个串，比如排名为第i的串，他提供的前缀数量是len-sa[i]，但由于有H[i]个和排名第i-1的是一样的，所以我们要减去h[i]，这样扫一遍就可以了。  

#include<bits/stdc++.h>
using namespace std; 
int const N=1000+3;  
int wa[N<<1],wb[N<<1],wv[N],rk[N],h[N],num[N],sa[N];    
char s[N]; 
int cmp(int *r,int x,int y,int z){
    return r[x]==r[y] && r[x+z]==r[y+z];  
} 
void build_sa(char *r,int *sa,int n,int m){
  int i,j,p,*x=wa,*y=wb;  
  for(i=0;i<m;i++) num[i]=0;  
  for(i=0;i<n;i++) num[x[i]=r[i]]++;  
  for(i=1;i<m;i++) num[i]+=num[i-1];  
  for(i=n-1;i>=0;i--) sa[--num[x[i]]]=i; 
  for(j=1,p=1;p<n;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<m;i++) num[i]=0; 
    for(i=0;i<n;i++) num[wv[i]=x[y[i]]]++;  
    for(i=1;i<m;i++) num[i]+=num[i-1];  
    for(i=n-1;i>=0;i--) sa[--num[wv[i]]]=y[i]; 
    swap(x,y); 
    for(i=1,p=1,x[sa[0]]=0;i<n;i++) 
      x[sa[i]]=cmp(y,sa[i],sa[i-1],j)? p-1:p++; 
  }
  for(i=0;i<n;i++) rk[i]=x[i]; 
}
void build_h(char *r,int *sa,int n){    
    int k=0; 
    for(int i=0;i<n;i++){
        if(k) k--; 
        int j=sa[rk[i]-1];  
        while (r[i+k]==r[j+k]) k++; 
        h[rk[i]]=k;
    }
}  
void solve(int n){
    int ans=0; 
    for(int i=1;i<=n;i++){
        int num=n-sa[i]; 
         ans+=num-h[i];  
    } 
    printf("%d\n",ans); 
}   
int main(){
  int cas;   
  scanf("%d",&cas);  
  while (cas--){
    scanf("%s",s);  
    int len=strlen(s);  
    build_sa(s,sa,len+1,130);   
    build_h(s,sa,len); 
    solve(len);  
  }
  return 0; 
}