[Luogu2408] 不同子串个数 [SA]

Link
Luogu - https://www.luogu.org/problemnew/show/P2408

---

Suffix Array 密切相关的是 LCP Array 或者也被叫做 Height 数组
通常我们规定 Height[i] = LCP(Suffix[SA[i]], Suffix[SA[i-1]])
同时有重要性质 Height[Rank[i]] ≥ Height[Rank[i-1]] - 1
现在记 H[i]=Height[Rank[i] 。有 H[i] ≥ H[i-1] - 1

证明：设 Suffix[k] 排在 Suffix[i-1] 前一名。那么它们的 LCP 是 H[i-1] 。
假如 H[i-1] > 1 那么有：Suffix[k+1] 必定排在 Suffix[i] 之前，并且他们的 LCP 是 (H[i-1] - 1) 。
（证明很显然，因为 Suffix[k+1] 就是 Suffix[k] 去掉首字符， Suffix[i] 同理）
那么 Suffix[i] 和排在它前一位的后缀的 LCP 也就是 H[i] 至少是 (H[i-1] - 1) 。
否则排在它前一位的就是 Suffix[k+1] 。
（这个也很显然，如果想不通的话，罚你重抄高精度大数比较 or 字符串字典序比较）
如果 H[i-1] < 1 那么。。。H[i] ≥ 0 还用说吗
证毕。
不过合起来就不好一眼看出来啦。如果记住去掉首字符的思路的话也许比较容易推？

重要性质
$\mathrm{LCP}(\mathrm{S}\mathrm{u}\mathrm{f}\mathrm{f}\mathrm{i}\mathrm{x}[\mathrm{i}],\mathrm{S}\mathrm{u}\mathrm{f}\mathrm{f}\mathrm{i}\mathrm{x}[\mathrm{i}])=\mathrm{S}\mathrm{u}\mathrm{f}\mathrm{f}\mathrm{i}\mathrm{x}[\mathrm{i}].\mathrm{l}\mathrm{e}\mathrm{n}\mathrm{g}\mathrm{t}\mathrm{h}=\mathrm{n}-\mathrm{i}+1$
$\mathrm{LCP}(\mathrm{S}\mathrm{u}\mathrm{f}\mathrm{f}\mathrm{i}\mathrm{x}[\mathrm{S}\mathrm{A}[\mathrm{i}]],\mathrm{S}\mathrm{u}\mathrm{f}\mathrm{f}\mathrm{i}\mathrm{x}[\mathrm{S}\mathrm{A}[\mathrm{j}]])=\underset{\mathrm{k}\in (\mathrm{i},\mathrm{j}]}{\min }\mathrm{H}\mathrm{e}\mathrm{i}\mathrm{g}\mathrm{h}\mathrm{t}[\mathrm{k}]$

---

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<cmath>
using namespace std;
#define R register
const int MAXN = 3e5 + 233;
int T, n, m, Q, qL, qR;
long long Ans;
string S;
int Rk[MAXN], SA[MAXN], Tax[MAXN], Tp[MAXN];
void Radix_Sort()
{
    fill(Tax, Tax + 1 + m, 0);
    for (R int i = 1; i <= n; ++i) ++Tax[Rk[i]];
    for (R int i = 1; i <= m; ++i) Tax[i] += Tax[i-1];
    for (R int i = n; i >= 1; --i) SA[Tax[Rk[Tp[i]]]--] = Tp[i];
}
void Suffix_Sort()
{
    m = 127;
    for (R int i = 1; i <= n; ++i) Rk[i] = S[i], Tp[i] = i;
    Radix_Sort();
    for (R int Len = 1, tot = 0; tot < n; Len <<= 1, m = tot)
    {
        tot = 0;
        for (R int i = n - Len + 1; i <= n; ++i) Tp[++tot] = i;
        for (R int i = 1; i <= n; ++i) if (SA[i] > Len) Tp[++tot] = SA[i] - Len;
        Radix_Sort();
        swap(Rk, Tp);
        Rk[SA[1]] = tot = 1;
        for (R int i = 2; i <= n; ++i)
        {
            if (!( Tp[SA[i]] == Tp[SA[i-1]] && (Tp[SA[i]+Len] == Tp[SA[i-1]+Len]) )) ++tot;
            Rk[SA[i]] = tot;
        }
    }
}
void Get_Height()
{
    for (R int k = 0, j, i = 1; i <= n; ++i)
    {
        if (k) --k;
        j = SA[Rk[i] - 1];
        while (S[i + k] == S[j + k]) ++k;
        Ans += n - SA[Rk[i]] + 1 - k;
    }
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin >> n;
    cin >> S;
    for (R int i = n; i >= 1; --i) S[i] = S[i-1];
    Suffix_Sort();
    Get_Height();
    cout << Ans;
    return 0;
}