Advertisement
newb_ie

Suffix array -CP-Algo Implementation

Dec 17th, 2020
175
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.41 KB | None | 0 0
  1. /*
  2. ======================
  3. [     ___T_          ]
  4. [    | 6=6 | =>HI :-)]
  5. [    |__o__|         ]
  6. [ >===]__o[===<      ]
  7. [     [o__]          ]
  8. [      .".           ]
  9. [      |_|           ]
  10. [                    ]
  11. ======================
  12. */
  13.  
  14. #include <bits/stdc++.h>
  15. using namespace std;
  16.  
  17. vector<int> sort_cyclic_shifts (string const& s) {
  18.     int n = (int) s.size();
  19.     const int alphabet = 256;
  20.     vector<int> p(n),c(n),cnt(max(alphabet,n),0);
  21.     for (int i = 0; i < n; ++i) ++cnt[s[i]];
  22.     for (int i = 1; i < alphabet; ++i) cnt[i] += cnt[i - 1];
  23.     for (int i = 0; i < n; ++i) p[--cnt[s[i]]] = i;
  24.     c[p[0]] = 0;
  25.     int classes = 1;
  26.     for (int i = 1; i < n; ++i) {
  27.         if (s[p[i]] != s[p[i - 1]]) ++ classes;
  28.         c[p[i]] = classes - 1;
  29.     }
  30.     vector<int> pn(n),cn(n);
  31.     for (int k = 0; (1 << k) < n; ++k) {
  32.         for (int i = 0; i < n; ++i) {
  33.             pn[i] = p[i] - (1 << k);
  34.             if (pn[i] < 0) {
  35.                 pn[i] += n;
  36.             }
  37.         }
  38.         fill(cnt.begin(),cnt.begin() + classes,0);
  39.         for (int i = 0; i < n; ++i) ++cnt[c[pn[i]]];
  40.         for (int i = 1; i < classes; ++i) cnt[i] += cnt[i - 1];
  41.         for (int i = n - 1; i >= 0; --i) p[--cnt[c[pn[i]]]] = pn[i];
  42.         cn[p[0]] = 0;
  43.         classes = 1;
  44.         for (int i = 1; i < n; ++i) {
  45.             pair<int,int> cur = make_pair(c[p[i]],c[(p[i] + (1 << k)) % n]);
  46.             pair<int,int> prev = make_pair(c[p[i - 1]],c[(p[i - 1] + (1 << k)) % n]);
  47.             if (cur != prev) ++classes;
  48.             cn[p[i]] = classes - 1;
  49.         }
  50.         c.swap(cn);
  51.     }
  52.     return p;
  53. }
  54.  
  55. vector<int> suffix_array_construction (string s) {
  56.     s += '$';
  57.     vector<int> sorted_shifts = sort_cyclic_shifts(s);
  58.     sorted_shifts.erase(sorted_shifts.begin());
  59.     return sorted_shifts;
  60. }
  61.  
  62. int main () {
  63.      ios::sync_with_stdio(false);
  64.      cin.tie(nullptr);
  65.      cout.tie(nullptr);
  66.      int T = 1;
  67.      //~ cin >> T;
  68.      for (int test_case = 1; test_case <= T; ++test_case) {
  69.          string s;
  70.          cin >> s;
  71.          vector<int> p = suffix_array_construction(s);
  72.          int q;
  73.          cin >> q;
  74.          for (int query = 1; query <= q; ++query) {
  75.              string in;
  76.              cin >> in;
  77.              int l = 0,r = (int) p.size() - 1;
  78.              bool yes = false;
  79.              while (l <= r) {
  80.                  int mid = l + (r - l) / 2;
  81.                  string sub = s.substr(p[mid],(int) in.size());
  82.                  if (sub == in) {
  83.                      yes = true;
  84.                      break;
  85.                  }
  86.                  if (sub > in) {
  87.                      r = mid - 1;
  88.                  } else {
  89.                      l = mid + 1;
  90.                  }
  91.              }
  92.              cout << (yes ? "Yes\n" : "No\n");
  93.          }
  94.      }
  95.      //cerr << "Time elapsed :" << clock() * 1000.0 / CLOCKS_PER_SEC << " ms" << '\n';
  96. }
  97.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement