Построение массива суффиксов - PullRequest
0 голосов
/ 19 октября 2019

Я читаю о построении суффиксного массива из [https://cp -algorithms.com / string / suffix-array.html # toc-tgt-3]

Вотcode

using namespace std;

vector<int> sort_cyclic_shifts(string const& s) {
   int n = s.size();
   const int alphabet = 256;
     vector<int> p(n), c(n), cnt(max(alphabet, n), 0);
   for (int i = 0; i < n; i++)
       cnt[s[i]]++;
   for (int i = 1; i < alphabet; i++)
       cnt[i] += cnt[i-1];
   for (int i = 0; i < n; i++)
       p[--cnt[s[i]]] = i;
   c[p[0]] = 0;
   int classes = 1;
   for (int i = 1; i < n; i++) {
       if (s[p[i]] != s[p[i-1]])
           classes++;
       c[p[i]] = classes - 1;
   }
    vector<int> pn(n), cn(n);
   for (int h = 0; (1 << h) < n; ++h) {
       for (int i = 0; i < n; i++) {
           pn[i] = p[i] - (1 << h);
           if (pn[i] < 0)
               pn[i] += n;
       }
       fill(cnt.begin(), cnt.begin() + classes, 0);
       for (int i = 0; i < n; i++)
           cnt[c[pn[i]]]++;
       for (int i = 1; i < classes; i++)
           cnt[i] += cnt[i-1];
       for (int i = n-1; i >= 0; i--)
           p[--cnt[c[pn[i]]]] = pn[i];
       cn[p[0]] = 0;
       classes = 1;
       for (int i = 1; i < n; i++) {
           pair<int, int> cur = make_pair(c[p[i]], c[(p[i] + (1 << h)) % n]);
           pair<int, int> prev = make_pair(c[p[i-1]], c[(p[i-1] + (1 << h)) % n]);
           if (cur != prev)
               ++classes;
           cn[p[i]] = classes - 1;
       }
       c.swap(cn);
   }
   return p;
}

 int main() {
   ios_base::sync_with_stdio(false);
   cin.tie(NULL);
   cout.tie(NULL);
   //code here;
   string s;
   cin>>s;
   vector<int> v=sort_cyclic_shifts(s);
   for(int i=0;i<v.size();i++) cout<<v[i]<<" ";
   return 0;
} 

Но алгоритм не дает правильного ответа на строку, содержащую любые два последовательных символа. Я не знаю почему?

Sample input:
ccccc
Sample output:
2 1 0 4 3
Correct output:
4 3 2 1 0

Может кто-нибудь объяснить, почему код дает неправильный ответ и как это исправить?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...