#include <bits/stdc++.h> template <typename... Ts> void __print(Ts &&...ts) {} #ifdef DEBUG #include "print.hpp" #endif // DEBUG using namespace std; template <class T> using pq_max = priority_queue<T>; template <class T> using pq_min = priority_queue<T, vector<T>, greater<T>>; #define FOR(i, a, b) for (int i = a; i < (b); ++i) #define F0R(i, a) for (int i = 0; i < (a); ++i) #define FORd(i, a, b) for (int i = (b)-1; i >= a; --i) #define F0Rd(i, a) for (int i = (a)-1; i >= 0; --i) #define uid(a, b) uniform_int_distribution<int>(a, b)(rng) mt19937 rng(chrono::steady_clock::now().time_since_epoch().count()); #define all(x) x.begin(), x.end() #define GET_FOR(_1, _2, _3, NAME, ...) NAME #define loop(...) GET_FOR(__VA_ARGS__, FOR, F0R)(__VA_ARGS__) #define GET_FORd(_1, _2, _3, NAME, ...) NAME #define loopr(...) GET_FOR(__VA_ARGS__, FORd, F0Rd)(__VA_ARGS__) #define sz(x) ((int)(x).size()) using ll = long long; template <typename T> using vec = vector<T>; template <typename T, size_t N> using arr = array<T, N>; using vi = vector<int>; using vvi = vector<vi>; using vl = vector<ll>; using vvl = vector<vl>; using pi = pair<int, int>; using pl = pair<ll, ll>; /////////////////////////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////////////////////// void solution() { int n; cin >> n; vl fishes(n); loop(i, n) { cin >> fishes[i]; } vi beg(n); beg[0] = 0; auto sorted = fishes; sort(all(sorted)); ll min_mass; if (sorted.front() == sorted.back()) { min_mass = int(1e9) + 1; } else { loop(i, 1, n) { ll a = sorted[i]; ll prev = sorted[i - 1]; if (prev == a) { beg[i] = beg[i - 1]; } else { beg[i] = i; } } vl sums(n + 1); sums[0] = 0; loop(i, n) { sums[i + 1] = sums[i] + sorted[i]; } min_mass = sorted.back(); loopr(i, n - 1) { if (sorted[i] + sums[beg[i]] > min_mass) { min_mass = sorted[i]; } } } for (int a : fishes) { cout << (a >= min_mass ? 'T' : 'N'); } cout << '\n'; } int main(void) { ios_base::sync_with_stdio(false); cin.tie(NULL); int qs = 1; // cin >> qs; while (qs--) { solution(); } 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 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 | #include <bits/stdc++.h> template <typename... Ts> void __print(Ts &&...ts) {} #ifdef DEBUG #include "print.hpp" #endif // DEBUG using namespace std; template <class T> using pq_max = priority_queue<T>; template <class T> using pq_min = priority_queue<T, vector<T>, greater<T>>; #define FOR(i, a, b) for (int i = a; i < (b); ++i) #define F0R(i, a) for (int i = 0; i < (a); ++i) #define FORd(i, a, b) for (int i = (b)-1; i >= a; --i) #define F0Rd(i, a) for (int i = (a)-1; i >= 0; --i) #define uid(a, b) uniform_int_distribution<int>(a, b)(rng) mt19937 rng(chrono::steady_clock::now().time_since_epoch().count()); #define all(x) x.begin(), x.end() #define GET_FOR(_1, _2, _3, NAME, ...) NAME #define loop(...) GET_FOR(__VA_ARGS__, FOR, F0R)(__VA_ARGS__) #define GET_FORd(_1, _2, _3, NAME, ...) NAME #define loopr(...) GET_FOR(__VA_ARGS__, FORd, F0Rd)(__VA_ARGS__) #define sz(x) ((int)(x).size()) using ll = long long; template <typename T> using vec = vector<T>; template <typename T, size_t N> using arr = array<T, N>; using vi = vector<int>; using vvi = vector<vi>; using vl = vector<ll>; using vvl = vector<vl>; using pi = pair<int, int>; using pl = pair<ll, ll>; /////////////////////////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////////////////////// void solution() { int n; cin >> n; vl fishes(n); loop(i, n) { cin >> fishes[i]; } vi beg(n); beg[0] = 0; auto sorted = fishes; sort(all(sorted)); ll min_mass; if (sorted.front() == sorted.back()) { min_mass = int(1e9) + 1; } else { loop(i, 1, n) { ll a = sorted[i]; ll prev = sorted[i - 1]; if (prev == a) { beg[i] = beg[i - 1]; } else { beg[i] = i; } } vl sums(n + 1); sums[0] = 0; loop(i, n) { sums[i + 1] = sums[i] + sorted[i]; } min_mass = sorted.back(); loopr(i, n - 1) { if (sorted[i] + sums[beg[i]] > min_mass) { min_mass = sorted[i]; } } } for (int a : fishes) { cout << (a >= min_mass ? 'T' : 'N'); } cout << '\n'; } int main(void) { ios_base::sync_with_stdio(false); cin.tie(NULL); int qs = 1; // cin >> qs; while (qs--) { solution(); } return 0; } |