제출 #1194784

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1194784		tutis	메시지 (IOI24_message)	C++20	큐에 대기중	0 ms	0 KiB

message

#include "message.h"
#include <algorithm>
#include <chrono>
#include <iostream>
#include <random>
#include <bitset>
#include <vector>
#include <math.h>

using namespace std;

int counter = 0;
bool initialized = false;

const int bitsCnt=8;
bitset<31> seqs[1<<bitsCnt];

void init()
{
    mt19937_64 rng(0);
    counter = 0;
    if (!initialized)
    {
        initialized = true;
        for (int i = 0; i < (1 < 16); i++)
        {
            seqs[i] = rng();
        }
    }
}

bitset<31> send(bitset<31> A)
{
    vector<bool> Avector(31);
    for (int i = 0; i < 31; i++)
    {
        Avector[i] = A[i];
    }
    vector<bool> recVector = send_packet(Avector);
    bitset<31> received;
    for (int j = 0; j < 31; j++)
    {
        received[j] = recVector[j];
    }
    counter++;
    return received;
}

struct X
{
    bitset<31> C;
};

X decode(X x, bitset<31> msg, bitset<31> bits)
{
    vector<int> options;
    for (int i = 0; i < (1<<bitsCnt); i++)
    {
        bitset<31> diff = msg ^ seqs[i];
        diff |= x.C;
        if (diff.count() > 15)
        {
            continue;
        }
        bitset<31> x;
        int bit = 0;
        for (int j = 0; j < 31; j++)
        {
            if (bits[j])
            {
                if ((i & (1 << bit)) != 0)
                {
                    x[j] = true;
                }
                bit++;
            }
        }
        if ((x & diff).count() != x.count())
        {
            continue;
        }
        options.push_back(i);
    }
    if (options.size() == 0)
    {
        exit(2);
    }
    bitset<31> diff;
    diff.flip();
    for (int i : options)
    {
        bitset<31> diff_i = msg ^ seqs[i];
        diff &= diff_i;
    }
    x.C |= diff;
    return x;
}

bitset<31> globC;

mt19937_64 rng(0);

bitset<31> get_bits(int step)
{
    mt19937_64 rng(step);
    bitset<31> ret;
    while (ret.count() < 8)
    {
        ret[rng() % 8] = true;
    }
    return ret;
}

void send_message(vector<bool> M, vector<bool> Cvect)
{
    bitset<31> C;
    for (int i = 0; i < 31; i++)
    {
        C[i] = Cvect[i];
    }
    globC = C;
    init();
    X x;
    int step = 0;
    while (x.C.count() < 15)
    {
        bitset<31> bits = get_bits(step);
        int val = 0;
        int cnt = 0;
        for (int i = 0; i < 31; i++)
        {
            if (bits[i])
            {
                if (C[i])
                {
                    val += 1 << cnt;
                }
                cnt++;
            }
        }
        cout << "c" << " " << x.C << endl;
        bitset<31> to_send = seqs[val % (1<<bitsCnt)];
        for (int i = 0; i < 31; i++)
        {
            if (C[i])
            {
                to_send[i] = rng() % 2;
            }
        }
        bitset<31> received = send(to_send);
        x = decode(x, received, bits);
        step++;
    }
    cout << "b" << endl;
    bitset<31> good = x.C.flip();
    int offset = (1024 - M.size()) % 16;
    vector<bool> X;
    for (int i = 0; i < offset; i++)
    {
        X.push_back(1);
    }
    X.push_back(0);
    for (int v : M)
    {
        X.push_back(v);
    }
    int i = 0;
    while (i < X.size())
    {
        bitset<31> packet;
        for (int v = 0; v < 31; v++)
        {
            if (good[v])
            {
                if (i < X.size())
                {
                    packet[v] = X[i];
                    i++;
                }
            }
        }
        send(packet);
    }
}

std::vector<bool> receive_message(vector<vector<bool>> Rvect)
{
    vector<bitset<31>> R(Rvect.size());
    for (int i = 0; i < Rvect.size(); i++)
    {
        bitset<31> b;
        for (int j = 0; j < 31; j++)
        {
            b[j] = Rvect[i][j];
        }
        R[i] = b;
    }
    init();
    X x;
    int i = 0;
    int step = 0;
    while (x.C.count() < 15)
    {
        x = decode(x, R[i], get_bits(step));
        i++;
        step++;
    }
    bitset<31> good = x.C.flip();
    vector<int> X;
    for (; i < R.size(); i++)
    {
        for (int v = 0; v < 31; v++)
        {
            if (good[v])
            {
                X.push_back(R[i][v]);
            }
        }
    }
    int offset = 0;
    while (X[offset])
    {
        offset++;
    }
    vector<bool> answer;
    for (int i = offset + 1; i < X.size(); i++)
    {
        answer.push_back(X[i]);
    }
    while ((1024 - answer.size()) % 16 != offset)
    {
        answer.pop_back();
    }
    return answer;
}

// vector<vector<bool>> sent;

// int main()
// {
//     vector<bool> msg(1024);

//     send_message(msg,
//                  {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
//                   1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1});
//     vector<bool> received = receive_message(sent);

//     cout << "Q = " << sent.size() << endl;
//     if (msg != received)
//     {
//         cout << "mismatch!" << endl;
//     }
//     else
//     {
//         cout << "ok" << endl;
//     }
// }

// vector<bool> send_packet(std::vector<bool> A)
// {
//     for (int i = 0; i < 31; i++)
//     {
//         if (globC[i])
//         {
//             A[i] = rng() % 2;
//         }
//     }
//     sent.push_back(A);
//     return A;
// }