C – Distinct Substrings （模板）

https://vjudge.net/problem/SPOJ-DISUBSTR

有两种方式来求去除重读的子串

 #include <bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 const int maxn = ; int t1[maxn],t2[maxn],c[maxn];
 bool cmp(int *r,int a,int b,int l)  {
     return r[a]==r[b] &&r[l+a] == r[l+b];
 } void get_sa(int str[],int sa[],int Rank[],int height[],int n,int m)
 {
     n++;
     int p,*x=t1,*y=t2;
     for(int i = ; i < m; i++) c[i] = ;
     for(int i = ; i < n; i++) c[x[i] = str[i]]++;
     for(int i = ; i < m; i++) c[i] += c[i-];
     for(int i = n-; i>=; i--) sa[--c[x[i]]] = i;
     for(int j = ; j <= n; j <<= )  {
         p = ;
         for(int i = n-j; i < n; i++) y[p++] = i;
         for(int i = ; i < n; i++) if(sa[i] >= j) y[p++] = sa[i]-j;
         for(int i = ; i < m; i++) c[i] = ;
         for(int i = ; i < n; i++) c[x[y[i]]]++ ;
         for(int i = ; i < m; i++) c[i] += c[i-];
         for(int i = n-; i >= ; i--)  sa[--c[x[y[i]]]] = y[i];
         swap(x,y);
         p = ;
         x[sa[]] = ;
         for(int i = ; i < n; i++)
             x[sa[i]] = cmp(y,sa[i-],sa[i],j)? p-:p++;
         if(p >= n) break;
         m = p;
     }
     int k = ;
     n--;
     for(int i = ; i <= n; i++)
         Rank[sa[i]] = i;
     for(int i = ; i < n; i++)  {
         if(k) k--;
         int j = sa[Rank[i]-];
         while(str[i+k] == str[j+k]) k++;
         height[Rank[i]] = k;
     }
 } int Rank[maxn],height[maxn];
 int sa[maxn];
 char str[maxn];
 int a[maxn];
 int n; int main()
 {
     int T;
     scanf("%d",&T);
     while(T--)
     {
         int tot = ;
         scanf("%s",str);
         int len = strlen(str);
         for(int i = ;i < len; i++)
             a[i] = str[i];
         a[len] = ;
         get_sa(a,sa,Rank,height,len,);
         long long res = ;
         long long ans = (len+)*len/;
         for(int i=;i<=len;i++){
             res+=len-sa[i]-height[i];
             ans-=height[i];
         }
         //printf("%lld\n",res);
         printf("%lld\n", ans);
     }
     return ;
 }