Distinct Substrings （ｓｐｏｊ694）（后缀数组）

核心：每一个后缀的每一个前缀都是一个子串

相邻后缀的ｌｃｐ的值是就是重复子串的个数

用所有的子串减去ｓｉｇｍａ（ｈｅｉｇｈｔ）就行了

 

#include"bits/stdc++.h"
using namespace std;
#define clr(x) memset(x,0,sizeof(x))

int T;
char a[2000];
int n,m;
int x[2000],y[2000],c[2000];
int sa[2000];
int rk[2000];
int ht[2000];

inline void SA()
{
   //cout<<"%^("<<endl;

   // cout<<n<<" "<<m<<endl;
 //cout<<a+1<<endl;
   /* memset(c,0,sizeof c);
    memset(x,0,sizeof x);
    memset(y,0,sizeof y);*/
    clr(c);
    clr(x);
    clr(y);


   for (int i=1;i<=n;i++) ++c[x[i] = a[i]-'A'];
   for (int i=1;i<=m;i++) c[i]+=c[i-1];
   for (int i=n;i;i--) sa[c[x[i]]--] = i;

   for (int k=1;k<=n;k*=2)
   {

      int num = 0;
      for (int i=n-k+1;i<=n;i++) y[++num] = i;
      for (int i=1;i<=n;i++) if (sa[i]>k) y[++num] = sa[i]-k;
      memset(c,0,sizeof c);
      for (int i=1;i<=n;i++) ++c[x[y[i]]];
      for (int i=1;i<=m;i++) c[i]+=c[i-1];
      for (int i=n;i;i--) sa[c[x[y[i]]]--] = y[i],y[i]=0;
      num=1;
      swap(x,y);
      x[sa[1]]=1;

      for (int i=2;i<=n;i++)
      {
        x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?num:++num;
      }

      if (num==n) break;
      m=num;


   }

   for (int i=1;i<=n;i++) rk[sa[i]] = i;
   int k = 0;

   for (int i=1;i<=n;i++)
   {
      if (rk[i]==1) continue;
      if (k) k--;
      int j =sa[rk[i]-1];


      while (a[i+k]==a[j+k]&&i+k<=n&&j+k<=n) k++;

      ht[rk[i]] = k;
   }

//
   //  for (int i=1;i<=n;i++ ) cout<<ht[i]<<" ";cout<<endl;
  // /   / for (int i=1;i<=n;i++) cout<<sa[i]<<" ";cout<<endl;


}



int main()
{
  cin>>T;
  while (T--)
  {

    m=233;
    cin>>a+1;
    n = strlen(a+1);
    SA();

  int ans = n*(n+1)/2;
  for (int i=1;i<=n;i++ ) ans -= ht[i];
  cout<<ans<<endl;



  }