p2408 不同子串个数

传送门

分析

首先我们不难求出一共有多少子串

之后我们只需要减掉重复个数即可

于是我们对于每个后缀减去它跟它前一名的最长公共前缀即可

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
#define int long long
inline int max(int x,int y){return x>y?x:y;}
char s[];
int Ans,n,m=,rk[],tp[],sa[],tax[],h[];
inline void qsort(){
    for(int i=;i<=m;i++)tax[i]=;
    for(int i=;i<=n;i++)tax[rk[i]]++;
    for(int i=;i<=m;i++)tax[i]+=tax[i-];
    for(int i=n;i>;i--)sa[tax[rk[tp[i]]]--]=tp[i];
}
inline void work(){
    for(int i=;i<=n;i++)rk[i]=s[i]-''+,tp[i]=i;qsort();
    for(int w=,p=;p<n;m=p,w<<=){
      p=;for(int i=;i<=w;i++)tp[++p]=n-w+i;
      for(int i=;i<=n;i++)if(sa[i]>w)tp[++p]=sa[i]-w;
      qsort();swap(rk,tp);rk[sa[]]=p=;
      for(int i=;i<=n;i++)rk[sa[i]]=(tp[sa[i-]]==tp[sa[i]]&&tp[sa[i-]+w]==tp[sa[i]+w])?p:++p;
    }
    for(int i=;i<=n;++i)rk[sa[i]]=i;h[]=;
    for(int i=;i<=n;++i){
      h[i]=max(h[i-]-,);
      while(i+h[i]<=n&&sa[rk[i]-]+h[i]<=n&&s[i+h[i]]==s[sa[rk[i]-]+h[i]])h[i]++;
    }
}
signed main(){
    scanf("%lld",&n);scanf("%s",s+);work();
    for(int i=;i<=n;i++)Ans+=(n-sa[i]+-h[i]);
    cout<<Ans;
    return ;
}