#include <vector>
#include <iostream>
#include <array>
#include <cinttypes>
#include <cstring>
#include <climits>
#include <string>
#include <algorithm>
using namespace std;
namespace wzor
{
}
string solve_wzor(int n, int* input)
{
using namespace wzor;
int min_val = input[0];
int max_val = input[0];
vector<pair<int, int>> ryby(n);
for(int i = 0; i < n; ++i)
{
min_val = min(min_val, input[i]);
max_val = max(max_val, input[i]);
ryby[i] = {input[i], i};
}
auto ans = std::string(n, 'N');
if(min_val == max_val)
{
//Wszystkie elementy są równe, nie istnieje król morza
return ans;
}
sort(ryby.begin(), ryby.end());
//Szukamy pierwszego króla morza wiemy że taki istnieje
auto czy_moze_byc_krolem_morza = [&](int index)
{
int64_t my_val = ryby[index].first;
for(int i = 0; i < n; ++i)
{
if(index == i) continue;
if(my_val <= ryby[i].first) return false;
else my_val += ryby[i].first;
}
return true;
};
int a = 0, b = n - 1, s;
while(a < b)
{
int mid = a + (b - a) / 2;
if(czy_moze_byc_krolem_morza(mid))
{
b = mid;
}
else
{
a = mid + 1;
}
}
int minimalny_krol_morza = a;
for(int i = minimalny_krol_morza; i < n; ++i)
{
ans[ryby[i].second] = 'T';
}
return ans;
}
#include <iostream>
int input[500100];
int main()
{
int n;
std::cin >> n;
for(int i = 0; i < n; ++i)
{
std::cin >> input[i];
}
std::cout << solve_wzor(n, input) << '\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 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 | #include <vector> #include <iostream> #include <array> #include <cinttypes> #include <cstring> #include <climits> #include <string> #include <algorithm> using namespace std; namespace wzor { } string solve_wzor(int n, int* input) { using namespace wzor; int min_val = input[0]; int max_val = input[0]; vector<pair<int, int>> ryby(n); for(int i = 0; i < n; ++i) { min_val = min(min_val, input[i]); max_val = max(max_val, input[i]); ryby[i] = {input[i], i}; } auto ans = std::string(n, 'N'); if(min_val == max_val) { //Wszystkie elementy są równe, nie istnieje król morza return ans; } sort(ryby.begin(), ryby.end()); //Szukamy pierwszego króla morza wiemy że taki istnieje auto czy_moze_byc_krolem_morza = [&](int index) { int64_t my_val = ryby[index].first; for(int i = 0; i < n; ++i) { if(index == i) continue; if(my_val <= ryby[i].first) return false; else my_val += ryby[i].first; } return true; }; int a = 0, b = n - 1, s; while(a < b) { int mid = a + (b - a) / 2; if(czy_moze_byc_krolem_morza(mid)) { b = mid; } else { a = mid + 1; } } int minimalny_krol_morza = a; for(int i = minimalny_krol_morza; i < n; ++i) { ans[ryby[i].second] = 'T'; } return ans; } #include <iostream> int input[500100]; int main() { int n; std::cin >> n; for(int i = 0; i < n; ++i) { std::cin >> input[i]; } std::cout << solve_wzor(n, input) << '\n'; } |
English