专栏文章

「笔记」后缀数组

算法·理论参与者 24已保存评论 25

文章操作

快速查看文章及其快照的属性,并进行相关操作。

当前评论
25 条
当前快照
1 份
快照标识符
@mhz5rx8n
此快照首次捕获于
2025/11/15 01:55
4 个月前
此快照最后确认于
2025/11/29 05:24
3 个月前
查看原文

小插曲

@ComeIntoPower 审日报的时候我正好在线,其实这篇一开始是没有过的。
草
心死了,犇犇都发出去了,伤心了 10min 又来看了一眼发现过了(((
于是顺便来扩充了一下内容。

一些约定

  1. \left| \sum \right|:字符集大小。
  2. S[i:j]S[i:j]:由字符串 SSSiSjS_i\sim S_j 构成的子串。
  3. S1<S2S_1<S_2:字符串 S1S_1 的字典序 <S2<S_2
  4. 后缀:从某个位置 ii 开始,到串末尾结束的子串,后缀 ii 等价于子串 S[i:n]S[i:n]。每一个后缀都与一个 1n1\sim n 的整数一一映射。

SA

SA 的定义

字符串 SS 的后缀数组定义为两个数组 sa,rksa,rk
sasa 储存 SS 的所有后缀 按字典序排序后的起始下标,满足 S[sai1:n]<S[sai:n]S[sa_{i-1}:n]<S[sa_i:n]
rkrk 储存 SS 的所有后缀的排名。
举例:这里有一个可爱的字符串 S=yuyukoS=\texttt{yuyuko}
k<o<u<y\texttt{k}<\texttt{o}<\texttt{u}<\texttt{y},则它的后缀数组 sa=[5,6,4,2,3,1]sa = [5,6,4,2,3,1]rk=[6,4,5,3,1,2]rk = [6,4,5,3,1,2]。具体地,有:
| 排名 | 1 | 2 | 3 | 4 | 5 | 6 | | :----------- | :----------- | :----------- | :----------- | :----------- | :----------- | :----------- | :----------- |
| 下标 | 55 | 66 | 44 | 22 | 33 | 11 | | 后缀 | ko\texttt{ko} | o\texttt{o} | uko\texttt{uko} |uyuko\texttt{uyuko} | yuko\texttt{yuko} | yuyuko\texttt{yuyuko} |
不同后缀的排名必然不同(长度不等),rkrk 中不会有重复值出现。

倍增法构造

考虑字符串大小的从前向后的比较过程,可以先将所有长度为 2k2^k 的子串进行排序,通过相邻子串合并并比较大小,求出所有长度为 2k+12^{k+1} 的子串的大小关系。
对于 S[i:i+2k1]S[i:i+2^k-1]S[j:j+2k1]S[j:j+2^k-1],分别将它们裂开,分成两长度为 2k12^{k-1} 的串。设 Ai=S[i:i+2k11]A_i = S[i:i+2^{k-1}-1]Bi=S[i+2k1:i+2k1]B_i = S[i+2^{k-1}:i+2^k-1]
考虑字典序排序的过程,则 S[i:i+2k1]<S[j:j+2k1]S[i:i+2^k-1] <S[j:j+2^k-1] 的条件为:
[Ai<Aj]or [Ai=AjandBi<Bj][A_i<A_j] \operatorname{or}\ [A_i=A_j \operatorname{and} B_i<B_j]
考虑每一次倍增时,都使用 sort 按双关键字 AiA_iBiB_i 进行排序,每次倍增都进行依次排序,时间复杂度为 O(nlog2n)O(n\log^2 n)

P3809 【模板】后缀排序
以下是一份简单易懂的代码。
这里定义了两个数组:
saisa_i:倍增中 排名为 ii 的长度为 2k12^{k-1} 的子串。
rkirk_i:倍增过程中子串 S[i:i+2k1]S[i:i+2^k-1] 的排名,
显然它们互为反函数,sarki=rksai=isa_{rk_i}=rk_{sa_i} = i
初始化 rki=Sirk_i = S_i,即 SiS_iASCII\text{ASCII} 值。
虽然这样不满足值域在 [1,n][1,n] 内,但体现了大小关系,可用于更新。 rkrk 的值之后还会更新。
CPP
//知识点:SA
/*
By:Luckyblock
*/
#include <algorithm>
#include <cctype>
#include <cstdio>
#include <cstring>
#define LL long long
const int kN = 1e6 + 10;
//=============================================================
char s[kN];
int n, m, w, sa[kN], rk[kN << 1], oldrk[kN << 1];
//rk[i]: 倍增过程中子串[i:i+2^k-1]的排名,
//sa[i] 排名为i的子串 [i:i+2^k-1],
//它们互为反函数。
//存在越界风险,如果不写特判,rk 和 oldrk 要开 2 倍空间。
//=============================================================
inline int read() {
  int f = 1, w = 0;
  char ch = getchar();
  for (; !isdigit(ch); ch = getchar())
    if (ch == '-') f = -1;
  for (; isdigit(ch); ch = getchar()) w = (w << 3) + (w << 1) + (ch ^ '0');
  return f * w;
}
void Chkmax(int &fir_, int sec_) {
  if (sec_ > fir_) fir_ = sec_;
}
void Chkmin(int &fir_, int sec_) {
  if (sec_ < fir_) fir_ = sec_;
}
bool CMP(int fir_, int sec_) {
  if (rk[fir_] != rk[sec_]) return rk[fir_] < rk[sec_];
  return rk[fir_ + w] < rk[sec_ + w];
}
int main() {
  scanf("%s", s + 1);
  n = strlen(s + 1);
  m = std::max(n, 300);
  //初始化 rk 和 sa。
  //观察下面的代码可知,每次倍增时都会根据 rk 来更新 sa,则仅须保证 sa 数组是一个 1~n 的排列即可。
  for (int i = 1; i <= n; ++ i) sa[i] = i, rk[i] = s[i];
  //倍增过程。w 是已经推出的子串长度,数值上等于上述分析中的 2^{k-1}。
  //注意此处的 sa 数组存的并不是后缀的排名,存的是倍增过程中指定长度子串的排名。
  for (w = 1; w < n; w <<= 1) {
    std::sort(sa + 1, sa + n + 1, CMP);
    for (int i = 1; i <= n; ++ i) oldrk[i] = rk[i];
    for (int i = 1, p = 0; i <= n; ++ i) {
      if (oldrk[sa[i]] == oldrk[sa[i - 1]] && //判断两个子串是否相等。
          oldrk[sa[i] + w] == oldrk[sa[i - 1] + w]) { //越界风险,2倍空间
        rk[sa[i]] = p;
      } else {
        rk[sa[i]] = ++p;
      }
    }
  }
  for (int i = 1; i <= n; ++ i) printf("%d ", sa[i]);
  return 0;
}

计数排序 与 基数排序

优化上述算法的前置知识。
计数排序是一种与桶排序类似的排序方法。
将长度为 nn 的数列 aa 排序后放入 bb 的代码如下, 其中 ww 为值域,即 max{ai}\max\{a_i\}
CPP
int a[kMaxn], b[kMaxn], cnt[kMaxw];
for (int i = 1; i <= n; ++ i) ++ cnt[a[i]];
for (int i = 1; i <= w; ++ i) cnt[i] += cnt[i - 1];
for (int i = n; i >= 1; -- i) b[cnt[a[i]] --] = a[i];
其中,在对 cntcnt 求前缀和后, cnticnt_i 为不大于 ii 的数的数量,即为 ii 的排名。
因此在下一步中,可以根据排名赋值。
复杂度为 O(n+w)O(n+w),值域与 nn 同阶时复杂度比较优秀。

个人认为基数排序只是一种思想,并不算一种独立的排序方法。
它仅仅是将 kk 个排序关键字分开,按优先级升序依次考虑,从而实现多比较字的排序。实际每次排序还是靠其他排序方法实现。常常与计数排序相结合。

优化

请确保完全理解上述朴素实现后再阅读下文。
发现后缀数组值域即为 nn,又是多关键字排序,考虑基数排序。
上面已经给出一个用于比较的式子:
[Ai<Aj]or [Ai=AjandBi<Bj][A_i<A_j] \operatorname{or}\ [A_i=A_j \operatorname{and} B_i<B_j]
倍增过程中 Ai,BiA_i,B_i 大小关系已知,先将 BiB_i 作为第二关键字排序,再将 AiA_i 作为第一关键字排序,两次计数排序实现即可。
单次计数排序复杂度 O(n+w)O(n + w)ww 为值域,显然最大与 nn 同阶),总时间复杂度变为 O(nlogn)O(n\log n)
实现时将所有排序替换为基数排序即可。注意初始化。
CPP
//知识点:SA
/*
By:Luckyblock
I love Marisa;
But Marisa has died;
*/
#include <cstdio>
#include <ctype.h>
#include <cstring>
#include <algorithm>
#define ll long long
const int kMaxn = 1e6 + 10;
//=============================================================
char S[kMaxn];
//rk[i]: 倍增过程中子串[i:i+2^k-1]的排名,
//sa[i] 排名为i的子串 [i:i+2^k-1],
//它们互为反函数。
//存在越界风险,如果不写特判,rk 和 oldrk 要开 2 倍空间。
int n, m, sa[kMaxn], rk[kMaxn << 1], oldrk[kMaxn << 1];
int id[kMaxn], cnt[kMaxn]; //用于计数排序的两个 temp 数组
//=============================================================
inline int read() {
  int f = 1, w = 0; char ch = getchar();
  for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1;
  for (; isdigit(ch); ch = getchar()) w = (w << 3) + (w << 1) + (ch ^ '0');
  return f * w;
}
//=============================================================
int main() {
  scanf("%s", S + 1);
  n = strlen(S + 1);
  m = std :: max(n, 300); //值域大小
  
  //初始化 rk 和 sa
  for (int i = 1; i <= n; ++ i) ++ cnt[rk[i] = S[i]];
  for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
  for (int i = n; i >= 1; -- i) sa[cnt[rk[i]] --] = i;

  //倍增过程。w 是已经推出的子串长度,数值上等于上述分析中的 2^{k-1}。
  //注意此处的 sa 数组存的并不是后缀的排名,存的是倍增过程中指定长度子串的排名。
  for (int w = 1; w < n; w <<= 1) {
    //按照后半截 rk[i+w] 作为第二关键字排序。
    memset(cnt, 0, sizeof (cnt));
    for (int i = 1; i <= n; ++ i) id[i] = i;
    for (int i = 1; i <= n; ++ i) ++ cnt[rk[id[i] + w]]; //越界风险,2倍空间
    for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
    for (int i = n; i >= 1; -- i) sa[cnt[rk[id[i] + w]] --] = id[i];

    //按照前半截 rk[i] 作为第一关键字排序。
    memset(cnt, 0, sizeof (cnt));
    for (int i = 1; i <= n; ++ i) id[i] = sa[i];
    for (int i = 1; i <= n; ++ i) ++ cnt[rk[id[i]]];
    for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
    for (int i = n; i >= 1; -- i) sa[cnt[rk[id[i]]] --] = id[i];

    //更新 rk 数组,可以滚动数组一下,但是可读性会比较差(
    for (int i = 1; i <= n; ++ i) oldrk[i] = rk[i];
    for (int p = 0, i = 1; i <= n; ++ i) {
      if (oldrk[sa[i]] == oldrk[sa[i - 1]] &&  //判断两个子串是否相等。
          oldrk[sa[i] + w] == oldrk[sa[i - 1] + w]) { //越界风险,2倍空间
        rk[sa[i]] = p;
      } else {
        rk[sa[i]] = ++ p;
      }
    }
  }
  for (int i = 1; i <= n; ++ i) printf("%d ", sa[i]);
  return 0;
}
有一点小问题,排后半截时会枚举到 idi+w>nid_i+w > n 怎么办?
考虑实际意义,出现此情况,表示该子串后半截为空。
空串字典序最小,考虑直接把 rkrk 开成两倍空间,则 rki=0 (i>n)rk_i=0\ (i>n) 恒成立。防止了越界,也处理了空串的字典序。

再优化

两次计排太慢啦! 观察对后半截排序时的特殊性质:
考虑更新前的 saisa_i 的含义:排名为 ii 的长度为 2k12^{k-1} 的子串 S[sai,sai+2k1]S[sa_i, sa_i + 2^{k-1}]
在本次排序中,S[sai,sai+2k1]S[sa_i, sa_i + 2^{k-1}] 是长度为 2k2^k 的子串 S[sai2k1:sai+2k1]S[sa_{i}-2^{k-1}:sa_i+2^{k-1}] 的后半截,saisa_i 的排名将作为排序的关键字。
S[sai,sai+2k1]S[sa_i, sa_i + 2^{k-1}] 的排名为 ii,则第一次计排后 S[sai2k1:sai+2k1]S[sa_{i}-2^{k-1}:sa_i+2^{k-1}] 的排名必为 ii。考虑直接赋值,那么原来的第一次计排就可以写成这样:
CPP
int p = 0;
for (int i = n; i > n - w; -- i) id[++ p] = i; //后半截为空的串
for (int i = 1; i <= n; ++ i) { //根据后半截,直接推整个串的排名
  if (sa[i] > w) id[++ p] = sa[i] - w;
}
注意后半截为空串的情况,这样的串排名相同且最小。
以及一些奇怪的常数优化:
  • 减小值域。 值域大小 mm 与排序复杂度有关,其最小值应为 rkrk 的最大值,更新 rkrk 时更新 mm 即可。
  • 减少数组嵌套的使用,减少不连续内存访问。 在第二次计数排序时,将 rkidirk_{id_i} 存下来。
  • 用 cmp 函数判断两个子串是否相同。同样是减少不连续内存访问,详见代码。
CPP
//知识点:SA
/*
By:Luckyblock
I love Marisa;
*/
#include <cstdio>
#include <ctype.h>
#include <cstring>
#include <algorithm>
#define ll long long
const int kMaxn = 1e6 + 10;
//=============================================================
char S[kMaxn];
int n, m, sa[kMaxn], rk[kMaxn << 1], oldrk[kMaxn << 1];
int id[kMaxn], cnt[kMaxn], rkid[kMaxn];
//=============================================================
inline int read() {
  int f = 1, w = 0; char ch = getchar();
  for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1;
  for (; isdigit(ch); ch = getchar()) w = (w << 3) + (w << 1) + (ch ^ '0');
  return f * w;
}
bool cmp(int x, int y, int w) { //判断两个子串是否相等。
  return oldrk[x] == oldrk[y] && 
         oldrk[x + w] == oldrk[y + w]; 
}
//=============================================================
int main() {
  scanf("%s", S + 1);
  n = strlen(S + 1);
  m = std :: max(n, 300); //值域大小
  
  //初始化 sa数组
  for (int i = 1; i <= n; ++ i) ++ cnt[rk[i] = S[i]];
  for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
  for (int i = n; i >= 1; -- i) sa[cnt[rk[i]] --] = i;

  //倍增过程。 
  //此处 w = 2^{k-1},是已经推出的子串长度。
  //注意此处的 sa 数组存的并不是后缀的排名,
  //存的是指定长度子串的排名。
  for (int p, w = 1; w < n; w <<= 1) {
    //按照后半截 rk[i+w] 作为第二关键字排序。
    p = 0;
    for (int i = n; i > n - w; -- i) id[++ p] = i; //后半截为空的串
    for (int i = 1; i <= n; ++ i) { //根据后半截,直接推整个串的排名
      if (sa[i] > w) id[++ p] = sa[i] - w;
    }

    //按照前半截 rk[i] 作为第一关键字排序。
    memset(cnt, 0, sizeof (cnt));
    for (int i = 1; i <= n; ++ i) ++ cnt[rkid[i] = rk[id[i]]];
    for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
    for (int i = n; i >= 1; -- i) sa[cnt[rkid[i]] --] = id[i];

    //更新 rk 数组。
    //这里可以滚动数组一下,但是可读性会比较差(
    //卡常可以写一下。
    std ::swap(rk, oldrk);
    m = 0; //直接更新值域 m
    for (int i = 1; i <= n; ++ i) {
      rk[sa[i]] = (m += (cmp(sa[i], sa[i - 1], w) ^ 1));
    }
  }
  for (int i = 1; i <= n; ++ i) printf("%d ", sa[i]);
  return 0;
}

线性构建

在大多数题目中,常数较小的倍增是完全够用的。走某些特殊题目中可以使用 DC3/SA-IS 算法实现线性构建后缀数组。
具体做法可以参考:诱导排序与 SA-IS 算法DC3:[2009]后缀数组——处理字符串的有力工具 by. 罗穗骞

LCP 问题

特别鸣谢:论文爷!后缀数组-许智磊

一些定义

lcp(S,T)\operatorname{lcp}(S,T) 定义为字符串 SSTT 的最长公共前缀 (Longest common prefix), 即最大的 lmin{S,T}l\le \min\{\left| S\right|,\left| T\right|\},满足 Si=Ti(1il)S_i=T_i(1\le i\le l)
在许多后缀数组相关问题中,都需要它的帮助。
下文以 lcp(i,j)\operatorname{lcp}(i,j) 表示后缀 iijj 的最长公共前缀,并延续上文中一些概念:saisa_i:排名为 ii 的后缀,rkirk_i:后缀 ii 的排名。
并将会用 saisa_i 直接代表排名为 ii 的后缀,即 sai=S[sai:n]sa_i = S[sa_i:n]
定义一些新的概念。 heighti\operatorname{height}_i 表示排名为 i1i-1ii 的两后缀的最长公共前缀。
heighti=lcp(sai1,sai)\operatorname{height}_i = \operatorname{lcp}(sa_{i-1},sa_i)
hih_i 表示后缀 ii 和排名在 ii 之前一位的后缀的最长公共前缀。
hi=heightrki=lcp(sarki1,sarki)=lcp(i,sarki1)h_i=\operatorname{height}_{rk_i} = \operatorname{lcp}(sa_{rk_i-1}, sa_{rk_i})= \operatorname{lcp}(i, sa_{rk_i -1})

引理:LCP Lemma

1i<j<kn,lcp(sai,sak)=min{lcp(sai,saj),lcp(saj,sak)} \forall 1\le i<j<k\le n, \,\operatorname{lcp}(sa_i,sa_k) = \min\{\operatorname{lcp}(sa_i,sa_j), \operatorname{lcp}(sa_j,sa_k)\}
此引理是证明其他引理的基础。
证明,设 p=min{lcp(sai,saj),lcp(saj,sak)}p = \min\{\operatorname{lcp}(sa_i,sa_j), \operatorname{lcp}(sa_j,sa_k)\},则有:
lcp(sai,saj)p,lcp(saj,sak)p\operatorname{lcp}(sa_i,sa_j)\ge p,\, \operatorname{lcp}(sa_j,sa_k)\ge p
sai[1:p]=saj[1:p]=sak[1:p]sa_i[1:p] = sa_j[1:p] = sa_k[1:p],可得 lcp(sai,sak)p\operatorname{lcp}(sa_i,sa_k)\ge p
再考虑反证法,设 lcp(sai,sak)=q>p\operatorname{lcp}(sa_i,sa_k) =q > p。则 sai[1:q]=sak[1:q]sa_i[1:q]=sa_k[1:q],有 sai[p+1]=sak[p+1]sa_i[p+1]=sa_k[p+1]。 对 pp 的取值分类讨论:
  1. p=lcp(sai,saj)<lcp(saj,sak)p=\operatorname{lcp}(sa_i,sa_j) < \operatorname{lcp}(sa_j,sa_k):则有 sai[p+1]<saj[p+1]=sak[p+1]sa_i[p+1] < sa_j[p+1] = sa_k[p+1]
  2. p=lcp(saj,sak)<lcp(sai,saj)p=\operatorname{lcp}(sa_j,sa_k) < \operatorname{lcp}(sa_i,sa_j):则有 sai[p+1]=saj[p+1]<sak[p+1]sa_i[p+1] = sa_j[p+1] < sa_k[p+1]
  3. p=lcp(saj,sak)=lcp(sai,saj)p=\operatorname{lcp}(sa_j,sa_k) = \operatorname{lcp}(sa_i,sa_j):则有 sai[p+1]<saj[p+1]<sak[p+1]sa_i[p+1] < sa_j[p+1] < sa_k[p+1]
sai[p+1]<sak[p+1]sa_i[p+1]<sa_k[p+1] 恒成立,与已知矛盾,则 lcp(sai,sak)p\operatorname{lcp}(sa_i,sa_k)\le p。 综合上述两个结论,得证引理成立。

引理:LCP Theorem

1i<jn,lcp(sai,saj)=mink=i+1j{heightk} \forall 1\le i < j\le n,\, \operatorname{lcp}(sa_i,sa_j) = \min_{k=i+1}^j\{\operatorname{height_k}\}
由 LCP Lemma,可知显然成立。
根据这个优美的式子,求解任意两个后缀的 lcp\operatorname{lcp} 变为求解 height\operatorname{height} 的区间最值问题。
可通过 st 表 实现 O(nlogn)O(n\log n) 预处理,O(1)O(1) 查询。
问题只剩下如何快速求 height\operatorname{height} 了。

推论:LCP Corollary

lcp(sai,saj)lcp(sai,sak)(ij<k) \operatorname{lcp}(sa_i,sa_j) \ge \operatorname{lcp}(sa_i, sa_k)\, (i\le j<k)
表示排名不相邻的两个后缀的 lcp\operatorname{lcp} 不超过它们之间任何相邻元素的 lcp\operatorname{lcp}
证明由引理 LCP Lemma 显然可得。
但是涛哥钦定我写一下证明,那我就不胜惶恐地写了(
类似 LCP Lemma,考虑反证法。设 lcp(sai,saj)<lcp(sai,sak)\operatorname{lcp}(sa_i,sa_j)< \operatorname{lcp}(sa_i, sa_k),则有下图:
Lb
考虑字典序比较的过程。若 sai<sajsa_i < sa_j,则有 sai[lcp(sai,saj)+1]<saj[lcp(sai,saj)+1]sa_i[{\operatorname{lcp}(sa_i,sa_j)+1}] <sa_j[{\operatorname{lcp}(sa_i,sa_j) + 1}]
即图中的字符 x<yx<y
此时考虑比较 sajsa_jsaksa_k 的字典序。由图,显然有 lcp(saj,sak)=lcp(sai,saj)\operatorname{lcp}(sa_j,sa_k) = \operatorname{lcp}(sa_i,sa_j)。而 lcp(sai,sak)>lcp(sai,saj)\operatorname{lcp}(sa_i,sa_k) > \operatorname{lcp}(sa_i,sa_j),则 sak[lcp(saj,sak)+1]=xsa_k[{\operatorname{lcp}(sa_j,sa_k)+1}] = x
x<yx<y,可得 saksa_k 的字典序小于 sajsa_j
与已知矛盾,反证原结论成立。

引理

1in,hihi11 \forall 1\le i\le n,\, h_i\ge h_{i-1}-1
hi=heightrki=lcp(sarki1,sarki)=lcp(i,sarki1)h_i=\operatorname{height}_{rk_i} = \operatorname{lcp}(sa_{rk_i-1}, sa_{rk_i})= \operatorname{lcp}(i, sa_{rk_i -1})
用来快速计算 height\operatorname{height} 的引理,个人喜欢叫它不完全单调性。
证明考虑数学归纳。首先当 hi11h_{i-1}\le 1 时,结论显然成立,因为 hi0h_i \ge 0
hi1>1h_{i-1}>1 时,设 u=i,v=sarki1u = i, \, v = sa_{rk_i-1},有 hi=lcp(u,v)h_i = \operatorname{lcp}(u,v)。同时,设 u=i1,v=sarki11u' = i-1, \, v' = sa_{rk_{i-1}-1},有 hi1=lcp(u,v)h_{i-1} = \operatorname{lcp}(u',v')
hi1=lcp(u,v)>1h_{i-1} = \operatorname{lcp}(u',v')>1,则 u,vu',v' 必有公共前缀。
考虑 删去 u,vu',v'第一个 字符,设其分别变成 x,yx,y。显然 lcp(x,y)=hi11\operatorname{lcp}(x,y) = h_{i-1}-1,且仍满足字典序 y<xy<x
u=i1u' = i-1,则删去第一个字符后,xx 等于后缀 ii
则对于他们在 sasa 中的排名,有 y<x=i=uy<x=i=u
sasa 中,vvuu 前一位置,则有 yv<uy\le v<u。根据 LCP Corollary,有:
hi=lcp(u,v)lcp(u,y)=lcp(x,y)=hi11h_i = \operatorname{lcp}(u,v)\ge \operatorname{lcp}(u,y) = \operatorname{lcp}(x,y) = h_{i-1}-1
得证。

快速求 height

定义 hi=heightrkih_i = \operatorname{height}_{rk_i},只需快速求出 hh,便可 O(n)O(n) 地获得 height\operatorname{height}
由引理已知 1in,hihi11\forall 1\le i\le n,\, h_i\ge h_{i-1}-1
hi=lcp(i,sarki1)h_i=\operatorname{lcp}(i, sa_{rk_i -1}) 具有不完全单调性,考虑正序枚举 ii 进行递推。
rki=1rk_i=1 时, sarki1sa_{rk_i-1} 不存在,特判 hi=0h_i=0
i=1i=1,暴力比较出 lcp(i,sarki1)\operatorname{lcp}(i,sa_{rk_i-1}),比较次数 <n<n
若上述情况均不满足,由引理知,hi=lcp(i,sarki1)hi11h_i=\operatorname{lcp}(i,sa_{rk_i-1})\ge h_{i-1}-1,两后缀前 hi11h_{i-1}-1 位相同。
可从第 hi1h_{i-1} 位开始比较两后缀计算出 hih_i,比较次数 =hihi1+2=h_i-h_{i-1}+2
代码中并没有专门开 hh 数组,其中hi=kh_i = k
CPP
void GetHeight() {
  for (int i = 1, k = 0; i <= n; ++ i) {
    if (rk[i] == 1) k = 0;
    else {
      if (k > 0) k --;
      int j = sa[rk[i] - 1];
      while (i + k <= n && j + k <= n && 
             S[i + k] == S[j + k]) {
        ++ k;
      }
    }
    height[rk[i]] = k;
  }
}
knk\le n,最多减 nn 次,则最多会在比较中加 2n2n 次。总复杂度为 O(n)O(n) 级别。

例题

「JSOI2007」字符加密

无法简述的题面。
断环成链,把字符串复制一遍扔到后面,跑 SA 即可。
板子背诵检查,可以练下手感。

SP705 SUBST1 - New Distinct Substrings

TT 组数据,每次给定一个字符串 ss,求该字符串本质不同的子串数量。
两个子串本质不同,当且仅当两个子串长度不等,或长度相等但有任意一位不同。
1T1s5×1041\le T\le 1\le|s|\le 5\times 10^4
280ms,1.46GB。
一种想法是用所有子串的个数 n(n+1)2\frac{n(n+1)}{2} 减去重复子串的个数,显然重复的串一定出现在某两个后缀的公共前缀部分。
考虑加入 saisa_i 后,新增的本质不同的子串的数量,显然即 length(sai)length(lcp(sai,sai1))\operatorname{length}(sa_i) - \operatorname{length}(\operatorname{lcp}(sa_i, sa_{i-1})),代表不作为之前加入的后缀的前缀的,saisa_i 的前缀的数量。最终答案即:
n(n+1)2i=2nheighti\frac{n(n+1)}{2} - \sum_{i = 2}^{n}\operatorname{height}_i SA 简单实现即可,总复杂度 O(nlogn)O(n\log n)
如果想了解直观的证明解释可以阅读这篇文章:「笔记」后缀树
CPP
//知识点:SA 
/*
By:Luckyblock
*/
#include <algorithm>
#include <cctype>
#include <cstdio>
#include <cstring>
#define LL long long
const int kN = 1e5 + 10;
//=============================================================
char s[kN];
int n, m, sa[kN], rk[kN << 1], oldrk[kN << 1], height[kN];
int id[kN], cnt[kN], rkid[kN];
//=============================================================
inline int read() {
  int f = 1, w = 0;
  char ch = getchar();
  for (; !isdigit(ch); ch = getchar())
    if (ch == '-') f = -1;
  for (; isdigit(ch); ch = getchar()) {
    w = (w << 3) + (w << 1) + (ch ^ '0');
  }
  return f * w;
}
void Chkmax(int &fir_, int sec_) {
  if (sec_ > fir_) fir_ = sec_;
}
void Chkmin(int &fir_, int sec_) {
  if (sec_ < fir_) fir_ = sec_;
}
bool cmp(int x_, int y_, int w_) {
  return oldrk[x_] == oldrk[y_] && 
         oldrk[x_ + w_] == oldrk[y_ + w_];
}
void GetHeight() {
  for (int i = 1, k = 0; i <= n; ++ i) {
    if (rk[i] == 1) k = 0;
    else {
      if (k > 0) -- k;
      int j = sa[rk[i] - 1];
      while (i + k <= n && j + k <=n && 
             s[i + k] == s[j + k]) {
               ++ k;
      }
    }
    height[rk[i]] = k;
  }
}
void SuffixSort() {
  scanf("%s", s + 1);
  m = 300;
  n = strlen(s + 1);
  
  memset(cnt, 0, sizeof (cnt));
  for (int i = 1; i <= n; ++ i) cnt[rk[i] = s[i]] ++;
  for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
  for (int i = n; i >= 1; -- i) sa[cnt[rk[i]] --] = i;
  
  for (int p, w = 1; w < n; w <<= 1) {
    p = 0;
    for (int i = n; i > n - w; -- i) id[++ p] = i;
    for (int i = 1; i <= n; ++ i) {
      if (sa[i] > w) id[++ p] = sa[i] - w;
    }
    
    memset(cnt, 0, sizeof (cnt));
    for (int i = 1; i <= n; ++ i) cnt[rkid[i] = rk[id[i]]] ++;
    for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
    for (int i = n; i >= 1; -- i) sa[cnt[rkid[i]] --] = id[i];
    
    m = 0;
    memcpy(oldrk, rk, sizeof (rk));
    for (int i = 1; i <= n; ++ i) {
      m += (cmp(sa[i], sa[i - 1], w) ^ 1);
      rk[sa[i]] = m;
    }
  }
  GetHeight();
}
//=============================================================
int main() {
  int T = read();
  while (T --) {
    SuffixSort();
    LL ans = 1ll * n * (n + 1) / 2ll; 
    for (int i = 1; i <= n; ++ i) ans -= height[i];
    printf("%lld\n", ans);
  }
  return 0;
}

SP1811 LCS - Longest Common Substring

给定两字符串 S1,S2S_1, S_2,求它们的最长公共子串长度。
S1,S22.5×105|S_1|,|S_2|\le 2.5\times 10^5
294ms,1.46GB。
套路地把两个字符串连起来,答案即:
max1iS1<jS1+S2lcp(i,j)\max_{1\le i\le |S_1| < j\le |S_1+S_2|}\operatorname{lcp}(i,j)
显然答案即为满足 sai,sai1sa_i,sa_{i-1} 分属不同字符串heighti\operatorname{height}_{i} 的最大值。
正确性非常显然,留与读者自证。这里给出一种证明,可以参考这里:「双串最长公共子串」
CPP
//知识点:SA
/*
By:Luckyblock
*/
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <ctype.h>
#define ll long long
const int kMaxn = 5e5 + 10;
//=============================================================
char S[kMaxn];
int n1, n, m, ans, cnt[kMaxn], id[kMaxn], rkid[kMaxn];
int sa[kMaxn], rk[kMaxn << 1], oldrk[kMaxn << 1], height[kMaxn];
int MaxHeight[kMaxn][20], Log2[kMaxn];
//=============================================================
inline int read() {
  int f = 1, w = 0; char ch = getchar();
  for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1;
  for (; isdigit(ch); ch = getchar()) w = (w << 3) + (w << 1) + (ch ^ '0');
  return f * w;
}
void GetMax(int &fir, int sec) {
  if (sec > fir) fir = sec;
}
bool cmp(int x, int y, int w) { //判断两个子串是否相等。
  return oldrk[x] == oldrk[y] && 
         oldrk[x + w] == oldrk[y + w]; 
}
void GetHeight() {
  for (int i = 1, k = 0; i <= n; ++ i) {
    if (rk[i] == 1) k = 0;
    else {
      if (k > 0) k --;
      int j = sa[rk[i] - 1];
      while (i + k <= n && j + k <= n && 
             S[i + k] == S[j + k]) {
        ++ k;
      }
    }
    height[rk[i]] = k;
  }
}
void SuffixSort() {
  m = 300;
  for (int i = 1; i <= n; ++ i) ++ cnt[rk[i] = S[i]];
  for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
  for (int i = n; i >= 1; -- i) sa[cnt[rk[i]] --] = i;
  for (int p, w = 1; w < n; w <<= 1) {
    p = 0;
    for (int i = n; i > n - w; -- i) id[++ p] = i;
    for (int i = 1; i <= n; ++ i) {
      if (sa[i] > w) id[++ p] = sa[i] - w;
    }
    memset(cnt, 0, sizeof (cnt));
    for (int i = 1; i <= n; ++ i) ++ cnt[(rkid[i] = rk[id[i]])];
    for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
    for (int i = n; i >= 1; -- i) sa[cnt[rkid[i]] --] = id[i];
    std ::swap(rk, oldrk);
    m = 0;
    for (int i = 1; i <= n; ++ i) {
      m += (cmp(sa[i], sa[i - 1], w) ^ 1);
      rk[sa[i]] = m;
    }
  }
  GetHeight();
}
bool Judge(int x, int y) {
  return (x <= n1 && y > n1) || (x > n1 && y < n1);
}
//=============================================================
int main() {
  scanf("%s", S + 1); n1 = strlen(S + 1);
  S[n1 + 1] = 'z' + 1;
  scanf("%s", S + n1 + 1 + 1); n = strlen(S + 1);
  SuffixSort();
  for (int i = 2; i <= n; ++ i) {
    if (Judge(sa[i - 1], sa[i])) GetMax(ans, height[i]);
  }
  printf("%d", ans);
  return 0;
}

「HAOI2016」找相同字符

给定两字符串 S1,S2S_1, S_2,求出在两字符串中各取一个子串,使得这两个子串相同的方案数。
两方案不同当且仅当这两个子串中有一个位置不同。
1S1,S22×1051\le |S_1|, |S_2|\le 2\times 10^5
1S,256MB。
考察对 lcp\operatorname{lcp} 单调性的理解。
S1S_1 后面加个终止符,S2S_2 串扔到 S1S_1 后面,跑 SA。
显然答案即后半段的后缀,与前半段的后缀的所有 lcp\operatorname{lcp} 之和。
考虑按字典序枚举后半段的后缀,设当前枚举到的后缀为 saisa_i
仅考虑 字典序 <sai<sa_i 的 前半段的后缀 saj (j<i)sa_j\ (j<i) 的贡献。其对 saisa_i 的贡献为 lcp(sai,saj)\operatorname{lcp}(sa_i, sa_j)
lcp\operatorname{lcp} 的单调性,对于 最小 的大于 saisa_i 的后半段的后缀 sak(k>i)sa_k(k>i),有 lcp(sak,saj)lcp(sai,saj)\operatorname{lcp}(sa_{k}, sa_j)\le \operatorname{lcp}(sa_i,sa_j),考虑贡献的变化情况。
lcp(sak,saj)<lcp(sai,saj)\operatorname{lcp}(sa_{k}, sa_j)< \operatorname{lcp}(sa_i,sa_j),则 sajsa_jsaksa_k 的贡献应变为: lcp(sak,saj)=min{lcp(sai,saj),minl=i+1kheightl}\operatorname{lcp}(sa_k, sa_j) = \min\{\operatorname{lcp}(sa_i,sa_j), \min\limits_{l=i+1}^{k}{\operatorname{height}_l}\}
此外,若存在 sal,l(i,k)sa_l, l\in (i,k)前半段的后缀 时,作出贡献的元素增加。
考虑在枚举后缀的过程中,用权值线段树维护 字典序小于 saisa_i前半段 的后缀 saj (j<i)sa_j\ (j<i) 的不同长度的 lcp\operatorname{lcp}数量
上述两操作,即为区间赋值 与 单点插入。
再按字典序倒序枚举后缀,计算字典序 >sai>sa_i 的 前半段的后缀的贡献。
分析很屑,代码有详细注释。
总复杂度 O(nlogn)O(n\log n)。线段树写法是自己 YY 的,比较无脑,也可以单调栈简单维护,复杂度也为 O(nlogn)O(n\log n) 级别。
此外还有优美的广义 SAM 写法,可以参考:「HAOI2016」找相同字符
CPP
//知识点:SA,线段树
/*
By:Luckyblock 
*/
#include <cstdio>
#include <ctype.h>
#include <cstring>
#include <algorithm>
#define ll long long
#define lson (now_<<1)
#define rson (now_<<1|1)
const int kMaxn = 4e5 + 10;
//=============================================================
char S[kMaxn];
int n1, n, m, sa[kMaxn], rk[kMaxn << 1], oldrk[kMaxn << 1], height[kMaxn];
int id[kMaxn], cnt[kMaxn], rkid[kMaxn];
ll ans, size[kMaxn << 2], sum[kMaxn << 2]; //size 维护数量,sum 维护 lcp 之和。
bool tag[kMaxn << 2];
//=============================================================
inline int read() {
  int f = 1, w = 0; char ch = getchar();
  for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1;
  for (; isdigit(ch); ch = getchar()) w = (w << 3) + (w << 1) + (ch ^ '0');
  return f * w;
}
bool cmp(int x, int y, int w) { //判断两个子串是否相等。
  return oldrk[x] == oldrk[y] && 
         oldrk[x + w] == oldrk[y + w]; 
}
void GetHeight() {
  for (int i = 1, k = 0; i <= n; ++ i) {
    if (rk[i] == 1) k = 0;
    else {
      if (k > 0) k --;
      int j = sa[rk[i] - 1];
      while (i + k <= n && j + k <= n && 
             S[i + k] == S[j + k]) {
        ++ k;
      }
    }
    height[rk[i]] = k;
  }
}
void SuffixSort() {
  m = std :: max(n, 300);
  for (int i = 1; i <= n; ++ i) ++ cnt[rk[i] = S[i]];
  for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
  for (int i = n; i >= 1; -- i) sa[cnt[rk[i]] --] = i;
  for (int p, w = 1; w < n; w <<= 1) {
    p = 0;
    for (int i = n; i > n - w; -- i) id[++ p] = i;
    for (int i = 1; i <= n; ++ i) {
      if (sa[i] > w) id[++ p] = sa[i] - w;
    }
    memset(cnt, 0, sizeof (cnt));
    for (int i = 1; i <= n; ++ i) ++ cnt[(rkid[i] = rk[id[i]])];
    for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1];
    for (int i = n; i >= 1; -- i) sa[cnt[rkid[i]] --] = id[i];
    std ::swap(rk, oldrk);
    m = 0;
    for (int i = 1; i <= n; ++ i) {
      m += (cmp(sa[i], sa[i - 1], w) ^ 1);
      rk[sa[i]] = m;
    }
  }
  GetHeight();
}
void Build(int now_, int L_, int R_) {
  size[now_] = sum[now_] = 0ll;
  tag[now_] = false;
  if (L_ == R_) return ;
  int mid = (L_ + R_) >> 1;
  Build(lson, L_, mid), Build(rson, mid + 1, R_);
}
void Pushdown(int now_) {
  tag[lson] = tag[rson] = true;
  size[lson] = size[rson] = 0;
  sum[lson] = sum[rson] = 0;
  tag[now_] = false;
}
void Pushup(int now_) {
  size[now_] = size[lson] + size[rson];
  sum[now_] = sum[lson] + sum[rson];
}
ll Delete(int now_, int L_, int R_, int ql_, int qr_) {
  if (ql_ <= L_ && R_ <= qr_) {
    ll ret = size[now_];
    tag[now_] = true;
    size[now_] = sum[now_] = 0;
    return ret;
  }
  if(tag[now_]) Pushdown(now_);
  int mid = (L_ + R_) >> 1;
  ll ret = 0ll;
  if (ql_ <= mid) ret += Delete(lson, L_, mid, ql_, qr_);
  if (qr_ > mid) ret += Delete(rson, mid + 1, R_, ql_, qr_);
  Pushup(now_);
  return ret;
}
void Insert(int now_, int L_, int R_, int pos_, ll num) {
  if (! num) return ;
  if (L_ == R_) {
    size[now_] += num;
    sum[now_] += 1ll * num * (L_ - 1ll); //注意减去偏移量。
    return ;
  }
  if (tag[now_]) Pushdown(now_);
  int mid = (L_ + R_) >> 1;
  if (pos_ <= mid) Insert(lson, L_, mid, pos_, num);
  else Insert(rson, mid + 1, R_, pos_, num);
  Pushup(now_);
}
//=============================================================
int main() {
  scanf("%s", S + 1); n1 = strlen(S + 1);
  S[n1 + 1] = 'z' + 1;
  scanf("%s", S + n1 + 2); n = strlen(S + 1);
  SuffixSort();

  //正序枚举所有后缀,计算字典序 >sa_i 的 前半段的后缀的贡献。
  //当枚举到一个 后半段的后缀,仅用于更新 min(lcp)。
  //枚举到一个 前半段的后缀,用于更新 min(lcp),且需新插入一个后缀。
  //由于 lcp 可能为 0,线段树维护的区间加了偏移量 1。
  for (int i = 2; i <= n; ++ i) {
    //计算 lcp > height(i) 的 前半段后缀的数量,并将他们删除。
    ll num = Delete(1, 1, n + 1, height[i] + 1 + 1, n + 1); 
    Insert(1, 1, n + 1, height[i] + 1, num + (sa[i - 1] <= n1)); //插入被删除的后缀 与 新后缀。注意边界。
    if (sa[i] > n1 + 1) ans += sum[1]; //若枚举到一个 后半段后缀,计算贡献。 注意边界。
  }
  Build(1, 1, n + 1); //清空线段树
  //倒序枚举所有后缀,计算字典序 >sa_i 的 前半段的后缀的贡献。
  for (int i = n; i >= 2; -- i) {
    ll num = Delete(1, 1, n + 1, height[i] + 2, n + 1);
    Insert(1, 1, n + 1, height[i] + 1, num + (sa[i] <= n1)); //注意边界
    if (sa[i - 1] > n1 + 1) ans += sum[1]; //注意边界
  }
  printf("%lld", ans);
  return 0;
}

「AHOI2013」 差异

给定一长度为 nn 的字符串 SS,令 TiT_i 表示从第 ii 个字符开始的后缀,求:
1i<jn{len(Ti)+len(Tj)2×lcp(Ti,Tj)}\sum_{1\le i<j\le n}\{\operatorname{len}(T_i) +\operatorname{len}(T_j) - 2\times \operatorname{lcp} (T_i,T_j)\}
len(a)\operatorname{len}(a) 表示字符串 aa 的长度,lcp(a,b)\operatorname{lcp}(a,b) 表示字符串 a,ba,b 的最长公共前缀。
1n5×1051\le n\le 5\times 10^5
1S,512MB。
化下式子:
&\sum_{1\le i<j\le n}\{\operatorname{len}(T_i) +\operatorname{len}(T_j) - 2\times \operatorname{lcp} (T_i,T_j)\}\\ =& \sum_{1\le i<j\le n}\{(n-i+1) +(n-j+1) - 2\times \operatorname{lcp} (T_i,T_j)\}\\ =& \dfrac{(n-1)\times n \times (n+1)}{2} - 2\sum_{1\le i<j\le n}\operatorname{lcp} (T_i,T_j) \end{aligned}$$ 考虑如何快速求后一半,即所有 $\operatorname{lcp}$ 之和。 各后缀与 $sa$ 数组的元素,一一对应,则有下列等价关系: $$\sum_{1\le i<j\le n}\operatorname{lcp} (T_i,T_j) = \sum_{1\le i<j\le n}\operatorname{lcp}(sa_i, sa_j)$$ 类似上题的套路,考虑枚举 $sa_j$,用权值线段树维护 $sa_i (i<j)$ 的不同长度的 $\operatorname{lcp}(sa_i, sa_j)$ 的数量。 有一引理: $$\forall 1\le i < j\le n,\, \operatorname{lcp}(sa_i,sa_j) = \min\limits_{k=i+1}^j\{\operatorname{height_k}\}$$ 模拟引理,当 $j+1$ 时将权值线段树中所有 $>\operatorname{height}_{j+1}$ 的元素删除,并添加相同个数个 元素 $\operatorname{height}_{j+1}$。 添加一个 $\operatorname{height}_{j+1}$,代表新增的 $sa_j$ 的贡献。 贡献求和即可,总复杂度 $O(n\log n)$。 --- 线段树太傻逼了,考虑单调数据结构。发现有下列等价关系: $$\sum_{1\le i<j\le n}\operatorname{lcp}(sa_i, sa_j) = \sum_{1\le i<j\le n}\min_{k=i+1}^{j}\{\operatorname{height}_k\}$$ 即求 $\operatorname{height}$ 中所有子区间的区间最小值之和。 这是个经典问题,单调栈维护 $\operatorname{height}$ 作为最小值的区间的 最大左右端点,答案即 $\sum\limits_{i=2}^{n}(i-l_i)\times (r_i-i)\times \operatorname{height}_i$。 总复杂度 $O(n\log n)$。注意特判区间长度不能为 1。 线段树代码可以参考:[这里](https://www.cnblogs.com/luckyblock/p/13520374.html)。 ```cpp //知识点:SA,单调性 /* By:Luckyblock */ #include <cstdio> #include <ctype.h> #include <cstring> #include <iostream> #include <algorithm> #define ll long long const int kMaxn = 5e5 + 10; //============================================================= char S[kMaxn]; int n, m, sa[kMaxn], rk[kMaxn << 1], oldrk[kMaxn << 1], height[kMaxn]; int cnt[kMaxn], id[kMaxn], rkid[kMaxn]; int top, st[kMaxn], l[kMaxn], r[kMaxn]; //============================================================= inline int read() { int f = 1, w = 0; char ch = getchar(); for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1; for (; isdigit(ch); ch = getchar()) w = (w << 3) + (w << 1) + (ch ^ '0'); return f * w; } void GetMax(int &fir, int sec) { if (sec > fir) fir = sec; } void GetMin(int &fir, int sec) { if (sec < fir) fir = sec; } int cmp(int x, int y, int w) { return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w]; } void GetHeight() { for (int i = 1, k = 0; i <= n; ++ i) { if (rk[i] == 1) k = 0; else { if (k > 0) k --; int j = sa[rk[i] - 1]; while (i + k <= n && j + k <= n && S[i + k] == S[j + k]) { ++ k; } } height[rk[i]] = k; } } void SuffixSort() { n = strlen(S + 1); m = 1010; for (int i = 1; i <= n; ++ i) cnt[rk[i] = S[i]] ++; for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1]; for (int i = n; i; -- i) sa[cnt[rk[i]] --] = i; for (int p, w = 1; w < n; w <<= 1) { p = 0; for (int i = n; i > n - w; -- i) id[++ p] = i; for (int i = 1; i <= n; ++ i) { if (sa[i] > w) id[++ p] = sa[i] - w; } memset(cnt, 0, sizeof (cnt)); for (int i = 1; i <= n; ++ i) cnt[rkid[i] = rk[id[i]]] ++; for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1]; for (int i = n; i; -- i) sa[cnt[rkid[i]] --] = id[i]; std :: swap(rk, oldrk); m = 0; for (int i = 1; i <= n; ++ i) { m += (cmp(sa[i], sa[i - 1], w) ^ 1); rk[sa[i]] = m; } } GetHeight(); } //============================================================= int main() { scanf("%s", S + 1); SuffixSort(); ll ans = 1ll * ((n - 1ll) * n / 2ll) * (n + 1ll) ; st[(top = 1)] = 1; for (int i = 2; i <= n; ++ i) { while (top && height[st[top]] > height[i]) { r[st[top]] = i; top --; } l[i] = st[top]; st[++ top] = i; } while (top) r[st[top --]] = n + 1; for (int i = 2; i <= n; ++ i) { ans -= 2ll * (i - l[i]) * (r[i] - i) * height[i]; } printf("%lld", ans); return 0; } ``` ### [「TJOI / HEOI2016」字符串](https://loj.ac/problem/2059) >给定一长度为 $n$ 的字符串,$m$ 个询问。 >每次询问给定参数 $a,b,c,d$,求子串 $S[a:b]$ 的**所有子串**,与子串 $S[c:d]$ 的最长公共前缀的最大值。 >$1\le n,m\le 10^5, 1\le a\le b\le n, 1\le c\le d\le n$。 >2S,256MB。 对于每一个询问,答案满足单调性,考虑二分答案。 设 $l$ 为当前二分到的最长的,子串 $S[a:b]$ 的 **所有子串**,与子串 $S[c:d]$ 的最长公共前缀。 分析可知,若 $l$ 合法,那么会存在至少一个后缀 $x$ 满足: - 开头在 $[a:b-l+1]$ 中。 - $\operatorname{lcp}(x, c)\ge l$。 对于第二个限制,考虑 $\operatorname{lcp}$ 的单调性。$\operatorname{lcp}(x,c)$ 是一个在 $c$ 处取极值的单峰函数。 则满足条件的 $x$ 的取值,一定是 $sa$ 数组上连续的一段。 套路地对 $\operatorname{height}$ 建立 st 表,即可 $O(1)$ 查询 $\operatorname{lcp}$,于是可以通过二分排名快速得到后缀 $x$ 排名的取值范围,将限制二也转化为了区间限制形式。 限制一限制了后缀的区间,限制二限制了 $rk$ 的区间。查询这样的后缀的存在性,变成了一个静态二维数点问题。 对 $sa$ 数组建立主席树维护即可,总复杂度 $O(n\log^2 n)$。 ```cpp //知识点:SA,二分答案,主席树 /* By:Luckyblock */ #include <algorithm> #include <cstdio> #include <cstring> #include <ctype.h> #define ll long long const int kMaxn = 1e5 + 10; //============================================================= char S[kMaxn]; int n, m, ans, cnt[kMaxn], id[kMaxn], rkid[kMaxn]; int sa[kMaxn], rk[kMaxn << 1], oldrk[kMaxn << 1], height[kMaxn]; int MaxHeight[kMaxn][20], Log2[kMaxn];; int lson[kMaxn << 5], rson[kMaxn << 5], size[kMaxn << 5]; int node_num, root[kMaxn]; //============================================================= inline int read() { int f = 1, w = 0; char ch = getchar(); for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1; for (; isdigit(ch); ch = getchar()) w = (w << 3) + (w << 1) + (ch ^ '0'); return f * w; } void GetMax(int &fir, int sec) { if (sec > fir) fir = sec; } bool cmp(int x, int y, int w) { //判断两个子串是否相等。 return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w]; } void GetHeight() { for (int i = 1, k = 0; i <= n; ++ i) { if (rk[i] == 1) k = 0; else { if (k > 0) k --; int j = sa[rk[i] - 1]; while (i + k <= n && j + k <= n && S[i + k] == S[j + k]) { ++ k; } } height[rk[i]] = k; } } int QueryLcp(int l_, int r_) { int k = Log2[r_ - l_ + 1]; return std :: min(MaxHeight[l_][k], MaxHeight[r_ - (1 << k) + 1][k]); } void MakeSt() { for (int i = 2; i <= n; ++ i) MaxHeight[i][0] = height[i]; for (int i = 2; i <= n; ++ i) { Log2[i] = Log2[i >> 1] + 1; } for (int j = 1; j < 20; ++ j) { for (int i = 1; i + (1 << j) - 1 <= n; ++ i) { MaxHeight[i][j] = std :: min(MaxHeight[i][j - 1], MaxHeight[i + (1 << (j - 1))][j - 1]); } } } void SuffixSort() { int M = 300; for (int i = 1; i <= n; ++ i) ++ cnt[rk[i] = S[i]]; for (int i = 1; i <= M; ++ i) cnt[i] += cnt[i - 1]; for (int i = n; i >= 1; -- i) sa[cnt[rk[i]] --] = i; for (int p, w = 1; w < n; w <<= 1) { p = 0; for (int i = n; i > n - w; -- i) id[++ p] = i; for (int i = 1; i <= n; ++ i) { if (sa[i] > w) id[++ p] = sa[i] - w; } memset(cnt, 0, sizeof (cnt)); for (int i = 1; i <= n; ++ i) ++ cnt[(rkid[i] = rk[id[i]])]; for (int i = 1; i <= M; ++ i) cnt[i] += cnt[i - 1]; for (int i = n; i >= 1; -- i) sa[cnt[rkid[i]] --] = id[i]; std ::swap(rk, oldrk); M = 0; for (int i = 1; i <= n; ++ i) { M += (cmp(sa[i], sa[i - 1], w) ^ 1); rk[sa[i]] = M; } } GetHeight(); MakeSt(); } void Insert(int pre_, int &now_, int L_, int R_, int val_) { now_ = ++ node_num; size[now_] = size[pre_] + 1; lson[now_] = lson[pre_], rson[now_] = rson[pre_]; if(L_ >= R_) return ; int mid = (L_ + R_) >> 1; if(val_ > mid) Insert(rson[pre_], rson[now_], mid + 1, R_, val_); else Insert(lson[pre_], lson[now_], L_, mid, val_); } int Query(int u_, int v_, int L_, int R_, int ql_, int qr_) { if (ql_ <= L_ && R_ <= qr_) return size[v_] - size[u_]; int mid = (L_ + R_) >> 1, ret = 0; if (ql_ <= mid) ret += Query(lson[u_], lson[v_], L_, mid, ql_, qr_); if (qr_ > mid) ret += Query(rson[u_], rson[v_], mid + 1, R_, ql_, qr_); return ret; } bool Judge(int len_, int a_, int b_, int c_) { int l = 1, r = rk[c_], up, down; while (l < r) { int mid = (l + r) >> 1; if (QueryLcp(mid + 1, rk[c_]) < len_) l = mid + 1; else r = mid; } up = r, l = rk[c_], r = n; while (l < r) { int mid = (l + r + 1) >> 1; if (QueryLcp(rk[c_] + 1, mid) < len_) r = mid - 1; else l = mid; } down = r; return Query(root[up - 1], root[down], 1, n, a_, b_ - len_ + 1) > 0; } int Solve(int a_, int b_, int c_, int d_) { int l = 0, r = std :: min(b_ - a_ + 1, d_ - c_ + 1); while (l < r) { int len = (l + r + 1) >> 1; if (Judge(len, a_, b_, c_)) l = len; else r = len - 1; } return r; } //============================================================= int main() { n = read(), m = read(); scanf("%s", S + 1); SuffixSort(); for (int i = 1; i <= n; ++ i) Insert(root[i - 1], root[i], 1, n, sa[i]); for (int i = 1; i <= m; ++ i) { int a = read(), b = read(), c = read(), d = read(); printf("%d\n", Solve(a, b, c, d)); } return 0; } ``` ### [「NOI2015」品酒大会](https://loj.ac/problem/2133) 「そして谁もいなくなるか?」 >给定一字符串 $S$,位置 $i$ 的属性值为 $a_i$。 >定义位置 $p,q$ 为「 $r$ 相似」,当且仅当 $S[p:p+r-1] = S[q:q+r-1]$ 成立。 >特别地,对于任意 $1\le p,q\le n, p\not ={q}$,$p,q$ 都是「 $0$ 相似」的。 >求:选出两个 「 $0\sim n-1$ 相似」 的位置的 **方案数**,及选出两个 「 $0\sim n-1$ 相似」的位置属性值 **乘积的最大值**。 >1S,256MB。 显然,若两个位置 $i,j$ 是「$r$ 相似」的,那么它们也是「$0\sim (r-1)$ 相似」的。它们会对「$0\sim r$ 相似」的答案做出贡献,于是考虑倒序枚举「$r$ 相似」的位置计算。 发现「$r$ 相似」问题的实质即 $\operatorname{lcp}$ 问题,先对原串跑 SA,求得 $\operatorname{height}$ 数组。 考虑「$r$ 相似」的定义,则对于任意两位置 $i,j$,他们最大是 「$\operatorname{lcp}(S[i:n],S[j:n])$ 相似」的。 --- 根据引理: $$\forall 1\le i < j\le n,\, \operatorname{lcp}(sa_i,sa_j) = \min_{k=i+1}^j\{\operatorname{height_k}\}$$ 考虑按照 $\operatorname{height}$ 将后缀排序后的后缀进行划分。 若 $\operatorname{height}_i\ge r$,将 $sa_{i-1}$ 与 $sa_i$ 划入一个集合,否则划入不同的集合。这样划分后,对于所有大小 $\ge 2$ 的集合,集合中所有后缀两两的 $\operatorname{lcp}\ge r$,它们都会对 「$r$ 相似」的答案做出贡献。 将所有集合的贡献合并,即为 「$r$ 相似」的答案。 定义上述划分方式为 「$r$ 划分」,考虑如何在此基础上获得 「$r-1$ 划分」。 显然,只需将 $\operatorname{height}_i = r-1$ 的后缀 $sa_{i-1}$ 和 $sa_i$ 所在集合合并即可。 考虑将 $\operatorname{height}$ 降序排序,用并查集维护集合,按上述过程依次进行合并,即可依次得到「$n\sim 1$ 相似」的答案。 --- 考虑如何维护集合的贡献。 先考虑第一问,对于「$r$ 划分」中一个大小 $\ge 2$ 的集合,集合中任意两后缀的 $\operatorname{lcp} \ge r$,该集合的贡献即为 $(\operatorname{size}-1)\times \operatorname{size}$。 合并时直接将 $\operatorname{size}$ 累加即可。 再考虑第二问,由于可能存在 $a_i<0$,考虑维护集合的最大值,次大值,最小值,次小值。为保证答案合法,四个值中的任意两个都不能来自 **同一个位置**。 集合的贡献为 $\max\{max_1\times max_2, min_1 \times min_2\}$。 合并时注意四个值的大小关系,代码中使用了 multiset 维护。 复杂度瓶颈是倍增,总复杂度 $O(n\log n)$ 级别,使用炫酷 DC3/SA-IS 魔术可做到 $O(n)$ 级别的复杂度。 ```cpp //知识点:SA,并查集 /* By:Luckyblock */ #include <algorithm> #include <cstdio> #include <cstring> #include <ctype.h> #include <set> #define ll long long const int kMaxn = 3e5 + 10; //============================================================= char S[kMaxn]; int n, m, a[kMaxn]; int cnt[kMaxn], id[kMaxn], rkid[kMaxn]; int sa[kMaxn], rk[kMaxn << 1], oldrk[kMaxn << 1], height[kMaxn]; ll ans1[kMaxn], ans2[kMaxn]; int fa[kMaxn], size[kMaxn]; std :: multiset <int> s[kMaxn]; //============================================================= inline int read() { int f = 1, w = 0; char ch = getchar(); for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1; for (; isdigit(ch); ch = getchar()) w = (w << 3) + (w << 1) + (ch ^ '0'); return f * w; } void GetMax(ll &fir, ll sec) { if (sec > fir) fir = sec; } bool cmp(int x, int y, int w) { //判断两个子串是否相等。 return oldrk[x] == oldrk[y] && oldrk[x + w] == oldrk[y + w]; } void GetHeight() { for (int i = 1, k = 0; i <= n; ++ i) { if (rk[i] == 1) k = 0; else { if (k > 0) k --; int j = sa[rk[i] - 1]; while (i + k <= n && j + k <= n && S[i + k] == S[j + k]) { ++ k; } } height[rk[i]] = k; } } void SuffixSort() { m = 300; for (int i = 1; i <= n; ++ i) ++ cnt[rk[i] = S[i]]; for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1]; for (int i = n; i >= 1; -- i) sa[cnt[rk[i]] --] = i; for (int p, w = 1; w < n; w <<= 1) { p = 0; for (int i = n; i > n - w; -- i) id[++ p] = i; for (int i = 1; i <= n; ++ i) { if (sa[i] > w) id[++ p] = sa[i] - w; } memset(cnt, 0, sizeof (cnt)); for (int i = 1; i <= n; ++ i) ++ cnt[(rkid[i] = rk[id[i]])]; for (int i = 1; i <= m; ++ i) cnt[i] += cnt[i - 1]; for (int i = n; i >= 1; -- i) sa[cnt[rkid[i]] --] = id[i]; std ::swap(rk, oldrk); m = 0; for (int i = 1; i <= n; ++ i) { m += (cmp(sa[i], sa[i - 1], w) ^ 1); rk[sa[i]] = m; } } GetHeight(); } bool Compare(int fir, int sec) { return height[fir] > height[sec]; } int Find(int x) { return fa[x] == x ? x : fa[x] = Find(fa[x]); } void Union(int x, int y, int z) { x = Find(x), y = Find(y); if (size[x] < size[y]) std :: swap(x, y); ans1[z] += 1ll * size[x] * size[y]; for (std :: set <int> :: iterator it = s[y].begin(); it != s[y].end(); ++ it) { s[x].insert(* it); } int t[5]; t[1] = *s[x].begin(), t[2] = *(++ s[x].begin()); t[3] = *(-- s[x].end()), t[4] = *(-- (-- s[x].end())); GetMax(ans2[z], std :: max(1ll * t[1] * t[2], 1ll * t[3] * t[4])); fa[y] = x; size[x] += size[y]; if (s[x].size() > 5) { s[x].clear(); for (int i = 1; i <= 4; ++ i) s[x].insert(t[i]); } } //============================================================= int main() { n = read(); scanf("%s", S + 1); for (int i = 1; i <= n; ++ i) a[i] = read(); SuffixSort(); memset(ans2, - 63, sizeof (ans2)); for (int i = 2; i <= n; ++ i) id[i] = i; for (int i = 1; i <= n; ++ i) fa[i] = i; for (int i = 1; i <= n; ++ i) size[i] = 1; std :: sort(id + 2, id + n + 1, Compare); for (int i = 1; i <= n; ++ i) s[i].insert(a[i]); for (int i = 2; i <= n; ++ i) { Union(sa[id[i] - 1], sa[id[i]], height[id[i]]); } for (int i = n - 1; i >= 0; -- i) { ans1[i] += ans1[i + 1]; GetMax(ans2[i], ans2[i + 1]); } for (int i = 0; i < n; ++ i) { printf("%lld %lld\n", ans1[i], ans1[i] ? ans2[i] : 0); } return 0; } ``` ### [「十二省联考 2019」字符串问题](https://loj.ac/p/3049) >$T$ 组数据,每次给定一字符串 $S$。 >在 $S$ 中存在 $n_a$ 个 A 类子串 $(la_i, ra_i)$,$n_b$ 个 B 类子串 $(lb_i,rb_i)$。且存在 $m$ 组支配关系,支配关系 $(x,y)$ 表示第 $x$ 个 A 类串支配第 $y$ 个 B 类串。 >要求构造一个目标串 $T$,满足: >- $T$ 由若干 A 类串拼接而成。 >- 对于分割中所有相邻的串,后一个串存在一个被前一个串支配的前缀。 > >求该目标串的最大长度,若目标串可以无限长输出 $-1$。 >$1\le T\le 100$,$n_a,n_b,|S|,m\le 2\times 10^5$。 >6S,1G。 首先建立图论模型,从每个 A 类子串向其支配的 B 串连边,从每个 B 串向以它为前缀的 A 串连边。A 串节点的权值为其长度,B 串节点权值为 0。 在图上 DP 求得最长路即为答案,若图中存在环则无解。 第一类边有 $m$ 条,但第二类边数可以达到 $n_an_b$ 级别,考虑优化建图。 对于某 A 串 $(la_i, ra_i)$,它以 B 串 $(lb_j, rb_j)$ 作为一个前缀的充要条件是 $\operatorname{lcp}(S[la_i:n],S[lb_j:n]) \ge rb_j-lb_j+1$ 且 $ra_i - la_i + 1\ge rb_j-lb_j+1$。 对于限制一,考虑求得 $S$ 的 SA,对 $\operatorname{height}$ 建立 ST 表,可在 $sa$ 上二分求得满足 $\operatorname{lcp}\ge rb_j-lb_j+1$ 的区间的左右边界,满足条件的 A 串一定都在这段区间内。第二类边转化为区间连边问题。 此时不考虑限制二直接线段树优化建图,可以拿到 80 分的好成绩。 限制二实际上限定了 B 连边的对象的长度。 考虑将所有 A,B 串按长度递减排序,按长度递减枚举 A 串并依次加入可持久化线段树。 对于每一个 B 串,先找到使得 A 串长度大于其长度的最晚的历史版本,此时线段树上的所有 A 串长度都大于其长度,再向这棵线段树上的节点连边。 时间复杂度 $O((|S| + n_a + n_b)\log n)$,空间复杂度 $O(m + (|S| + n_a + n_b)\log n)$,不需要刻意卡常就能稳过。 注意边数在极限数据下可以达到 $10^7$ 级别,不注意空间大小和清空时的实现会被卡到 60。这个时空限制显然就是给选手乱搞的,数组往大了开就行。 在线段树优化建图中,实点会在建树操作中就与虚点连好边。本题的实点是代表 A,B 串的节点,在本题的可持久化线段树优化中,实点与虚点的连边发生在动态开点的插入过程中。 在新建节点时,需要将该节点连向上一个版本中对应位置的节点。 对于 A 串 $(la_i, ra_i)$,它应该被插入到线段树中 $rk_{la_i}$ 的位置,即叶节点 $rk_{la_i}$ 与该实点相连。 开 `long long`。 ```cpp //知识点:SA,可持久化线段树,优化建图,DAGDP /* By:Luckyblock */ #include <algorithm> #include <cctype> #include <cstdio> #include <cstring> #include <queue> #define LL long long const int kN = 2e5 + 10; //============================================================= struct Str { int l, r, lth, id; } subs[kN << 1]; int node_num, n, na, nb, m, into[kN <<5]; int e_num, head[kN << 5], v[50 * kN], ne[50 * kN]; LL val[kN << 5], f[kN << 5]; char s[kN]; //============================================================= inline int read() { int f = 1, w = 0; char ch = getchar(); for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -1; for (; isdigit(ch); ch = getchar()) { w = (w << 3) + (w << 1) + (ch ^ '0'); } return f * w; } void Chkmax(LL &fir_, LL sec_) { if (sec_ > fir_) fir_ = sec_; } void Chkmin(int &fir_, int sec_) { if (sec_ < fir_) fir_ = sec_; } bool cmp(Str fir_, Str sec_) { if (fir_.lth != sec_.lth) return fir_.lth > sec_.lth; return fir_.id < sec_.id; } void Add(int u_, int v_) { v[++ e_num] = v_, ne[e_num] = head[u_], head[u_] = e_num; into[v_] ++; } namespace ST { int Minn[kN][21], Log2[kN]; void MakeST(int *a_) { for (int i = 1; i <= n; ++ i) Minn[i][0] = a_[i]; for (int i = 2; i <= n; ++ i) Log2[i] = Log2[i >> 1] + 1; for (int j = 1; j <= 20; ++ j) { for (int i = 1; i + (1 << j) - 1 <= n; ++ i) { // Minn[i][j] = std::min(Minn[i][j - 1], Minn[i + (1 << (j - 1))][j - 1]); } } } int Query(int l_, int r_) { int k = Log2[r_ - l_ + 1]; return std::min(Minn[l_][k], Minn[r_ - (1 << k) + 1][k]); } } namespace SA { int sa[kN], rk[kN << 1]; int oldrk[kN << 1], cnt[kN], id[kN]; int height[kN]; void SuffixSort() { int rknum = std::max(n, 300); memset(cnt, 0, sizeof (cnt)); for (int i = 1; i <= n; ++ i) cnt[rk[i] = s[i]] ++; for (int i = 1; i <= rknum; ++ i) cnt[i] += cnt[i - 1]; for (int i = n; i >= 1; -- i) sa[cnt[rk[i]] --] = i; for (int w = 1, p; w < n; w <<= 1) { p = 0; for (int i = n; i > n - w; -- i) id[++ p] = i; for (int i = 1; i <= n; ++ i) { if (sa[i] > w) id[++ p] = sa[i] - w; } memset(cnt, 0, sizeof (cnt)); for (int i = 1; i <= n; ++ i) ++ cnt[rk[id[i]]]; for (int i = 1; i <= rknum; ++ i) cnt[i] += cnt[i - 1]; for (int i = n; i >= 1; -- i) sa[cnt[rk[id[i]]] --] = id[i]; for (int i = 1; i <= n; ++ i) oldrk[i] = rk[i]; rknum = 0; for (int i = 1; i <= n; ++ i) { rknum += (oldrk[sa[i]] == oldrk[sa[i - 1]] && oldrk[sa[i] + w] == oldrk[sa[i - 1] + w]) ^ 1; rk[sa[i]] = rknum; } } } void GetHeight() { for (int i = 1, k = 0; i <= n; ++ i) { if (rk[i] == 1) { k = 0; } else { if (k) -- k; int j = sa[rk[i] - 1]; while (i + k <= n && j + k <= n && s[i + k] == s[j + k]) { ++ k; } } height[rk[i]] = k; } } int Lcp(int x_, int y_) { if (x_ > y_) std::swap(x_, y_); return ST::Query(x_ + 1, y_); } void Init() { SuffixSort(); GetHeight(); ST::MakeST(SA::height); } } namespace Hjt { #define ls lson[now_] #define rs rson[now_] #define mid ((L_+R_)>>1) int root[kN], lson[kN << 5], rson[kN << 5]; void Insert(int &now_, int pre_, int L_, int R_, int pos_, int id_) { now_ = ++ node_num; ls = lson[pre_], rs = rson[pre_]; if (pre_) Add(now_, pre_); if (L_ == R_) { Add(now_, id_); return ; } if (pos_ <= mid) { Insert(ls, lson[pre_], L_, mid, pos_, id_); Add(now_, ls); } else { Insert(rs, rson[pre_], mid + 1, R_, pos_, id_); Add(now_, rs); } } void AddEdge(int now_, int L_, int R_, int l_, int r_, int id_) { if (! now_) return ; if (l_ <= L_ && R_ <= r_) { Add(id_, now_); return ; } if (l_ <= mid) AddEdge(ls, L_, mid, l_, r_, id_); if (r_ > mid) AddEdge(rs, mid + 1, R_, l_, r_, id_); } #undef ls #undef rs #undef mid } void Init() { e_num = 0; for (int i = 0; i <= node_num; ++ i) { head[i] = val[i] = into[i] = f[i] = 0; } scanf("%s", s + 1); n = strlen(s + 1); SA::Init(); na = read(); for (int i = 1; i <= na; ++ i) { int l_ = read(), r_ = read(); subs[i] = (Str) {l_, r_, r_ - l_ + 1, i}; val[i] = subs[i].lth; } nb = read(); for (int i = 1; i <= nb; ++ i) { int l_ = read(), r_ = read(); subs[na + i] = (Str) {l_, r_, r_ - l_ + 1, na + i}; } m = read(); for (int i = 1; i <= m; ++ i) { int u_ = read(), v_ = read(); Add(u_, v_ + na); //Add(read(), read()+na) 会倒着读 } node_num = na + nb; } bool Check(int x_, int y_, int lth_) { return SA::Lcp(x_, y_) >= lth_; } void AddEdgeB(int id_, int now_) { int pos = SA::rk[subs[id_].l], l_ = pos, r_ = pos; //l_,r_ 初始值 for (int l = 1, r = pos - 1; l <= r; ) { int mid = (l + r) >> 1; if (Check(mid, pos, subs[id_].lth)) { r = mid - 1; l_ = mid; } else { l = mid + 1; } } for (int l = pos + 1, r = n; l <= r; ) { int mid = (l + r) >> 1; if (Check(pos, mid, subs[id_].lth)) { l = mid + 1; r_ = mid; } else { r = mid - 1; } } Hjt::AddEdge(Hjt::root[now_], 1, n, l_, r_, subs[id_].id); } void Build() { node_num = na + nb; std::sort(subs + 1, subs + na + nb + 1, cmp); for (int now = 0, i = 1; i <= na + nb; ++ i) { if (subs[i].id > na) { AddEdgeB(i, now); continue; } ++ now; Hjt::Insert(Hjt::root[now], Hjt::root[now - 1], 1, n, SA::rk[subs[i].l], subs[i].id); } } void TopSort() { std::queue <int> q; for (int i = 1; i <= node_num; ++ i) { if (!into[i]) { f[i] = val[i]; q.push(i); } } while (! q.empty()) { int u_ = q.front(); q.pop(); for (int i = head[u_]; i; i = ne[i]) { int v_ = v[i]; Chkmax(f[v_], f[u_] + val[v_]); -- into[v_]; if (!into[v_]) q.push(v_); } } LL ans = 0; for (int i = 1; i <= node_num; ++ i) { Chkmax(ans, f[i]); if (into[i]) { printf("-1\n"); return ; } } printf("%lld\n", ans); } //============================================================= int main() { int T = read(); while (T --) { Init(); Build(); TopSort(); } return 0; } ``` ## 与后缀树的联系 后缀树相关介绍可以参考:[「笔记」后缀树](https://www.cnblogs.com/luckyblock/p/14222851.html)。 SA 可以看作由后缀树的所有叶结点按字典序排列得到的数组。 SA 的 $\operatorname{height}$ 数组可以看做后缀树中相邻两关键点的 $\operatorname{lca}$,从这个角度上可以直观地理解 SA 的许多性质。 同时,这也证明了后缀树的适用范围一定比 SA 大。 奈何 SA 配合 $\operatorname{height}$ 可以实现大部分后缀树的功能,又好写,时间上与后缀树又几乎相同,空间又小,所以我喜欢 SA( 做题的时候可以先直观地从后缀树的角度出发思考,再用 SA 进行实现。 可以避免抽象的思考过程,比较适合我这样的无脑选手( 后缀数组结合虚树的思想可以用于构建后缀树,详见这篇博客:[利用后缀数组构造后缀树_AZUI](https://blog.csdn.net/u011542204/article/details/51231962)。 ## 写在最后 参考资料: [OI-wiki SA](https://oi-wiki.org/string/sa/) [后缀数组详解 - 自为风月马前卒](https://www.cnblogs.com/zwfymqz/p/8413523.html) [「后缀排序SA」学习笔记 - Rainy7](https://www.luogu.com.cn/blog/Rainy7/post-hou-zhui-pai-xu-sa-xue-xi-bi-ji) [后缀数组-许智磊](https://wenku.baidu.com/view/0dc03d2b1611cc7931b765ce0508763230127479.html) [后缀数组学习笔记 _ Menci's Blog](https://oi.men.ci/suffix-array-notes/) [利用后缀数组构造后缀树_AZUI](https://blog.csdn.net/u011542204/article/details/51231962) [P5284 [十二省联考2019]字符串问题 - xht37 的洛谷博客](https://www.luogu.com.cn/blog/xht37/p5284-shi-er-xing-lian-kao-2019-zi-fu-chuan-wen-ti)

评论

25 条评论,欢迎与作者交流。

正在加载评论...