Advertisement
Guest User

Suffix Array (DC3) + LCP

a guest
Jul 13th, 2013
2,436
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.20 KB | None | 0 0
  1. char str [MAX_N];
  2. int N, m, SA [MAX_N], LCP [MAX_N];
  3. int x [MAX_N], y [MAX_N], w [MAX_N], c [MAX_N];
  4.  
  5. inline bool cmp (const int a, const int b, const int l) { return (y [a] == y [b] && y [a + l] == y [b + l]); }
  6.  
  7. void Sort () {
  8.     for (int i = 0; i < m; ++i) w [i] = 0;
  9.     for (int i = 0; i < N; ++i) ++w [x [y [i]]];
  10.     for (int i = 0; i < m - 1; ++i) w [i + 1] += w [i];
  11.     for (int i = N - 1; i >= 0; --i) SA [--w [x [y [i]]]] = y [i];
  12. }
  13.  
  14. void DA () {
  15.     for (int i = 0; i < N; ++i) x [i] = str [i], y[i] = i;
  16.     Sort ();
  17.     for (int i, j = 1, p = 1; p < N; j <<= 1, m = p) {
  18.         for (p = 0, i = N - j; i < N; i++) y [p++] = i;
  19.         for (int k = 0; k < N; ++k) if (SA [k] >= j) y [p++] = SA [k] - j;
  20.         Sort ();
  21.         for (swap (x, y), p = 1, x [SA [0]] = 0, i = 1; i < N; ++i) x [SA [i]] = cmp (SA [i - 1], SA [i], j) ? p - 1 : p++;
  22.     }
  23. }
  24.  
  25. void kasaiLCP () {
  26.     for (int i = 0; i < N; i++) c [SA [i]] = i;
  27.     for (int i = 0, j, k = 0; i < N; LCP [c [i++]] = k)
  28.         if (c [i] > 0) for (k ? k-- : 0, j = SA [c [i] - 1]; str [i + k] == str [j + k]; k++);
  29.         else k = 0;
  30. }
  31.  
  32. void suffixArray () {
  33.     m = 256;
  34.     N = strlen (str);
  35.     DA ();
  36.     kasaiLCP ();
  37. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement