#include "message.h"
#include <bits/stdc++.h>
using namespace std;
void send_message(vector<bool> M, vector<bool> C) {
    vector<bool> msg[67];
    int arr[16], cnt = 0;
    for(int i = 0; i < 31; i++) {
        if(!C[i]) arr[cnt++] = i;
    }
    for(int i = 0; i < 67; i++) msg[i] = vector<bool>(31, 0);
    int grr = arr[0];
    for(int i = 0; i < 4; i++) {
        for(int j = 0; j < 31; j++) msg[0][j] = grr & 1;
        send_packet(msg[0]);
        grr /= 2;
    }
    msg[0] = vector<bool>(31, 0);
    for(int i = 0; i < 15; i++) {
        grr = arr[i + 1] - arr[i];
        msg[grr - 1][arr[i]] = 1;
    }
    cnt = M.size();
    vector<bool> sz(10);
    for(int i = 0; i < 10; i++) {
        sz[i] = cnt & 1;
        cnt /= 2;
    }
    int c0 = 0, c1 = 0;
    for(int i = 0; i < 16; i++) {
        grr = (i == 15);
        for(int j = 0; j < 67; j++) {
            if(grr) {
                if(c0 == 10) msg[j][arr[i]] = (c1 < (int)M.size()) ? M[c1++] : 0;
                else msg[j][arr[i]] = sz[c0++];
            }
            if(msg[j][arr[i]]) grr = 1;
        }
    }
    for(int i = 0; i < 67; i++) send_packet(msg[i]);
}

vector<bool> receive_message(vector<vector<bool>> R) {
    // for(auto i : R) {
    //     for(auto j : i) cout << j;
    //     cout << "\n";
    // }
    int arr[16], cnt, grr = 0;
    for(int i = 0; i < 4; i++) {
        cnt = 0;
        for(int j = 0; j < 31; j++) cnt += R[i][j];
        if(cnt > 15) grr += (1 << i);
    }
    arr[0] = grr;
    vector<bool> ans;
    int sz = 0;
    cnt = 0;
    for(int i = 0; i < 16; i++) {
        grr = (i == 15);
        for(int j = 4; j < 71; j++) {
            if(grr) {
                if(cnt < 9) sz += (R[j][arr[i]] << cnt), cnt++;
                else if(cnt == 9) sz += (R[j][arr[i]] << cnt), sz = (sz ? sz : 1024), cnt++;
                else if(sz > 0) ans.push_back(R[j][arr[i]]), sz--;
            }
            if(!grr && R[j][arr[i]]) arr[i + 1] = arr[i] + j - 3, grr = 1;
        }
    }
    return ans;
}