后缀数组简介

一些约定

字符串相关的定义请参考字符串基础。

字符串下标从 \(1\) 开始。

字符串 \(s\) 的长度为 \(n\)。

" 后缀 \(i\)" 代指以第 \(i\) 个字符开头的后缀，存储时用 \(i\) 代表字符串 \(s\) 的后缀 \(s[i\dots n]\)。

后缀数组是什么？

后缀数组（Suffix Array）主要关系到两个数组：\(sa\) 和 \(rk\)。

其中，\(sa[i]\) 表示将所有后缀排序后第 \(i\) 小的后缀的编号，也是所说的后缀数组，后文也称编号数组 \(sa\)；

\(rk[i]\) 表示后缀 \(i\) 的排名，是重要的辅助数组，后文也称排名数组 \(rk\)。

这两个数组满足性质：\(sa[rk[i]]=rk[sa[i]]=i\)。

解释

后缀数组示例：

后缀数组怎么求？

O(n^2logn) 做法

我相信这个做法大家还是能自己想到的：将盛有全部后缀字符串的数组进行 sort 排序，由于排序进行 \(O(n\log n)\) 次字符串比较，每次字符串比较要 \(O(n)\) 次字符比较，所以这个排序是 \(O(n^2\log n)\) 的时间复杂度。

O(nlog^2n) 做法

这个做法要用到倍增的思想。

首先对字符串 \(s\) 的所有长度为 \(1\) 的子串，即每个字符进行排序，得到排序后的编号数组 \(sa_1\) 和排名数组 \(rk_1\)。

倍增过程：

用两个长度为 \(1\) 的子串的排名，即 \(rk_1[i]\) 和 \(rk_1[i+1]\)，作为排序的第一第二关键字，就可以对字符串 \(s\) 的每个长度为 \(2\) 的子串：\(\{s[i\dots \min(i+1, n)]\ |\ i \in [1,\ n]\}\) 进行排序，得到 \(sa_2\) 和 \(rk_2\)；
之后用两个长度为 \(2\) 的子串的排名，即 \(rk_2[i]\) 和 \(rk_2[i+2]\)，作为排序的第一第二关键字，就可以对字符串 \(s\) 的每个长度为 \(4\) 的子串：\(\{s[i\dots \min(i+3, n)]\ |\ i \in [1,\ n]\}\) 进行排序，得到 \(sa_4\) 和 \(rk_4\)；
以此倍增，用长度为 \(w/2\) 的子串的排名，即 \(rk_{w/2}[i]\) 和 \(rk_{w/2}[i+w/2]\)，作为排序的第一第二关键字，就可以对字符串 \(s\) 的每个长度为 \(w\) 的子串 \(s[i\dots \min(i+w-1,\ n)]\) 进行排序，得到 \(sa_w\) 和 \(rk_w\)。其中，类似字母序排序规则，当 \(i+w>n\) 时，\(rk_w[i+w]\) 视为无穷小；
\(rk_w[i]\) 即是子串 \(s[i\dots i + w - 1]\) 的排名，这样当 \(w \geqslant n\) 时，得到的编号数组 \(sa_w\)，也就是我们需要的后缀数组。

过程

倍增排序示意图：

显然倍增的过程是 \(O(\log n)\)，而每次倍增用 sort 对子串进行排序是 \(O(n\log n)\)，而每次子串的比较花费 \(2\) 次字符比较；

除此之外，每次倍增在 sort 排序完后，还有额外的 \(O(n)\) 时间复杂度的，更新 \(rk\) 的操作，但是相对于 \(O(n\log n)\) 被忽略不计；

所以这个算法的时间复杂度就是 \(O(n\log^2n)\)。

实现

C++
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 1000010;

char s[N];
int n, w, sa[N], rk[N << 1], oldrk[N << 1];

// 为了防止访问 rk[i+w] 导致数组越界，开两倍数组。
// 当然也可以在访问前判断是否越界，但直接开两倍数组方便一些。

int main() {
  int i, p;

  scanf("%s", s + 1);
  n = strlen(s + 1);
  for (i = 1; i <= n; ++i) sa[i] = i, rk[i] = s[i];

  for (w = 1; w < n; w <<= 1) {
    sort(sa + 1, sa + n + 1, [](int x, int y) {
      return rk[x] == rk[y] ? rk[x + w] < rk[y + w] : rk[x] < rk[y];
    });  // 这里用到了 lambda
    memcpy(oldrk, rk, sizeof(rk));
    // 由于计算 rk 的时候原来的 rk 会被覆盖，要先复制一份
    for (p = 0, i = 1; i <= n; ++i) {
      if (oldrk[sa[i]] == oldrk[sa[i - 1]] &&
          oldrk[sa[i] + w] == oldrk[sa[i - 1] + w]) {
        rk[sa[i]] = p;
      } else {
        rk[sa[i]] = ++p;
      }  // 若两个子串相同，它们对应的 rk 也需要相同，所以要去重
    }
  }

  for (i = 1; i <= n; ++i) printf("%d ", sa[i]);

  return 0;
}

O(nlogn) 做法

在刚刚的 \(O(n\log^2n)\) 做法中，单次排序是 \(O(n\log n)\) 的，如果能 \(O(n)\) 排序，就能 \(O(n\log n)\) 计算后缀数组了。

前置知识：计数排序，基数排序。

由于计算后缀数组的过程中排序的关键字是排名，值域为 \(O(n)\)，并且是一个双关键字的排序，可以使用基数排序优化至 \(O(n)\)。

实现

C++
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 1000010;

char s[N];
int n, sa[N], rk[N << 1], oldrk[N << 1], id[N], cnt[N];

int main() {
  int i, m, p, w;

  scanf("%s", s + 1);
  n = strlen(s + 1);
  m = 127;
  for (i = 1; i <= n; ++i) ++cnt[rk[i] = s[i]];
  for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
  for (i = n; i >= 1; --i) sa[cnt[rk[i]]--] = i;
  memcpy(oldrk + 1, rk + 1, n * sizeof(int));
  for (p = 0, i = 1; i <= n; ++i) {
    if (oldrk[sa[i]] == oldrk[sa[i - 1]]) {
      rk[sa[i]] = p;
    } else {
      rk[sa[i]] = ++p;
    }
  }

  for (w = 1; w < n; w <<= 1, m = n) {
    // 对第二关键字：id[i] + w进行计数排序
    memset(cnt, 0, sizeof(cnt));
    memcpy(id + 1, sa + 1,
           n * sizeof(int));  // id保存一份儿sa的拷贝，实质上就相当于oldsa
    for (i = 1; i <= n; ++i) ++cnt[rk[id[i] + w]];
    for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
    for (i = n; i >= 1; --i) sa[cnt[rk[id[i] + w]]--] = id[i];

    // 对第一关键字：id[i]进行计数排序
    memset(cnt, 0, sizeof(cnt));
    memcpy(id + 1, sa + 1, n * sizeof(int));
    for (i = 1; i <= n; ++i) ++cnt[rk[id[i]]];
    for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
    for (i = n; i >= 1; --i) sa[cnt[rk[id[i]]]--] = id[i];

    memcpy(oldrk + 1, rk + 1, n * sizeof(int));
    for (p = 0, i = 1; i <= n; ++i) {
      if (oldrk[sa[i]] == oldrk[sa[i - 1]] &&
          oldrk[sa[i] + w] == oldrk[sa[i - 1] + w]) {
        rk[sa[i]] = p;
      } else {
        rk[sa[i]] = ++p;
      }
    }
  }

  for (i = 1; i <= n; ++i) printf("%d ", sa[i]);

  return 0;
}

一些常数优化

如果你把上面那份代码交到 LOJ #111: 后缀排序上：

这是因为，上面那份代码的常数的确很大。

第二关键字无需计数排序

思考一下第二关键字排序的实质，其实就是把超出字符串范围（即 \(sa[i] + w > n\)）的 \(sa[i]\) 放到 \(sa\) 数组头部，然后把剩下的依原顺序放入：

C++
for (p = 0, i = n; i > n - w; --i) id[++p] = i;

for (i = 1; i <= n; ++i) {
  if (sa[i] > w) id[++p] = sa[i] - w;
}

优化计数排序的值域

每次对 \(rk\) 进行更新之后，我们都计算了一个 \(p\)，这个 \(p\) 即是 \(rk\) 的值域，将值域改成它即可。

将 rk[id[i]] 存下来，减少不连续内存访问

这个优化在数据范围较大时效果非常明显。

用函数 cmp 来计算是否重复

同样是减少不连续内存访问，在数据范围较大时效果比较明显。

把 oldrk[sa[i]] == oldrk[sa[i - 1]] && oldrk[sa[i] + w] == oldrk[sa[i - 1] + w]

替换成 cmp(sa[i], sa[i - 1], w)，

bool cmp(int x, int y, int w) { return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w]; }。

若排名都不相同可直接生成后缀数组

考虑新的 \(rk\) 数组，若其值域为 \([1,n]\) 那么每个排名都不同，此时无需再排序。

实现

C++
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 1000010;

char s[N];
// key1[i] = rk[id[i]]（作为基数排序的第一关键字数组）
int n, sa[N], rk[N], oldrk[N << 1], id[N], key1[N], cnt[N];

bool cmp(int x, int y, int w) {
  return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w];
}

int main() {
  int i, m = 127, p, w;

  scanf("%s", s + 1);
  n = strlen(s + 1);
  for (i = 1; i <= n; ++i) ++cnt[rk[i] = s[i]];
  for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
  for (i = n; i >= 1; --i) sa[cnt[rk[i]]--] = i;

  for (w = 1;; w <<= 1, m = p) {  // m=p 就是优化计数排序值域
    for (p = 0, i = n; i > n - w; --i) id[++p] = i;
    for (i = 1; i <= n; ++i)
      if (sa[i] > w) id[++p] = sa[i] - w;

    memset(cnt, 0, sizeof(cnt));
    for (i = 1; i <= n; ++i) ++cnt[key1[i] = rk[id[i]]];
    // 注意这里px[i] != i，因为rk没有更新，是上一轮的排名数组

    for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
    for (i = n; i >= 1; --i) sa[cnt[key1[i]]--] = id[i];
    memcpy(oldrk + 1, rk + 1, n * sizeof(int));
    for (p = 0, i = 1; i <= n; ++i)
      rk[sa[i]] = cmp(sa[i], sa[i - 1], w) ? p : ++p;
    if (p == n) {
      for (int i = 1; i <= n; ++i) sa[rk[i]] = i;
      break;
    }
  }

  for (i = 1; i <= n; ++i) printf("%d ", sa[i]);

  return 0;
}

O(n) 做法

在一般的题目中，常数较小的倍增求后缀数组是完全够用的，求后缀数组以外的部分也经常有 \(O(n\log n)\) 的复杂度，倍增求解后缀数组不会成为瓶颈。

但如果遇到特殊题目、时限较紧的题目，或者是你想追求更短的用时，就需要学习 \(O(n)\) 求后缀数组的方法。

SA-IS

可以参考诱导排序与 SA-IS 算法，另外它的评论页面也有参考价值。

DC3

可以参考[2009] 后缀数组——处理字符串的有力工具 by. 罗穗骞。

后缀数组的应用

寻找最小的循环移动位置

将字符串 \(S\) 复制一份变成 \(SS\) 就转化成了后缀排序问题。

例题：「JSOI2007」字符加密。

在字符串中找子串

任务是在线地在主串 \(T\) 中寻找模式串 \(S\)。在线的意思是，我们已经预先知道知道主串 \(T\)，但是当且仅当询问时才知道模式串 \(S\)。我们可以先构造出 \(T\) 的后缀数组，然后查找子串 \(S\)。若子串 \(S\) 在 \(T\) 中出现，它必定是 \(T\) 的一些后缀的前缀。因为我们已经将所有后缀排序了，我们可以通过在 \(p\) 数组中二分 \(S\) 来实现。比较子串 \(S\) 和当前后缀的时间复杂度为 \(O(|S|)\)，因此找子串的时间复杂度为 \(O(|S|\log |T|)\)。注意，如果该子串在 \(T\) 中出现了多次，每次出现都是在 \(p\) 数组中相邻的。因此出现次数可以通过再次二分找到，输出每次出现的位置也很轻松。

从字符串首尾取字符最小化字典序

例题：「USACO07DEC」Best Cow Line。

题意：给你一个字符串，每次从首或尾取一个字符组成字符串，问所有能够组成的字符串中字典序最小的一个。

题解

暴力做法就是每次最坏 \(O(n)\) 地判断当前应该取首还是尾（即比较取首得到的字符串与取尾得到的反串的大小），只需优化这一判断过程即可。

由于需要在原串后缀与反串后缀构成的集合内比较大小，可以将反串拼接在原串后，并在中间加上一个没出现过的字符（如 #，代码中可以直接使用空字符），求后缀数组，即可 \(O(1)\) 完成这一判断。

参考代码

C++
#include <cctype>
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 1000010;

char s[N];
int n, sa[N], id[N], oldrk[N * 2], rk[N * 2], px[N], cnt[N];

bool cmp(int x, int y, int w) {
  return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w];
}

int main() {
  int i, w, m = 200, p, l = 1, r, tot = 0;

  cin >> n;
  r = n;

  for (i = 1; i <= n; ++i)
    while (!isalpha(s[i] = getchar()))
      ;
  for (i = 1; i <= n; ++i)
    rk[i] = rk[2 * n + 2 - i] = s[i];  // 拼接正反两个字符串，中间空出一个字符

  n = 2 * n + 1;
  // 求后缀数组
  for (i = 1; i <= n; ++i) ++cnt[rk[i]];
  for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
  for (i = n; i >= 1; --i) sa[cnt[rk[i]]--] = i;

  for (w = 1; w < n; w *= 2, m = p) {  // m=p 就是优化计数排序值域
    for (p = 0, i = n; i > n - w; --i) id[++p] = i;
    for (i = 1; i <= n; ++i)
      if (sa[i] > w) id[++p] = sa[i] - w;
    memset(cnt, 0, sizeof(cnt));
    for (i = 1; i <= n; ++i) ++cnt[px[i] = rk[id[i]]];
    for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
    for (i = n; i >= 1; --i) sa[cnt[px[i]]--] = id[i];
    memcpy(oldrk, rk, sizeof(rk));
    for (p = 0, i = 1; i <= n; ++i)
      rk[sa[i]] = cmp(sa[i], sa[i - 1], w) ? p : ++p;
  }
  // 利用后缀数组O(1)进行判断
  while (l <= r) {
    printf("%c", rk[l] < rk[n + 1 - r] ? s[l++] : s[r--]);
    if ((++tot) % 80 == 0) puts("");  // 回车
  }

  return 0;
}

height 数组

LCP（最长公共前缀）

两个字符串 \(S\) 和 \(T\) 的 LCP 就是最大的 \(x\)(\(x\le \min(|S|, |T|)\)) 使得 \(S_i=T_i\ (\forall\ 1\le i\le x)\)。

下文中以 \(lcp(i,j)\) 表示后缀 \(i\) 和后缀 \(j\) 的最长公共前缀（的长度）。

height 数组的定义

\(height[i]=lcp(sa[i],sa[i-1])\)，即第 \(i\) 名的后缀与它前一名的后缀的最长公共前缀。

\(height[1]\) 可以视作 \(0\)。

O(n) 求 height 数组需要的一个引理

\(height[rk[i]]\ge height[rk[i-1]]-1\)

证明

当 \(height[rk[i-1]]\le1\) 时，上式显然成立（右边小于等于 \(0\)）。

当 \(height[rk[i-1]]>1\) 时：

根据 \(height\) 定义，有 \(lcp(sa[rk[i-1]], sa[rk[i-1]-1]) = height[rk[i-1]] > 1\)。

既然后缀 \(i-1\) 和后缀 \(sa[rk[i-1]-1]\) 有长度为 \(height[rk[i-1]]\) 的最长公共前缀，

那么不妨用 \(aA\) 来表示这个最长公共前缀。（其中 \(a\) 是一个字符，\(A\) 是长度为 \(height[rk[i-1]]-1\) 的字符串，非空）

那么后缀 \(i-1\) 可以表示为 \(aAD\)，后缀 \(sa[rk[i-1]-1]\) 可以表示为 \(aAB\)。（\(B < D\)，\(B\) 可能为空串，\(D\) 非空）

进一步地，后缀 \(i\) 可以表示为 \(AD\)，存在后缀（\(sa[rk[i-1]-1]+1\)）\(AB\)。

因为后缀 \(sa[rk[i]-1]\) 在大小关系的排名上仅比后缀 \(sa[rk[i]]\) 也就是后缀 \(i\)，小一位，而 \(AB < AD\)。

所以 \(AB \leqslant\) 后缀 \(sa[rk[i]-1] < AD\)，显然后缀 \(i\) 和后缀 \(sa[rk[i]-1]\) 有公共前缀 \(A\)。

于是就可以得出 \(lcp(i,sa[rk[i]-1])\) 至少是 \(height[rk[i-1]]-1\)，也即 \(height[rk[i]]\ge height[rk[i-1]]-1\)。

O(n) 求 height 数组的代码实现

利用上面这个引理暴力求即可：

C++
for (i = 1, k = 0; i <= n; ++i) {
  if (rk[i] == 0) continue;
  if (k) --k;
  while (s[i + k] == s[sa[rk[i] - 1] + k]) ++k;
  height[rk[i]] = k;
}

\(k\) 不会超过 \(n\)，最多减 \(n\) 次，所以最多加 \(2n\) 次，总复杂度就是 \(O(n)\)。

height 数组的应用

两子串最长公共前缀

\(lcp(sa[i],sa[j])=\min\{height[i+1..j]\}\)

感性理解：如果 \(height\) 一直大于某个数，前这么多位就一直没变过；反之，由于后缀已经排好序了，不可能变了之后变回来。

严格证明可以参考[2004] 后缀数组 by. 徐智磊。

有了这个定理，求两子串最长公共前缀就转化为了 RMQ 问题。

比较一个字符串的两个子串的大小关系

假设需要比较的是 \(A=S[a..b]\) 和 \(B=S[c..d]\) 的大小关系。

若 \(lcp(a, c)\ge\min(|A|, |B|)\)，\(A<B\iff |A|<|B|\)。

否则，\(A<B\iff rk[a]< rk[c]\)。

不同子串的数目

子串就是后缀的前缀，所以可以枚举每个后缀，计算前缀总数，再减掉重复。

「前缀总数」其实就是子串个数，为 \(n(n+1)/2\)。

如果按后缀排序的顺序枚举后缀，每次新增的子串就是除了与上一个后缀的 LCP 剩下的前缀。这些前缀一定是新增的，否则会破坏 \(lcp(sa[i],sa[j])=\min\{height[i+1..j]\}\) 的性质。只有这些前缀是新增的，因为 LCP 部分在枚举上一个前缀时计算过了。

所以答案为：

\(\frac{n(n+1)}{2}-\sum\limits_{i=2}^nheight[i]\)

出现至少 k 次的子串的最大长度

例题：「USACO06DEC」Milk Patterns。

题解

出现至少 \(k\) 次意味着后缀排序后有至少连续 \(k\) 个后缀以这个子串作为公共前缀。

所以，求出每相邻 \(k-1\) 个 \(height\) 的最小值，再求这些最小值的最大值就是答案。

可以使用单调队列 \(O(n)\) 解决，但使用其它方式也足以 AC。

参考代码

C++
#include <cstdio>
#include <cstring>
#include <iostream>
#include <set>

using namespace std;

const int N = 40010;

int n, k, a[N], sa[N], rk[N], oldrk[N], id[N], px[N], cnt[1000010], ht[N], ans;
multiset<int> t;  // multiset 是最好写的实现方式

bool cmp(int x, int y, int w) {
  return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w];
}

int main() {
  int i, j, w, p, m = 1000000;

  scanf("%d%d", &n, &k);
  --k;

  for (i = 1; i <= n; ++i) scanf("%d", a + i);  // 求后缀数组
  for (i = 1; i <= n; ++i) ++cnt[rk[i] = a[i]];
  for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
  for (i = n; i >= 1; --i) sa[cnt[rk[i]]--] = i;

  for (w = 1; w < n; w <<= 1, m = p) {
    for (p = 0, i = n; i > n - w; --i) id[++p] = i;
    for (i = 1; i <= n; ++i)
      if (sa[i] > w) id[++p] = sa[i] - w;
    memset(cnt, 0, sizeof(cnt));
    for (i = 1; i <= n; ++i) ++cnt[px[i] = rk[id[i]]];
    for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
    for (i = n; i >= 1; --i) sa[cnt[px[i]]--] = id[i];
    memcpy(oldrk, rk, sizeof(rk));
    for (p = 0, i = 1; i <= n; ++i)
      rk[sa[i]] = cmp(sa[i], sa[i - 1], w) ? p : ++p;
  }

  for (i = 1, j = 0; i <= n; ++i) {  // 求 height
    if (j) --j;
    while (a[i + j] == a[sa[rk[i] - 1] + j]) ++j;
    ht[rk[i]] = j;
  }

  for (i = 1; i <= n; ++i) {  // 求所有最小值的最大值
    t.insert(ht[i]);
    if (i > k) t.erase(t.find(ht[i - k]));
    ans = max(ans, *t.begin());
  }

  cout << ans;

  return 0;
}

是否有某字符串在文本串中至少不重叠地出现了两次

可以二分目标串的长度 \(|s|\)，将 \(h\) 数组划分成若干个连续 LCP 大于等于 \(|s|\) 的段，利用 RMQ 对每个段求其中出现的数中最大和最小的下标，若这两个下标的距离满足条件，则一定有长度为 \(|s|\) 的字符串不重叠地出现了两次。

连续的若干个相同子串

我们可以枚举连续串的长度 \(|s|\)，按照 \(|s|\) 对整个串进行分块，对相邻两块的块首进行 LCP 与 LCS 查询，具体可见[2009] 后缀数组——处理字符串的有力工具。

例题：「NOI2016」优秀的拆分。

结合并查集

某些题目求解时要求你将后缀数组划分成若干个连续 LCP 长度大于等于某一值的段，亦即将 \(h\) 数组划分成若干个连续最小值大于等于某一值的段并统计每一段的答案。如果有多次询问，我们可以将询问离线。观察到当给定值单调递减的时候，满足条件的区间个数总是越来越少，而新区间都是两个或多个原区间相连所得，且新区间中不包含在原区间内的部分的 \(h\) 值都为减少到的这个值。我们只需要维护一个并查集，每次合并相邻的两个区间，并维护统计信息即可。

经典题目：「NOI2015」品酒大会

结合线段树

某些题目让你求满足条件的前若干个数，而这些数又在后缀排序中的一个区间内。这时我们可以用归并排序的性质来合并两个结点的信息，利用线段树维护和查询区间答案。

结合单调栈

例题：「AHOI2013」差异

题解

被加数的前两项很好处理，为 \(n(n-1)(n+1)/2\)（每个后缀都出现了 \(n-1\) 次，后缀总长是 \(n(n+1)/2\)），关键是最后一项，即后缀的两两 LCP。

我们知道 \(lcp(i,j)=k\) 等价于 \(\min\{height[i+1..j]\}=k\)。所以，可以把 \(lcp(i,j)\) 记作 \(\min\{x|i+1\le x\le j, height[x]=lcp(i,j)\}\) 对答案的贡献。

考虑每个位置对答案的贡献是哪些后缀的 LCP，其实就是从它开始向左若干个连续的 \(height\) 大于它的后缀中选一个，再从向右若干个连续的 \(height\) 不小于它的后缀中选一个。这个东西可以用单调栈计算。

单调栈部分类似于 Luogu P2659 美丽的序列以及悬线法。

参考代码

C++
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 500010;

char s[N];
int n, sa[N], rk[N << 1], oldrk[N << 1], id[N], px[N], cnt[N], ht[N], sta[N],
    top, l[N];
long long ans;

bool cmp(int x, int y, int w) {
  return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w];
}

int main() {
  int i, k, w, p, m = 300;

  scanf("%s", s + 1);
  n = strlen(s + 1);
  ans = 1ll * n * (n - 1) * (n + 1) / 2;
  // 求后缀数组
  for (i = 1; i <= n; ++i) ++cnt[rk[i] = s[i]];
  for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
  for (i = n; i >= 1; --i) sa[cnt[rk[i]]--] = i;

  for (w = 1; w < n; w <<= 1, m = p) {
    for (p = 0, i = n; i > n - w; --i) id[++p] = i;
    for (i = 1; i <= n; ++i)
      if (sa[i] > w) id[++p] = sa[i] - w;
    memset(cnt, 0, sizeof(cnt));
    for (i = 1; i <= n; ++i) ++cnt[px[i] = rk[id[i]]];
    for (i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];
    for (i = n; i >= 1; --i) sa[cnt[px[i]]--] = id[i];
    memcpy(oldrk, rk, sizeof(rk));
    for (p = 0, i = 1; i <= n; ++i)
      rk[sa[i]] = cmp(sa[i], sa[i - 1], w) ? p : ++p;
  }
  // 求 height
  for (i = 1, k = 0; i <= n; ++i) {
    if (k) --k;
    while (s[i + k] == s[sa[rk[i] - 1] + k]) ++k;
    ht[rk[i]] = k;
  }
  // 维护单调栈
  for (i = 1; i <= n; ++i) {
    while (ht[sta[top]] > ht[i]) --top;  // top类似于一个指针
    l[i] = i - sta[top];
    sta[++top] = i;
  }
  // 最后利用单调栈算 ans
  sta[++top] = n + 1;
  ht[n + 1] = -1;
  for (i = n; i >= 1; --i) {
    while (ht[sta[top]] >= ht[i]) --top;
    ans -= 2ll * ht[i] * l[i] * (sta[top] - i);
    sta[++top] = i;
  }

  cout << ans;

  return 0;
}

类似的题目：「HAOI2016」找相同字符。

习题

参考资料

本页面中（4070a9b 引入的部分）主要译自博文 Суффиксный массив 与其英文翻译版 Suffix Array。其中俄文版版权协议为 Public Domain + Leave a Link；英文版版权协议为 CC-BY-SA 4.0。

论文：

本页面的全部内容在 小熊老师 - 莆田青少年编程俱乐部 0594codes.cn 协议之条款下提供，附加条款亦可能应用