Я читаю о построении суффиксного массива из [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
Может кто-нибудь объяснить, почему код дает неправильный ответ и как это исправить?