#include <iostream>
#include <vector>
#include "message.h"
using namespace std;

void send_message(std::vector<bool> M, std::vector<bool> C) {
	int S = M.size();
	std::vector<int> pos;
	for (int i = 0; i < 30; i ++) {
		send_packet(vector<bool>(31, C[i]));
		if (!C[i]) pos.push_back(i);
	}
	if (!C[30]) pos.push_back(30);

	std::vector<bool> len(31);
	for (int i = 0; i < 16; i ++)
		len[pos[i]] = S >> i & 1;

	send_packet(len);
	for (int i = 0; i < S; i += 16) {
		std::vector<bool> tmp(31);
		for (int j = i; j < min(i + 16, S); j ++)
			tmp[pos[j - i]] = M[j];
		send_packet(tmp);
	}
}
std::vector<bool> receive_message(std::vector<vector<bool>> R) {
	std::vector<int> pos;
	int sum = 0;
	for (int i = 0; i < 30; i ++) {
		int C[2] = {0};
		for (int j = 0; j < 31; j ++)
			C[R[i][j]] ++;
		if (C[0] > C[1]) pos.push_back(i);
		sum += C[1] > C[0];
	}
	if (sum == 15) pos.push_back(30);

	int S = 0;
	for (int i = 0; i < 16; i ++)
		S |= (R[30][pos[i]] << i);

	std::vector<bool> res;
	for (int i = 31; i < R.size(); i ++)
		for (auto j : pos)
			res.push_back(R[i][j]);
	while (res.size() > S) res.pop_back();

	return res;
}