Update suffix_array.cpp

This commit is contained in:
Anthony Wang 2019-09-04 13:49:18 -05:00 committed by GitHub
parent 12181073db
commit 8ad47e9d37
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,8 +1,3 @@
#include <algorithm>
#include <vector>
#include <string>
using namespace std;
vector<int> suffix_array(string& S) { vector<int> suffix_array(string& S) {
int N = S.length(); int N = S.length();
vector<int> SA(N), rank(N); vector<int> SA(N), rank(N);
@ -21,8 +16,7 @@ vector<int> suffix_array(string& S) {
vector<int> cnt(N); vector<int> cnt(N);
for (int i = 0; i < N; i++) cnt[i] = i; for (int i = 0; i < N; i++) cnt[i] = i;
for (int i = 0; i < N; i++) { for (int i = 0; i < N; i++) {
int s = tmp[i] - t; if (tmp[i] >= t) SA[cnt[rank[tmp[i] - t]]++] = tmp[i] - t;
if (s >= 0) SA[cnt[rank[s]]++] = s;
} }
} }
return SA; return SA;