## 题目描述:
给你一个长为 $N$ $(N<=10^5)$ 的字符串,求不同的子串的个数
我们定义两个子串不同,当且仅当有这两个子串长度不一样 或者长度一样且有任意一位不一样。子串的定义:原字符串中连续的一段字符组成的字符串
很妙的一道题,考察了对 $Height$ 数组的理解。
$1.$首先,不难发现任意子串都可以被字符串中后缀串的前缀表达出来
$2.$我们知道, $Height[i]$ 被定义为排名为 $i$ 的后缀串与排名为 $i-1$ 的后缀串的 $LCP$.
而与排名为 $i$ 得后缀串 $LCP$ 值最大的字符串必定是排名为 $i-1$ 的后缀串,他们的 $LCP$
值恰好就是 $Height$ 数组的值,即$Height[i]$.
考虑向后缀串集合中新加入一个后缀串 $sa[k]$, 共会产生 $n-sa[k]+1$ 个前缀串,但是有一些
前缀串在先前就已经会被计算到,会被计算到的前缀部分的最大值是 $Height[k]$,直接减去
$Height[k]$ 即可. 即贡献为 $n-sa[k]+1-Height[k]$.
Code:
#include <cstdio>
#include <algorithm>
#include <cstring>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 1000000
using namespace std;
char str[maxn];
int tr[maxn],rk[maxn],sa[maxn],arr[maxn],c[maxn],height[maxn];
int n,m;
struct Suffix_Array
{
void qsort()
{
for(int i=0;i<=m;++i) c[i]=0;
for(int i=1;i<=n;++i) ++c[rk[tr[i]]];
for(int i=1;i<=m;++i) c[i]+=c[i-1];
for(int i=n;i>=1;--i) sa[c[rk[tr[i]]]--]=tr[i];
}
void build()
{
for(int i=1;i<=n;++i) rk[i]=arr[i],tr[i]=i;
qsort();
for(int k=1;k<=n;k<<=1)
{
int num=0;
for(int i=n-k+1;i<=n;++i) tr[++num]=i;
for(int i=1;i<=n;++i) if(sa[i]>k) tr[++num]=sa[i]-k;
qsort();
swap(rk,tr);
rk[sa[1]]=1;
num=1;
for(int i=2;i<=n;++i)
rk[sa[i]]=(tr[sa[i]]==tr[sa[i-1]]&&tr[sa[i]+k]==tr[sa[i-1]+k])?num:++num;
if(num>=n) break;
m=num;
}
}
void get_height()
{
int k=0;
for(int i=1;i<=n;++i) rk[sa[i]]=i;
for(int i=1;i<=n;++i){
if(k) --k;
int j=sa[rk[i]-1];
while(arr[i+k]==arr[j+k]) ++k;
height[rk[i]]=k;
}
}
}T;
int main()
{
//setIO("input");
scanf("%d",&n);
scanf("%s",str),m=120;
for(int i=1;i<=n;++i) arr[i]=str[i-1]-'0';
T.build();
T.get_height();
long long ans=0;
for(int i=1;i<=n;++i)
ans+=(long long) (n-sa[i]+1-height[i]);
printf("%lld",ans);
return 0;
}