#include <stdio.h>
#include <utility>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef pair<ll, ll> pll;
int main(void) {
size_t N; scanf("%lu", &N);
pll *arr = new pll[N];
for (size_t i=0; i<N; ++i) {
scanf("%lli", &arr[i].first);
arr[i].second = i;
}
sort(arr, arr+N);
ll *prefix = new ll[N];
bool all_same = false;
size_t i = 0;
do {
prefix[i] = arr[i].first;
++i;
} while (arr[i].first == arr[i-1].first && i < N);
if (i < N) { prefix[i] = i * arr[i-1].first + arr[i].first; ++i; }
else all_same = true;
for (; i < N; ++i) {
prefix[i] = prefix[i-1] + arr[i].first;
}
bool *king = new bool[N];
i = N-1;
if (!all_same) king[i] = true; // todo: if not the same as the others
for (--i; i != -1; --i) {
if (prefix[i] > arr[i+1].first)
king[i] = king[i+1];
else
king[i] = false;
}
// for (i = 0; i < N; ++i) {
// printf("%lli ", arr[i].first);
// }
// puts("");
// for (i = 0; i < N; ++i) {
// printf("%lli ", prefix[i]);
// }
// puts("");
// for (i = 0; i < N; ++i) {
// printf("%lu ", king[i]);
// }
// puts("");
// for (i = 0; i < N; ++i) {
// printf("%c ", king[i] == N-1 ? 'T' : 'N');
// }
// puts("");
pll *result = new pll[N];
for (i = 0; i < N; ++i) {
result[i] = { arr[i].second, king[i] };
}
sort(result, result+N);
for (i = 0; i < N; ++i) {
printf("%c", result[i].second ? 'T' : 'N');
}
puts("");
return 0;
}
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 | #include <stdio.h> #include <utility> #include <algorithm> using namespace std; typedef long long ll; typedef pair<ll, ll> pll; int main(void) { size_t N; scanf("%lu", &N); pll *arr = new pll[N]; for (size_t i=0; i<N; ++i) { scanf("%lli", &arr[i].first); arr[i].second = i; } sort(arr, arr+N); ll *prefix = new ll[N]; bool all_same = false; size_t i = 0; do { prefix[i] = arr[i].first; ++i; } while (arr[i].first == arr[i-1].first && i < N); if (i < N) { prefix[i] = i * arr[i-1].first + arr[i].first; ++i; } else all_same = true; for (; i < N; ++i) { prefix[i] = prefix[i-1] + arr[i].first; } bool *king = new bool[N]; i = N-1; if (!all_same) king[i] = true; // todo: if not the same as the others for (--i; i != -1; --i) { if (prefix[i] > arr[i+1].first) king[i] = king[i+1]; else king[i] = false; } // for (i = 0; i < N; ++i) { // printf("%lli ", arr[i].first); // } // puts(""); // for (i = 0; i < N; ++i) { // printf("%lli ", prefix[i]); // } // puts(""); // for (i = 0; i < N; ++i) { // printf("%lu ", king[i]); // } // puts(""); // for (i = 0; i < N; ++i) { // printf("%c ", king[i] == N-1 ? 'T' : 'N'); // } // puts(""); pll *result = new pll[N]; for (i = 0; i < N; ++i) { result[i] = { arr[i].second, king[i] }; } sort(result, result+N); for (i = 0; i < N; ++i) { printf("%c", result[i].second ? 'T' : 'N'); } puts(""); return 0; } |
English