#include <bits/stdc++.h>
using namespace std;
struct TestCase
{
size_t letter_count;
size_t ones;
size_t zeros;
};
TestCase read_test_case();
void solve_test_case(const TestCase&);
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
solve_test_case(read_test_case());
}
TestCase read_test_case()
{
TestCase tc;
cin >> tc.letter_count;
tc.ones = tc.zeros = 0;
string data;
cin >> data;
for (auto c : data)
if (c == '0')
tc.zeros++;
else
tc.ones++;
return tc;
}
void solve_test_case(const TestCase& tc)
{
const size_t min_popcount = 3;
const size_t max_popcount = 6;
const map<size_t, char> popcount_to_letter = {
{3, 'p'},
{4, 'x'},
{5, 'z'},
{6, 'w'},
};
if (tc.ones < min_popcount * tc.letter_count ||
tc.ones > max_popcount * tc.letter_count)
{
cout << "NIE\n";
return;
}
size_t remaining = tc.ones - 3 * tc.letter_count;
for (size_t i = 0; i < tc.letter_count; i++)
{
size_t add = min(remaining, static_cast<size_t>(3));
cout << popcount_to_letter.at(3 + add);
remaining -= add;
}
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 | #include <bits/stdc++.h> using namespace std; struct TestCase { size_t letter_count; size_t ones; size_t zeros; }; TestCase read_test_case(); void solve_test_case(const TestCase&); int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); solve_test_case(read_test_case()); } TestCase read_test_case() { TestCase tc; cin >> tc.letter_count; tc.ones = tc.zeros = 0; string data; cin >> data; for (auto c : data) if (c == '0') tc.zeros++; else tc.ones++; return tc; } void solve_test_case(const TestCase& tc) { const size_t min_popcount = 3; const size_t max_popcount = 6; const map<size_t, char> popcount_to_letter = { {3, 'p'}, {4, 'x'}, {5, 'z'}, {6, 'w'}, }; if (tc.ones < min_popcount * tc.letter_count || tc.ones > max_popcount * tc.letter_count) { cout << "NIE\n"; return; } size_t remaining = tc.ones - 3 * tc.letter_count; for (size_t i = 0; i < tc.letter_count; i++) { size_t add = min(remaining, static_cast<size_t>(3)); cout << popcount_to_letter.at(3 + add); remaining -= add; } cout << "\n"; } |
English