#include <bits/stdc++.h>
#include "message.h"

using i64 = long long;
using u64 = unsigned long long;

void send_message(std::vector<bool> M, std::vector<bool> C) {
    M.push_back(true);
    M.resize(1025, false);
    
    std::vector<std::vector<bool>> S(66, std::vector<bool>(31));
    std::vector<int> a;
    for (int i = 0; i < 31; i++) {
        if (C[i] == 0) {
            a.push_back(i);
        }
    }
    
    int cnt = 0;
    for (int i = 0; i < 16; i++) {
        int d = i == 15 ? a[0] + 31 - a[i] : a[i + 1] - a[i];
        for (int j = 0; j < 66; j++) {
            if (j < d - 1) {
                S[j][a[i]] = 0;
            } else if (j == d - 1) {
                S[j][a[i]] = 1;
            } else {
                S[j][a[i]] = M[cnt++];
            }
        }
    }
    assert(cnt == 1025);
    for (int i = 0; i < 66; i++) {
        send_packet(S[i]);
    }
}

std::vector<bool> receive_message(std::vector<std::vector<bool>> R) {
    std::vector<int> f(31), a;
    for (int i = 0; i < 31; i++) {
        int j = 0;
        while (j < 66 && !R[j][i]) {
            j++;
        }
        f[i] = (i + j + 1) % 31;
    }
    
    for (int i = 0; i < 31; i++) {
        int x = i;
        int len = 0;
        do {
            x = f[x];
            len++;
        } while (x != i && len < 16);
        if (x == i && len == 16) {
            a.push_back(i);
        }
    }
    
    std::vector<bool> M;
    for (auto i : a) {
        int j = 0;
        while (!R[j][i]) {
            j++;
        }
        for (j++; j < 66; j++) {
            M.push_back(R[j][i]);
        }
    }
    while (!M.back()) {
        M.pop_back();
    }
    M.pop_back();
    return M;
}