#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; string word; word.reserve(8 * n); cin >> word; int ones = count(all(word), '1'); int min_ones = n * 3; int max_ones = n * 6; if (ones < min_ones || max_ones < ones) { cout << "NIE\n"; return; } char map_c[] = { 0, 0, 0, 'a', // 3 'c', // 4 'g', // 5 'o' // 6 }; int k = ones / n; int t = ones - k * n; string res; res.reserve(n); loop(i, t) { res += map_c[k + 1]; } loop(i, t, n) { res += map_c[k]; } cout << res << '\n'; } int main(void) { ios_base::sync_with_stdio(false); cin.tie(NULL); int qs = 1; 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 | #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; string word; word.reserve(8 * n); cin >> word; int ones = count(all(word), '1'); int min_ones = n * 3; int max_ones = n * 6; if (ones < min_ones || max_ones < ones) { cout << "NIE\n"; return; } char map_c[] = { 0, 0, 0, 'a', // 3 'c', // 4 'g', // 5 'o' // 6 }; int k = ones / n; int t = ones - k * n; string res; res.reserve(n); loop(i, t) { res += map_c[k + 1]; } loop(i, t, n) { res += map_c[k]; } cout << res << '\n'; } int main(void) { ios_base::sync_with_stdio(false); cin.tie(NULL); int qs = 1; while (qs--) { solution(); } return 0; } |