#include <iostream>
#include <vector>
#include <string>
#include <numeric>
char canBeKing(std::vector<long long> arr, size_t index)
{
long long sum = std::accumulate(arr.begin(), arr.end(), decltype(arr)::value_type(0));
long long catfish = arr[index];
arr[index] = 0;
bool ateOther;
do
{
ateOther = false;
for (size_t i = 0; i < arr.size(); i++)
{
if (arr[i] != 0 && arr[i] < catfish)
{
catfish += arr[i];
arr[i] = 0;
ateOther = true;
}
}
} while (ateOther);
if (catfish == sum)
{
return 'T';
}
return 'N';
}
int main()
{
size_t n; // Number of catfish
std::vector<long long> arr;
std::cin >> n;
long long input;
// Load data
for (size_t i = 0; i < n; i++)
{
std::cin >> input;
arr.push_back(input);
}
for (size_t i = 0; i < n; i++)
{
std::cout << canBeKing(arr, i);
}
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 | #include <iostream> #include <vector> #include <string> #include <numeric> char canBeKing(std::vector<long long> arr, size_t index) { long long sum = std::accumulate(arr.begin(), arr.end(), decltype(arr)::value_type(0)); long long catfish = arr[index]; arr[index] = 0; bool ateOther; do { ateOther = false; for (size_t i = 0; i < arr.size(); i++) { if (arr[i] != 0 && arr[i] < catfish) { catfish += arr[i]; arr[i] = 0; ateOther = true; } } } while (ateOther); if (catfish == sum) { return 'T'; } return 'N'; } int main() { size_t n; // Number of catfish std::vector<long long> arr; std::cin >> n; long long input; // Load data for (size_t i = 0; i < n; i++) { std::cin >> input; arr.push_back(input); } for (size_t i = 0; i < n; i++) { std::cout << canBeKing(arr, i); } return 0; } |
English