oj.uz

Submission #927966

# Submission time Handle Problem Language Result Execution time Memory
927966 2024-02-15T15:17:46 Z boris_mihov Two Transportations (JOI19_transportations) C++17
68 / 100
 445 ms 49052 KB 
#include "Azer.h"
#include <algorithm>
#include <iostream>
#include <numeric>
#include <cassert>
#include <random>
#include <vector>
#include <queue>

typedef long long llong;
const int DIST_BITS = 9;
const int NODE_BITS = 11;
const int MAXN = 2000 + 10;
const int MAXM = 500000 + 10;
const int INF  = 1e9;

static int n, m;
static int lastDist;
static int dist[MAXN];
static bool vis[MAXN];
static std::mt19937 rng(123456789);
static std::vector <bool> sequence;

struct PQelement
{
    int node;
    int currDist;

    friend bool operator < (const PQelement &a, const PQelement &b)
    {
        return a.currDist > b.currDist || (a.currDist == b.currDist && a.node > b.node);
    }
};

static std::vector <std::pair <int,int>> g[MAXN];
static std::priority_queue <PQelement> pq;
static std::queue <int> receivedQueue;
static int cntDoneTurns;
static int cntEntries;

int readA(int);
void writeA(int, int);
void manageA();

std::pair <int,int> readPairA()
{
    if (receivedQueue.front() == 1)
    {
        receivedQueue.pop();
        return {n, 1e9};
    }

    receivedQueue.pop();
    return {readA(NODE_BITS), readA(DIST_BITS)};
}

void sendMyTopA()
{
    cntEntries++;
    if (pq.size())
    {
        // std::cout << "send from A: " << pq.top().node << ' ' << pq.top().currDist << '\n';
        SendA(0);
        writeA(pq.top().node, NODE_BITS);
        writeA(pq.top().currDist - lastDist, DIST_BITS);
    } else
    {
        // std::cout << "send from A: " << n << ' ' << -1 << '\n';
        SendA(1);
    }
}

void receiveSendPairA()
{   
    auto [otherNode, otherDist] = readPairA();
    // std::cout << "receive from B: " << otherNode << ' ' << otherDist << '\n';
    otherDist += lastDist;

    if (otherNode < n) pq.push({otherNode, otherDist});
    while (pq.size() && vis[pq.top().node])
    {
        pq.pop();
    }

    if (pq.size())
    {
        // std::cout << "hereAAA: " << pq.top().currDist << ' ' << lastDist << '\n';
        // std::cout << "topA: " << pq.top().node << ' ' << pq.top().currDist << ' ' << lastDist << '\n';
        auto [node, currDist] = pq.top();
        pq.pop();

        vis[node] = true;
        dist[node] = currDist;
        lastDist = currDist;
        for (const auto &[u, edge] : g[node])
        {
            if (dist[u] > dist[node] + edge)
            {
                dist[u] = dist[node] + edge;
                pq.push({u, dist[u]});
            }
        }
    }

    while (pq.size() && vis[pq.top().node])
    {
        pq.pop();
    }

    cntDoneTurns++;
    if (cntDoneTurns < n && sequence[cntDoneTurns] == 1)
    {
        SendA(0);
    }
}

void receiveResendPairA()
{   
    cntEntries++;
    auto [otherNode, otherDist] = readPairA();
    otherDist += lastDist;
    if (otherNode < n) pq.push({otherNode, otherDist});
    // std::cout << "from A: " << otherNode << ' ' << otherDist << '\n';
    while (pq.size() && vis[pq.top().node])
    {
        pq.pop();
    }

    bool whatShouldSend = false;
    
    PQelement other = {otherNode, otherDist};
    // if (pq.size()) std::cout << "try b: " << pq.top().node << ' ' << pq.top().currDist << '\n';
    if (pq.size() && (otherNode >= n || other < pq.top()))
    {
        // std::cout << "send from B: " << pq.top().node << ' ' << pq.top().currDist << ' ' << lastDist << '\n';
        SendA(0);
        writeA(pq.top().node, NODE_BITS);
        writeA(pq.top().currDist - lastDist, DIST_BITS);
    } else if (pq.size())
    {
        SendA(1);
    }

    if (pq.size())
    {
        auto [node, currDist] = pq.top();
        // std::cout << "topB: " << node << ' ' << currDist << "\n";
        pq.pop();

        vis[node] = true;
        dist[node] = currDist;
        lastDist = currDist;
        for (const auto &[u, edge] : g[node])
        {
            // std::cout << "B edge: " << u << ' ' << node << ' ' << edge << ' ' << dist[u] << ' ' << dist[node] + edge << '\n';
            if (dist[u] > dist[node] + edge)
            {
                dist[u] = dist[node] + edge;
                // assert(!vispu)
                pq.push({u, dist[u]});
            }
        }
    }

    while (pq.size() && vis[pq.top().node])
    {
        pq.pop();
    }

    cntDoneTurns++;
    if (cntDoneTurns < n && sequence[cntDoneTurns] == 1)
    {
        SendA(0);
    }
}

void ReceiveA(bool x) 
{
    receivedQueue.push(x);
    if (cntEntries == cntDoneTurns && sequence[cntEntries] == 0)
    {
        receivedQueue.pop();
        manageA();
        return;
    }

    if (receivedQueue.size() == NODE_BITS + DIST_BITS + 1 || (receivedQueue.size() == 1 && receivedQueue.front() == 1))
    {
        if (sequence[cntDoneTurns] == 0) receiveSendPairA();
        else receiveResendPairA();
    }
}

int readA(int bits)
{
    int res = 0;
    for (int i = 0 ; i < bits ; ++i)
    {
        res <<= 1;
        // std::cout << "bitA: " << receivedQueue.front() << ' ' << res << '\n';
        res += receivedQueue.front();
        receivedQueue.pop();
    }

    return res;
}

void writeA(int number, int bits)
{
    for (int i = bits - 1 ; i >= 0 ; --i)
    {
        SendA((number & (1 << i)) > 0);
    }
}

void manageA()
{
    sendMyTopA();
}

void InitA(int N, int A, std::vector <int> U, std::vector <int> V,std::vector <int> C)
{
    n = N; m = A;
    for (int i = 0 ; i < A ; ++i)
    {
        g[U[i]].push_back({V[i], C[i]});
        g[V[i]].push_back({U[i], C[i]});
    }

    while (sequence.size() < n)
    {
        sequence.push_back(rng() & 1);
    }

    sequence[0] = 0;
    std::fill(dist, dist + n, INF);
    pq.push({0, 0});
    dist[0] = 0;
}

std::vector <int> Answer() 
{
    std::vector <int> answer(n);
    for (int i = 0 ; i < n ; ++i)
    {
        answer[i] = dist[i];
    }

    return answer;
}

#include "Baijan.h"
#include <algorithm>
#include <iostream>
#include <numeric>
#include <cassert>
#include <cassert>
#include <random>
#include <queue>

typedef long long llong;
const int DIST_BITS = 9;
const int NODE_BITS = 11;
const int MAXN = 2000 + 10;
const int MAXM = 500000 + 10;
const int INF  = 1e9;

static int n, m;
static int lastDist;
static int dist[MAXN];
static bool vis[MAXN];
static std::vector <std::pair <int,int>> g[MAXN];
static std::mt19937 rng(123456789);
static std::vector <bool> sequence;
static int cntDoneTurns;
static int cntEntries;

struct PQelement
{
    int node;
    int currDist;

    friend bool operator < (const PQelement &a, const PQelement &b)
    {
        return a.currDist > b.currDist || (a.currDist == b.currDist && a.node > b.node);
    }
};

static std::priority_queue <PQelement> pq;
static std::queue <int> receivedQueue;

int readB(int);
void writeB(int, int);
void receiveSendPairB();
void receiveResendPairB();
void manageB();

std::pair <int,int> readPairB()
{
    if (receivedQueue.front() == 1)
    {
        receivedQueue.pop();
        return {n, 1e9};
    }

    receivedQueue.pop();
    return {readB(NODE_BITS), readB(DIST_BITS)};
}

void ReceiveB(bool x) 
{
    receivedQueue.push(x);
    if (cntEntries == cntDoneTurns && sequence[cntEntries])
    {
        receivedQueue.pop();
        manageB();
        return;
    }

    if (receivedQueue.size() == NODE_BITS + DIST_BITS + 1 || (receivedQueue.size() == 1 && receivedQueue.front() == 1))
    {
        if (sequence[cntDoneTurns] == 1) receiveSendPairB();
        else receiveResendPairB();
    }
}

void sendMyTopB()
{
    cntEntries++;
    if (pq.size())
    {
        // std::cout << "send from A: " << pq.top().node << ' ' << pq.top().currDist << '\n';
        SendB(0);
        writeB(pq.top().node, NODE_BITS);
        writeB(pq.top().currDist - lastDist, DIST_BITS);
    } else 
    {
        // std::cout << "send from A: " << n << ' ' << -1 << '\n';
        SendB(1);
    }
}

void receiveSendPairB()
{   
    auto [otherNode, otherDist] = readPairB();
    // std::cout << "receive from B: " << otherNode << ' ' << otherDist << '\n';
    otherDist += lastDist;

    if (otherNode < n) pq.push({otherNode, otherDist});
    while (pq.size() && vis[pq.top().node])
    {
        pq.pop();
    }

    if (pq.size())
    {
        // std::cout << "hereAAA: " << pq.top().currDist << ' ' << lastDist << '\n';
        // std::cout << "topA: " << pq.top().node << ' ' << pq.top().currDist << ' ' << lastDist << '\n';
        auto [node, currDist] = pq.top();
        pq.pop();

        vis[node] = true;
        dist[node] = currDist;
        lastDist = currDist;
        for (const auto &[u, edge] : g[node])
        {
            if (dist[u] > dist[node] + edge)
            {
                dist[u] = dist[node] + edge;
                pq.push({u, dist[u]});
            }
        }
    }

    while (pq.size() && vis[pq.top().node])
    {
        pq.pop();
    }

    cntDoneTurns++;
    if (cntDoneTurns < n && sequence[cntDoneTurns] == 0)
    {
        SendB(0);
    }
}

void receiveResendPairB()
{   
    cntEntries++;
    auto [otherNode, otherDist] = readPairB();
    otherDist += lastDist;
    if (otherNode < n) pq.push({otherNode, otherDist});
    // std::cout << "from A: " << otherNode << ' ' << otherDist << '\n';
    while (pq.size() && vis[pq.top().node])
    {
        pq.pop();
    }

    bool whatShouldSend = false;
    
    PQelement other = {otherNode, otherDist};
    // if (pq.size()) std::cout << "try b: " << pq.top().node << ' ' << pq.top().currDist << '\n';
    if (pq.size() && (otherNode >= n || other < pq.top()))
    {
        // std::cout << "send from B: " << pq.top().node << ' ' << pq.top().currDist << ' ' << lastDist << '\n';
        SendB(0);
        writeB(pq.top().node, NODE_BITS);
        writeB(pq.top().currDist - lastDist, DIST_BITS);
    } else if (pq.size())
    {
        SendB(1);
    }

    if (pq.size())
    {
        auto [node, currDist] = pq.top();
        // std::cout << "topB: " << node << ' ' << currDist << "\n";
        pq.pop();

        vis[node] = true;
        dist[node] = currDist;
        lastDist = currDist;
        for (const auto &[u, edge] : g[node])
        {
            // std::cout << "B edge: " << u << ' ' << node << ' ' << edge << ' ' << dist[u] << ' ' << dist[node] + edge << '\n';
            if (dist[u] > dist[node] + edge)
            {
                dist[u] = dist[node] + edge;
                // assert(!vispu)
                pq.push({u, dist[u]});
            }
        }
    }

    while (pq.size() && vis[pq.top().node])
    {
        pq.pop();
    }

    cntDoneTurns++;
    if (cntDoneTurns < n && sequence[cntDoneTurns] == 0)
    {
        SendB(0);
    }
}

void manageB()
{
    sendMyTopB();
}

int readB(int bits)
{
    int res = 0;
    for (int i = 0 ; i < bits ; ++i)
    {
        res <<= 1;
        res += receivedQueue.front();
        receivedQueue.pop();
    }

    return res;
}

void writeB(int number, int bits)
{
    for (int i = bits - 1 ; i >= 0 ; --i)
    {
        // std::cout << "bitB: " << number << ' ' << (number & (1 << i)) << ' ' << ((number & (1 << i)) > 0) << '\n';
        SendB((number & (1 << i)) > 0);
    }
}

void InitB(int N, int B, std::vector <int> U, std::vector <int> V, std::vector <int> C)
{
    n = N; m = B;
    for (int i = 0 ; i < B ; ++i)
    {
        g[U[i]].push_back({V[i], C[i]});
        g[V[i]].push_back({U[i], C[i]});
    }

    while (sequence.size() < n)
    {
        sequence.push_back(rng() & 1);
    }

    sequence[0] = 0;
    std::fill(dist, dist + n, INF);
    pq.push({0, 0});
    dist[0] = 0;
    SendB(0);
}

Compilation message

Azer.cpp: In function 'void receiveResendPairA()':
Azer.cpp:129:10: warning: unused variable 'whatShouldSend' [-Wunused-variable]
  129 |     bool whatShouldSend = false;
      |          ^~~~~~~~~~~~~~
Azer.cpp: In function 'void InitA(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
Azer.cpp:230:28: warning: comparison of integer expressions of different signedness: 'std::vector<bool>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  230 |     while (sequence.size() < n)
      |            ~~~~~~~~~~~~~~~~^~~

Baijan.cpp: In function 'void receiveResendPairB()':
Baijan.cpp:148:10: warning: unused variable 'whatShouldSend' [-Wunused-variable]
  148 |     bool whatShouldSend = false;
      |          ^~~~~~~~~~~~~~
Baijan.cpp: In function 'void InitB(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
Baijan.cpp:232:28: warning: comparison of integer expressions of different signedness: 'std::vector<bool>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
  232 |     while (sequence.size() < n)
      |            ~~~~~~~~~~~~~~~~^~~

Subtask #1
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 235 ms 1052 KB Output is correct
2 Correct 0 ms 664 KB Output is correct
3 Correct 223 ms 1056 KB Output is correct
4 Correct 265 ms 10516 KB Output is correct
5 Correct 10 ms 920 KB Output is correct
6 Correct 231 ms 2468 KB Output is correct

Subtask #2
0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 744 KB Output is correct
2 Correct 217 ms 892 KB Output is correct
3 Correct 278 ms 944 KB Output is correct
4 Runtime error 215 ms 27256 KB Execution killed with signal 13
5 Halted 0 ms 0 KB -

Subtask #3
0 / 8.0

# Verdict Execution time Memory Grader output
1 Runtime error 131 ms 540 KB Execution killed with signal 13
2 Halted 0 ms 0 KB -

Subtask #4
38.0 / 38.0

# Verdict Execution time Memory Grader output
1 Correct 111 ms 836 KB Output is correct
2 Correct 131 ms 816 KB Output is correct
3 Correct 183 ms 13580 KB Output is correct
4 Correct 159 ms 664 KB Output is correct
5 Correct 160 ms 10128 KB Output is correct
6 Correct 124 ms 920 KB Output is correct
7 Correct 127 ms 664 KB Output is correct
8 Correct 113 ms 868 KB Output is correct
9 Correct 185 ms 18484 KB Output is correct
10 Correct 174 ms 18624 KB Output is correct
11 Correct 283 ms 35696 KB Output is correct
12 Correct 266 ms 31008 KB Output is correct
13 Correct 125 ms 664 KB Output is correct
14 Correct 0 ms 664 KB Output is correct

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 111 ms 836 KB Output is correct
2 Correct 131 ms 816 KB Output is correct
3 Correct 183 ms 13580 KB Output is correct
4 Correct 159 ms 664 KB Output is correct
5 Correct 160 ms 10128 KB Output is correct
6 Correct 124 ms 920 KB Output is correct
7 Correct 127 ms 664 KB Output is correct
8 Correct 113 ms 868 KB Output is correct
9 Correct 185 ms 18484 KB Output is correct
10 Correct 174 ms 18624 KB Output is correct
11 Correct 283 ms 35696 KB Output is correct
12 Correct 266 ms 31008 KB Output is correct
13 Correct 125 ms 664 KB Output is correct
14 Correct 0 ms 664 KB Output is correct
15 Correct 173 ms 840 KB Output is correct
16 Correct 169 ms 816 KB Output is correct
17 Correct 110 ms 664 KB Output is correct
18 Correct 208 ms 10580 KB Output is correct
19 Correct 162 ms 664 KB Output is correct
20 Correct 181 ms 10696 KB Output is correct
21 Correct 176 ms 664 KB Output is correct
22 Correct 162 ms 664 KB Output is correct
23 Correct 244 ms 22380 KB Output is correct
24 Correct 254 ms 22448 KB Output is correct
25 Correct 378 ms 43684 KB Output is correct
26 Correct 311 ms 36948 KB Output is correct
27 Correct 162 ms 1176 KB Output is correct

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 111 ms 836 KB Output is correct
2 Correct 131 ms 816 KB Output is correct
3 Correct 183 ms 13580 KB Output is correct
4 Correct 159 ms 664 KB Output is correct
5 Correct 160 ms 10128 KB Output is correct
6 Correct 124 ms 920 KB Output is correct
7 Correct 127 ms 664 KB Output is correct
8 Correct 113 ms 868 KB Output is correct
9 Correct 185 ms 18484 KB Output is correct
10 Correct 174 ms 18624 KB Output is correct
11 Correct 283 ms 35696 KB Output is correct
12 Correct 266 ms 31008 KB Output is correct
13 Correct 125 ms 664 KB Output is correct
14 Correct 0 ms 664 KB Output is correct
15 Correct 173 ms 840 KB Output is correct
16 Correct 169 ms 816 KB Output is correct
17 Correct 110 ms 664 KB Output is correct
18 Correct 208 ms 10580 KB Output is correct
19 Correct 162 ms 664 KB Output is correct
20 Correct 181 ms 10696 KB Output is correct
21 Correct 176 ms 664 KB Output is correct
22 Correct 162 ms 664 KB Output is correct
23 Correct 244 ms 22380 KB Output is correct
24 Correct 254 ms 22448 KB Output is correct
25 Correct 378 ms 43684 KB Output is correct
26 Correct 311 ms 36948 KB Output is correct
27 Correct 162 ms 1176 KB Output is correct
28 Correct 224 ms 852 KB Output is correct
29 Correct 203 ms 804 KB Output is correct
30 Correct 297 ms 24300 KB Output is correct
31 Correct 147 ms 808 KB Output is correct
32 Correct 256 ms 21268 KB Output is correct
33 Correct 219 ms 876 KB Output is correct
34 Correct 220 ms 1176 KB Output is correct
35 Correct 196 ms 1176 KB Output is correct
36 Correct 274 ms 24984 KB Output is correct
37 Correct 278 ms 25108 KB Output is correct
38 Correct 445 ms 49052 KB Output is correct
39 Correct 404 ms 44124 KB Output is correct
40 Correct 202 ms 1176 KB Output is correct

Subtask #7
0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 235 ms 1052 KB Output is correct
2 Correct 0 ms 664 KB Output is correct
3 Correct 223 ms 1056 KB Output is correct
4 Correct 265 ms 10516 KB Output is correct
5 Correct 10 ms 920 KB Output is correct
6 Correct 231 ms 2468 KB Output is correct
7 Correct 0 ms 744 KB Output is correct
8 Correct 217 ms 892 KB Output is correct
9 Correct 278 ms 944 KB Output is correct
10 Runtime error 215 ms 27256 KB Execution killed with signal 13
11 Halted 0 ms 0 KB -