#include <iostream> #include <string> using namespace std; bool decode(const string &input, string &output, int len) { int zeros = 0, ones = 0, pos = 0; for(int i=0; i<8*len; ++i) { if(input[i] == '0') ++zeros; else ++ones; } if(zeros > ones) { int numCh = (zeros-ones) >> 1; output.append(numCh, 'a'); zeros -= 5*numCh; ones -= 3*numCh; if(zeros < 0 || ones < 0 || zeros%4 != 0 || ones%4 != 0) return false; output.append((zeros >> 2), 'c'); return true; } if(ones > zeros) { int diff = ones - zeros; for(int quads=0; quads<= (diff>>2); ++quads) { int dbls = (diff-(quads<<2))>>1; int remZeros = zeros - quads*2 - dbls*3; int remOnes = ones - quads*6 - dbls*5; if(remZeros != remOnes || remZeros < 0 || remOnes < 0 || remZeros % 4 || remOnes % 4) continue; int equals = remZeros / 4; if(quads + dbls + equals == len) { output.append(quads, 'o'); output.append(dbls, 'g'); output.append(equals, 'c'); return true; } } return false; } output.append((zeros >> 2), 'c'); return true; } int main() { string input, output; int n; ios::sync_with_stdio(false); cin >> n >> input; if(decode(input, output, n)) cout << output << endl; else cout << "NIE" << endl; 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 | #include <iostream> #include <string> using namespace std; bool decode(const string &input, string &output, int len) { int zeros = 0, ones = 0, pos = 0; for(int i=0; i<8*len; ++i) { if(input[i] == '0') ++zeros; else ++ones; } if(zeros > ones) { int numCh = (zeros-ones) >> 1; output.append(numCh, 'a'); zeros -= 5*numCh; ones -= 3*numCh; if(zeros < 0 || ones < 0 || zeros%4 != 0 || ones%4 != 0) return false; output.append((zeros >> 2), 'c'); return true; } if(ones > zeros) { int diff = ones - zeros; for(int quads=0; quads<= (diff>>2); ++quads) { int dbls = (diff-(quads<<2))>>1; int remZeros = zeros - quads*2 - dbls*3; int remOnes = ones - quads*6 - dbls*5; if(remZeros != remOnes || remZeros < 0 || remOnes < 0 || remZeros % 4 || remOnes % 4) continue; int equals = remZeros / 4; if(quads + dbls + equals == len) { output.append(quads, 'o'); output.append(dbls, 'g'); output.append(equals, 'c'); return true; } } return false; } output.append((zeros >> 2), 'c'); return true; } int main() { string input, output; int n; ios::sync_with_stdio(false); cin >> n >> input; if(decode(input, output, n)) cout << output << endl; else cout << "NIE" << endl; return 0; } |