# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1008756 | biank | Parrots (IOI11_parrots) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "encoder.h"
#include "encoderlib.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int D = 512;
struct BigInt {
bitset<D> num;
BigInt(int x = 0) : num(x) {}
bool operator[](int i) const { return num[i]; }
friend BigInt operator+(const BigInt &x, const BigInt &y) {
bool carry = 0;
BigInt z;
for (int i = 0; i < D; i++) {
char s = x[i] + y[i] + carry;
if (s >= 2) s -= 2, carry = 1;
else carry = 0:
z.num[i] = s;
}
return z;
}
friend BigInt operator-(const BigInt &x, const BigInt &y) {//x >= y
bool carry = 0;
BigInt z;
for (int i = 0; i < D; i++) {
char s = x[i] - y[i] - carry;
if (s < 0) s += 2, carry = 1;
else carry = 0;
z.num[i] = s;
}
return z;
}
friend bool operator<(const BigInt &x, const BigInt &y) {
for (int i = D - 1; i >= 0; i--) {
if (x[i] ^ y[i]) return y[i];
}
return false;
}
friend bool operator>=(const BigInt &x, const BigInt &y) {
return !(x < y);
}
BigInt &operator+=(const BigInt &o) {
return *this = *this + o;
}
BigInt &operator-=(const BigInt &o) {
return *this = *this - o;
}
BigInt &operator<<=(int i) {
num <<= i;
return *this;
}
BigInt operator<<(int i) { return *this <<= i; }
};
const int MAXN = 320;
const int MAXA = 255;
BigInt dp[MAXN][MAXA];
bool flag = false;
void calcDP() {
for (int i = 0; i < MAXA; i++) {
dp[0][i] = 1;
}
for (int i = 1; i < MAXN; i++) {
dp[i][0] = dp[i - 1][0];
for (int j = 1; j < MAXA; j++) {
dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
}
}
flag = true;
}
void encode(int N, int M[]) {
if (!flag) calcDP();
BigInt pos = 0;
for (int i = 0; i < N; i++) {
pos = (pos << 8) + M[i];
}
pos += 1;
for (int i = 5 * N - 1; i >= 0; i--) {
int j = 0;
while (pos >= dp[i][j]) {
pos -= dp[i][j];
j++;
}
send(j);
}
}
#include "decoder.h"
#include "decoderlib.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int B = 512;
struct BigInt_de {
bitset<B> num;
BigInt_de(int x = 0) {
for (int i = 0; i < 30; i++) {
num[i] = x >> i & 1;
}
}
bool operator[](int i) const { return num[i]; }
friend BigInt_de operator+(const BigInt_de &x, const BigInt_de &y) {
bool carry = 0;
BigInt_de z;
for (int i = 0; i < B; i++) {
char s = x[i] + y[i] + carry;
if (s >= 2) s -= 2, carry = 1;
else carry = 0;
z.num[i] = s;
}
return z;
}
friend BigInt_de operator-(const BigInt_de &x, const BigInt_de &y) {//x >= y
bool carry = 0;
BigInt_de z;
for (int i = 0; i < B; i++) {
int s = x[i] - y[i] - carry;
if (s < 0) s += 2, carry = 1;
else carry = 0;
z.num[i] = s;
}
return z;
}
friend bool operator<(const BigInt_de &x, const BigInt_de &y) {
for (int i = B - 1; i >= 0; i--) {
if (x[i] ^ y[i]) return y[i];
}
return false;
}
inline friend bool operator>=(const BigInt_de &x, const BigInt_de &y) {
return !(x < y);
}
ll val() {
return num.to_ulong();
}
BigInt_de &operator+=(const BigInt_de &o) {
return *this = *this + o;
}
BigInt_de &operator-=(const BigInt_de &o) {
return *this = *this - o;
}
inline BigInt_de operator&(const BigInt_de &o) {
BigInt_de r;
r.num = num & o.num;
return r;
}
inline BigInt_de &operator>>=(int i) {
num >>= i;
return *this;
}
};
const int MAXN_DE = 320;
const int MAXA_DE = 255;
BigInt_de dp_de[MAXN_DE][MAXA_DE];
bool flag_de = false;
void calcDP_de() {
for (int i = 0; i < MAXA_DE; i++) {
dp_de[0][i] = 1;
}
for (int i = 1; i < MAXN_DE; i++) {
dp_de[i][0] = dp_de[i - 1][0];
for (int j = 1; j < MAXA_DE; j++) {
dp_de[i][j] = dp_de[i - 1][j] + dp_de[i][j - 1];
}
}
flag_de = true;
}
void decode(int N, int L, int X[]) {
if (!flag_de) calcDP_de();
sort(X, X + L);
BigInt_de pos = 0;
for (int i = L - 1; i >= 0; i--) {
for (int j = 0; j < X[i]; j++) pos += dp_de[i][j];
}
pos -= 1;
vector<int> res;
for (int i = 0; i < N; i++) {
BigInt_de x = pos & 255;
res.push_back(x.val());
pos >>= 8;
}
for (int i = N - 1; i >= 0; i--) {
output(res[i]);
}
}