#include <cstdio> #include <cassert> #include <vector> #include <algorithm> typedef long long int LL; const LL INF = 1e18; const int MAXN = 1000005; int T[MAXN]; char W[MAXN]; bool visited[MAXN]; int resolved[MAXN]; template<class T> struct Array { int a; std::vector<T> data; Array(int a, int b, T init = T()): a(a), data(b-a+1, init) {} T& operator[](int i) { return data[i-a]; } }; int GetValue(char c) { return c == 'W' ? 1 : -1; } int main() { int n; scanf("%d", &n); for (int i = 0; i < n; i++) scanf("%d", &T[i]); int m; scanf("%d%s", &m, W); LL result = INF; for (int i = 0; i < m; i++) { if (!visited[i]) { visited[i] = true; int s = GetValue(W[i]); std::vector<int> sum(1, s); int it = i, min = s, max = s; while ((it + n) % m != i) { it = (it + n) % m; visited[it] = true; s += GetValue(W[it]); min = std::min(min, s); max = std::max(max, s); sum.push_back(s); } int size = sum.size(); for (int t = 0; t < size; t++) { sum.push_back(sum.back() + sum[t] - (t > 0 ? sum[t - 1] : 0)); min = std::min(min, sum.back()); max = std::max(max, sum.back()); } for (int t = 0; t < 2 * size; t++) sum.push_back(sum[t]); /*printf("cykl: "); for (int t = 0; t < size; t++) printf("%d ", sum[t]); printf("\n");*/ std::vector<int> next(sum.size()); Array<int> first(min, max, -1), last(min, max, -1); for (int t = 0; t < sum.size(); t++) { if (first[sum[t]] == -1) { first[sum[t]] = t; last[sum[t]] = t; } if (last[sum[t]] != -1) { next[last[sum[t]]] = t; last[sum[t]] = t; } } int p = 0, q = i; it = i; for (int p = 0; p < size; p++) { it = q % n; if (resolved[it] != i + 1) { resolved[it] = i + 1; //printf("resolved %d\n", it); //printf("poczatek cyklu w %d, q = %d\n", p, q); for (int t = it; t < n; t += m) { //printf("solving %d\n", t); int v = (p > 0 ? sum[p - 1] : 0) - T[t]; int whole_sum = sum[size - 1]; //printf("whole sum = %d, v = %d\n", whole_sum, v); if (v >= min) { assert(first[v] != -1); LL turn_count = first[v] - p + 1; //printf("latwo, wyszlo %d, v na pozycji %d, p = %d\n", turn_count, first[v], p); result = std::min(result, (LL)(turn_count - 1) * n + t + 1); } else if (whole_sum < 0) { whole_sum *= -1; int gap = min - v; int k = (gap + whole_sum - 1) / whole_sum; int nv = v + k * whole_sum; LL turn_count = (LL)k * size + first[nv] - p + 1; LL res = (turn_count - 1) * n + t + 1; //printf("krece %d razy, starczy %d tur\n", k, turn_count); result = std::min(result, (LL)(turn_count - 1) * n + t + 1); } //printf("result = %Ld\n", result); } } first[sum[p]] = next[p]; q = (q + n) % m; } } } if (result == INF) printf("-1\n"); else printf("%Ld\n", result); }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 | #include <cstdio> #include <cassert> #include <vector> #include <algorithm> typedef long long int LL; const LL INF = 1e18; const int MAXN = 1000005; int T[MAXN]; char W[MAXN]; bool visited[MAXN]; int resolved[MAXN]; template<class T> struct Array { int a; std::vector<T> data; Array(int a, int b, T init = T()): a(a), data(b-a+1, init) {} T& operator[](int i) { return data[i-a]; } }; int GetValue(char c) { return c == 'W' ? 1 : -1; } int main() { int n; scanf("%d", &n); for (int i = 0; i < n; i++) scanf("%d", &T[i]); int m; scanf("%d%s", &m, W); LL result = INF; for (int i = 0; i < m; i++) { if (!visited[i]) { visited[i] = true; int s = GetValue(W[i]); std::vector<int> sum(1, s); int it = i, min = s, max = s; while ((it + n) % m != i) { it = (it + n) % m; visited[it] = true; s += GetValue(W[it]); min = std::min(min, s); max = std::max(max, s); sum.push_back(s); } int size = sum.size(); for (int t = 0; t < size; t++) { sum.push_back(sum.back() + sum[t] - (t > 0 ? sum[t - 1] : 0)); min = std::min(min, sum.back()); max = std::max(max, sum.back()); } for (int t = 0; t < 2 * size; t++) sum.push_back(sum[t]); /*printf("cykl: "); for (int t = 0; t < size; t++) printf("%d ", sum[t]); printf("\n");*/ std::vector<int> next(sum.size()); Array<int> first(min, max, -1), last(min, max, -1); for (int t = 0; t < sum.size(); t++) { if (first[sum[t]] == -1) { first[sum[t]] = t; last[sum[t]] = t; } if (last[sum[t]] != -1) { next[last[sum[t]]] = t; last[sum[t]] = t; } } int p = 0, q = i; it = i; for (int p = 0; p < size; p++) { it = q % n; if (resolved[it] != i + 1) { resolved[it] = i + 1; //printf("resolved %d\n", it); //printf("poczatek cyklu w %d, q = %d\n", p, q); for (int t = it; t < n; t += m) { //printf("solving %d\n", t); int v = (p > 0 ? sum[p - 1] : 0) - T[t]; int whole_sum = sum[size - 1]; //printf("whole sum = %d, v = %d\n", whole_sum, v); if (v >= min) { assert(first[v] != -1); LL turn_count = first[v] - p + 1; //printf("latwo, wyszlo %d, v na pozycji %d, p = %d\n", turn_count, first[v], p); result = std::min(result, (LL)(turn_count - 1) * n + t + 1); } else if (whole_sum < 0) { whole_sum *= -1; int gap = min - v; int k = (gap + whole_sum - 1) / whole_sum; int nv = v + k * whole_sum; LL turn_count = (LL)k * size + first[nv] - p + 1; LL res = (turn_count - 1) * n + t + 1; //printf("krece %d razy, starczy %d tur\n", k, turn_count); result = std::min(result, (LL)(turn_count - 1) * n + t + 1); } //printf("result = %Ld\n", result); } } first[sum[p]] = next[p]; q = (q + n) % m; } } } if (result == INF) printf("-1\n"); else printf("%Ld\n", result); } |