Suffix array

Codigos/String/Suffix-Array/README.md

@@ -4,5 +4,9 @@ Estrutura que conterá inteiros que representam os índices iniciais de todos os
 
 Também constrói a tabela LCP (Longest Common Prefix).
 
+- `sa[i]` = Índice inicial do i-ésimo menor sufixo.
+- `ra[i]` = Rank do sufixo que começa em `i`.
+- `LCP[i]` = Comprimento do maior prefixo comum entre os sufixos `sa[i]` e `sa[i-1]`.
+
 * Complexidade de tempo (Pré-Processamento): $\mathcal{O}(|S| \cdot \log(|S|))$
 * Complexidade de tempo (Contar ocorrências de \(S\) em \(T\)): $\mathcal{O}(|S| \cdot \log(|T|))$

Codigos/String/Suffix-Array/suffix_array.cpp

@@ -1,8 +1,7 @@
-const int MAX_N = 5e5 + 5;
-
+const int MAX = 5e5 + 5;
 struct suffix_array {
     string s;
-    int n, sum, r, ra[MAX_N], sa[MAX_N], auxra[MAX_N], auxsa[MAX_N], c[MAX_N], lcp[MAX_N];
+    int n, sum, r, ra[MAX], sa[MAX], auxra[MAX], auxsa[MAX], c[MAX], lcp[MAX];
     void counting_sort(int k) {
         memset(c, 0, sizeof(c));
         for (int i = 0; i < n; i++) c[(i + k < n) ? ra[i + k] : 0]++;
@@ -15,12 +14,10 @@ struct suffix_array {
             counting_sort(k);
             counting_sort(0);
             auxra[sa[0]] = r = 0;
-            for (int i = 1; i < n; i++) {
-                auxra[sa[i]] =
-                    (ra[sa[i]] == ra[sa[i - 1]] && ra[sa[i] + k] == ra[sa[i - 1] + k])
-                        ? r
-                        : ++r;
-            }
+            for (int i = 1; i < n; i++)
+                if (ra[sa[i]] == ra[sa[i - 1]] && ra[sa[i] + k] == ra[sa[i - 1] + k])
+                    auxra[sa[i]] = r;
+                else auxra[sa[i]] = ++r;
             for (int i = 0; i < n; i++) ra[i] = auxra[i];
             if (ra[sa[n - 1]] == n - 1) break;
         }