#include <bits/stdc++.h> using namespace std; typedef long long ll; constexpr int MAX_N = 5e5 + 7; bool ANS[MAX_N]; ll sumy_prefiksowe[MAX_N]; int main() { ios_base::sync_with_stdio(false), cin.tie(NULL); int n; cin >> n; vector<pair<int, int>> masa_numer; masa_numer.push_back(make_pair(0, 0)); int masa; for (int i = 1; i <= n; ++i) { cin >> masa; masa_numer.push_back(make_pair(masa, i)); } sort(masa_numer.begin(), masa_numer.end()); for (int i = 1; i <= n; ++i) { sumy_prefiksowe[i] = sumy_prefiksowe[i-1] + masa_numer[i].first; } for (int i = 0; i <= n-1; ++i) { if (masa_numer[n-i].first != masa_numer[1].first) { ANS[masa_numer[n].second] = true; if (i > 0 && sumy_prefiksowe[n-i] > masa_numer[n-i+1].first) { ANS[masa_numer[n-i].second] = true; } else if (i > 0) break; } } for (int i = 1; i <= n; ++i) { if (ANS[i]) cout << 'T'; else cout << 'N'; } }
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 | #include <bits/stdc++.h> using namespace std; typedef long long ll; constexpr int MAX_N = 5e5 + 7; bool ANS[MAX_N]; ll sumy_prefiksowe[MAX_N]; int main() { ios_base::sync_with_stdio(false), cin.tie(NULL); int n; cin >> n; vector<pair<int, int>> masa_numer; masa_numer.push_back(make_pair(0, 0)); int masa; for (int i = 1; i <= n; ++i) { cin >> masa; masa_numer.push_back(make_pair(masa, i)); } sort(masa_numer.begin(), masa_numer.end()); for (int i = 1; i <= n; ++i) { sumy_prefiksowe[i] = sumy_prefiksowe[i-1] + masa_numer[i].first; } for (int i = 0; i <= n-1; ++i) { if (masa_numer[n-i].first != masa_numer[1].first) { ANS[masa_numer[n].second] = true; if (i > 0 && sumy_prefiksowe[n-i] > masa_numer[n-i+1].first) { ANS[masa_numer[n-i].second] = true; } else if (i > 0) break; } } for (int i = 1; i <= n; ++i) { if (ANS[i]) cout << 'T'; else cout << 'N'; } } |