New Distinct Substrings(SAM)

New Distinct Substrings(SAM)

求本质不同子串个数，模板题辣。

严重怀疑此题数据有数字，害我 W A WA WA了好久

​​传送门​​

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5,M=2e4+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a) memset(a,0,sizeof a)
#define lx x<<1
#define rx x<<1|1
#define reg register
#define PII pair<int,int>
#define fi first
#define se second
#define pb push_back
char s[N];
int a[N],b[N],n;
struct SAM{
  int last,cnt;int ch[N<<1][130],fa[N<<1],len[N<<1];
  ll dp[N];
  void clear(){
    for(int i=0;i<N;i++)
       a[i]=b[i]=fa[i]=len[i]=dp[i]=0;
    mst(ch);
    last=cnt=1;
    }
  void insert(int c){
    int p=last,np=++cnt;last=np;len[np]=len[p]+1;
    for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
    if(!p) fa[np]=1;
    else {
      int q=ch[p][c];
      if(len[q]==len[p]+1) fa[np]=q;
      else  {
        int nq=++cnt;len[nq]=len[p]+1;
        memcpy(ch[nq],ch[q],sizeof ch[q]);
        fa[nq]=fa[q],fa[q]=fa[np]=nq;
        for(;ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
      }
    }
  }
  void build(){
    scanf("%s",s+1);
    for(int i=1;s[i]!='\0';i++) insert(s[i]);
  }
  void solve(){
    ll ans=0;
    for(int i=1;i<=cnt;i++) b[len[i]]++;
    for(int i=1;i<=cnt;i++) b[i]+=b[i-1];
    for(int i=1;i<=cnt;i++) a[b[len[i]]--]=i;
    for(int i=cnt;i;i--){
      int p=a[i];
      dp[p]=1;
      for(int i=0;i<128;i++) if(ch[p][i]) dp[p]+=dp[ch[p][i]];
    }
    printf("%lld\n",dp[1]-1);
  }
}sam;
int main(){
  int t;
  scanf("%d",&t);
  while(t--){
  sam.clear();
  sam.build();
  sam.solve();
  }
  return 0;
}