#include using namespace std; // thank you, tookunn // http://tookunn.hatenablog.com/entry/2016/07/13/211148 #include #include #ifndef _WIN32 #include #endif class RMQ_SparseTable { public: // table[k][i] means minimum value in [i, i + 2^k) std::vector> table; RMQ_SparseTable(int* A, size_t N) { table = std::vector>(log(N) + 1, vector(N)); for (int i = 0; i < N; ++i) { table[0][i] = A[i]; } for (int k = 1; (1 << k) <= N; k++) { for (int i = 0; i + (1 << k) <= N; ++i) { table[k][i] = min(table[k - 1][i], table[k - 1][i + (1 << (k - 1))]); } } } // returns min value in [s, t) inline int getMin(int s, int t) { int range = t - s; int k = log(range); int value1 = table[k][s]; int value2 = table[k][t - (1 << k)]; return min(value1, value2); } inline static int log(int n) { //int ret = 0; //while (n = n >> 1) //{ // ret += 1; //} //return ret; #ifndef _WIN32 return std::__lg(n); #else if (n >= (1 << 16)) { return 16; } else if (n >= (1 << 15)) { return 15; } else if (n >= (1 << 14)) { return 14; } else if (n >= (1 << 13)) { return 13; } else if (n >= (1 << 12)) { return 12; } else if (n >= (1 << 11)) { return 11; } else if (n >= (1 << 10)) { return 10; } else if (n >= (1 << 9)) { return 9; } else if (n >= (1 << 8)) { return 8; } else if (n >= (1 << 7)) { return 7; } else if (n >= (1 << 6)) { return 6; } else if (n >= (1 << 5)) { return 5; } else if (n >= (1 << 4)) { return 4; } else if (n >= (1 << 3)) { return 3; } else if (n >= (1 << 2)) { return 2; } else if (n >= (1 << 1)) { return 1; } else { return 0; } #endif } }; #include #include int N; char s_data[1000000]; char* s[100000]; int s_len[100000]; int si[100000]; int M; long long x, d; int i, j; int index_table[100000]; int LCP_table[100000]; // same as pekempey's submission // http://yukicoder.me/submissions/172137 int di, dj, ii, jj; void next_init() { di = d / (N - 1); dj = d % (N - 1); ii = x / (N - 1); jj = x % (N - 1); i = ii; j = jj; if (i <= j) { ++j; } } inline void next() { jj += dj; if (jj >= N - 1) { jj -= N - 1; ++ii; } ii += di; if (ii >= N) { ii -= N; } i = ii; j = jj; if (i <= j) { ++j; } } // multikey-quicksort // http://d.hatena.ne.jp/echizen_tm/20100815/1281872393 // <- small int si_temp_s[100000]; // <- middle ........ large -> int si_temp_ml[100000]; // sort [l, r) void multikey_qsort(int l, int r, int depth) { if (r - l < 2) { //// insertion sort //si[l] = si[l]; //for (int ins_i = l + 1; ins_i < r; ++ins_i) //{ // int ins_v = si[ins_i]; // int i = 0; // for (; i < ins_i; ++i) // { // if (strcmp(s[si[i]].data() + depth, s[ins_v].data() + depth) > 0) // { // for (int j = ins_i; j > i; --j) // { // si[j] = si[j - 1]; // } // si[i] = ins_v; // break; // } // } // if (i == ins_i) // { // si[ins_i] = ins_v; // } //} //return; } else { char pivot = *(s[si[l]] + depth); int sorted = 0; int small = 0; int middle = 0; int large = 0; for (int ins_i = l; ins_i < r; ++ins_i) { int index = si[ins_i]; char c = *(s[index] + depth); if (c == '\0') { si[l + sorted++] = index; } else if (c < pivot) { si_temp_s[small++] = index; } else if (c == pivot) { si_temp_ml[middle++] = index; } else { si_temp_ml[r - ++large] = index; } } memcpy(si + (l + sorted), si_temp_s, sizeof(int) * small); memcpy(si + (l + sorted + small), si_temp_ml, sizeof(int) * middle); memcpy(si + (l + sorted + small + middle), si_temp_ml + (r - large), sizeof(int) * large); multikey_qsort(l + sorted, l + sorted + small, depth); multikey_qsort(l + sorted + small, l + sorted + small + middle, depth + 1); multikey_qsort(l + sorted + small + middle, r, depth); } } void multikey_qsort() { multikey_qsort(0, N, 0); } int LCP(const char* s1, const char* s2, size_t min_length) { int ret = 0; //const uint64_t* s64i = reinterpret_cast(s1); //const uint64_t* s64j = reinterpret_cast(s2); //while (ret < min_length / 8) //{ // if (s64i[ret] == s64j[ret]) // { // ++ret; // } // else // { // break; // } //} //ret *= 8; while (ret < min_length) { if (s1[ret] == s2[ret]) { ++ret; } else { break; } } return ret; } #ifdef _WIN32 #include DWORD time = GetTickCount(); #define TimePrint(x) cout << x " " << GetTickCount() - time << endl; time = GetTickCount(); #else #define TimePrint(x) ; #endif #ifdef __linux__ #define getchar getchar_unlocked #endif int main() { TimePrint("(start)"); // get input scanf("%lld", &N); { char* t = s_data; for (int i = 0; i < N; ++i) { while ((*t = getchar()) < 'a') {} s[i] = t; ++t; while ((*t = getchar()) >= 'a') ++t; *t = '\0'; s_len[i] = t - s[i]; ++t; } } scanf("%d %lld %lld", &M, &x, &d); TimePrint("input"); // sort for (int i = 0; i < N; ++i) { si[i] = i; } //sort(si, si + N, // [&](int i, int j) // { // return s[i] < s[j]; // }); multikey_qsort(); TimePrint("sort "); // reindex for (int i = 0; i < N; ++i) { index_table[si[i]] = i; } TimePrint("reidx"); // calc LCP for (int i = 0; i < N - 1; ++i) { char* s1 = s[si[i]]; int s1_len = s_len[si[i]]; char* s2 = s[si[i + 1]]; int s2_len = s_len[si[i + 1]]; LCP_table[i] = LCP(s1, s2, min(s1_len, s2_len)); } TimePrint("LCP "); // make Sparse Table RMQ_SparseTable sparse_table(LCP_table, N - 1); TimePrint("SpTab"); // calc ans long long sum = 0; next_init(); for (int indx = 0; indx < M; ++indx) { int i_ = index_table[i]; int j_ = index_table[j]; if (i_ > j_) std::swap(i_, j_); sum += sparse_table.getMin(i_, j_); next(); } TimePrint("clAns"); cout << sum << endl; }