oj.uz

Submission #916272

# Submission time Handle Problem Language Result Execution time Memory
916272 2024-01-25T14:57:57 Z boris_mihov Minerals (JOI19_minerals) C++17
40 / 100
 22 ms 2132 KB 
#include "minerals.h"
#include <algorithm>
#include <iostream>
#include <numeric>
#include <cassert>
#include <random>
#include <vector>
#include <queue>

typedef long long llong;
const int MAXN = 86000 + 10;
const int INF  = 1e9;

int n;
struct Node
{
    int idx;
    int level;
    int l, r;
    int qL, qR;
};

int perm[MAXN];
bool isIn[MAXN];
std::vector <int> unique;
std::vector <int> second;
std::vector <int> secondCopy;
std::deque <Node> waitlist;
std::mt19937 rng(69420);

int countDiff;
int query(int idx)
{
    isIn[idx] ^= 1;
    return countDiff = Query(perm[idx]);
}

void answer(int a, int b)
{
    Answer(perm[a], perm[b]);
}

void Solve(int N) 
{
    n = N;
    std::iota(perm + 1, perm + 1 + 2 * n, 1);
    std::shuffle(perm + 1, perm + 1 + 2 * n, rng);

    unique.push_back(0);
    unique.push_back(1);
    second.push_back(0);
    query(1);
    int last = 1;

    for (int i = 2 ; i <= 2 * n ; ++i)
    {
        if (query(i) > last)
        {
            last++;
            unique.push_back(i);
        } else
        {
            second.push_back(i);
        }
    }

    waitlist.push_front({1, 1, 1, n, 1, n});
    assert(unique.size() == n + 1);
    assert(second.size() == n + 1);
    secondCopy.resize(n + 1);

    int ptr = n;
    int lastLevel = 1;
    int reversed = 1;

    while (waitlist.size())
    {
        auto [idx, level, l, r, qL, qR] = waitlist.front();
        if (level != reversed)
        {
            reversed = level;
            std::reverse(waitlist.begin(), waitlist.end());
            continue;
        }

        waitlist.pop_front();

        assert(r - l + 1 == qR - qL + 1);
        if (l == r)
        {
            assert(qL == qR);
            answer(unique[l], second[qL]);
            continue;
        }

        int mid = (l + r) / 2;
        if (level != lastLevel)
        {
            if (level & 1)
            {
                while (ptr < mid)
                {
                    ptr++;
                    query(unique[ptr]);
                }
            } else
            {
                while (ptr > mid)
                {
                    query(unique[ptr]);
                    ptr--;
                }
            }

            lastLevel = level;
        }

        if (level & 1)
        {
            while (ptr > mid)
            {
                query(unique[ptr]);
                ptr--;
            }
        } else
        {
            while (ptr < mid)
            {
                ptr++;
                query(unique[ptr]);
            }
        }

        int myL = qL;
        int myR = qR;
        for (int idx = qL ; idx <= qR ; ++idx)
        {
            int last = countDiff;
            int curr = query(second[idx]);
            if (curr == last)
            {
                secondCopy[myL++] = second[idx];
            } else
            {
                secondCopy[myR--] = second[idx];
            }
        }

        for (int idx = qL ; idx <= qR ; ++idx)
        {
            second[idx] = secondCopy[idx];
        }
        
        assert(qL < myL);
        assert(myR < qR);
        if (level & 1)
        {
            waitlist.push_back({2 * idx + 1, level + 1, mid + 1, r, myR + 1, qR});
            waitlist.push_back({2 * idx, level + 1, l, mid, qL, myL - 1});
        } else
        {
            waitlist.push_back({2 * idx, level + 1, l, mid, qL, myL - 1});
            waitlist.push_back({2 * idx + 1, level + 1, mid + 1, r, myR + 1, qR});
        }
    }
}

Compilation message

In file included from /usr/include/c++/10/cassert:44,
                 from minerals.cpp:5:
minerals.cpp: In function 'void Solve(int)':
minerals.cpp:68:26: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   68 |     assert(unique.size() == n + 1);
      |            ~~~~~~~~~~~~~~^~~~~~~~
minerals.cpp:69:26: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   69 |     assert(second.size() == n + 1);
      |            ~~~~~~~~~~~~~~^~~~~~~~

Subtask #1
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct

Subtask #2
25.0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 1 ms 600 KB Output is correct
3 Correct 2 ms 600 KB Output is correct
4 Correct 6 ms 856 KB Output is correct
5 Correct 9 ms 1368 KB Output is correct

Subtask #3
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
6 Correct 1 ms 600 KB Output is correct
7 Correct 2 ms 600 KB Output is correct
8 Correct 6 ms 856 KB Output is correct
9 Correct 9 ms 1368 KB Output is correct
10 Correct 1 ms 344 KB Output is correct
11 Correct 8 ms 1112 KB Output is correct
12 Correct 9 ms 1584 KB Output is correct
13 Correct 8 ms 1368 KB Output is correct
14 Correct 9 ms 1528 KB Output is correct

Subtask #4
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
6 Correct 1 ms 600 KB Output is correct
7 Correct 2 ms 600 KB Output is correct
8 Correct 6 ms 856 KB Output is correct
9 Correct 9 ms 1368 KB Output is correct
10 Correct 1 ms 344 KB Output is correct
11 Correct 8 ms 1112 KB Output is correct
12 Correct 9 ms 1584 KB Output is correct
13 Correct 8 ms 1368 KB Output is correct
14 Correct 9 ms 1528 KB Output is correct
15 Incorrect 22 ms 2132 KB Wrong Answer [2]
16 Halted 0 ms 0 KB -

Subtask #5
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
6 Correct 1 ms 600 KB Output is correct
7 Correct 2 ms 600 KB Output is correct
8 Correct 6 ms 856 KB Output is correct
9 Correct 9 ms 1368 KB Output is correct
10 Correct 1 ms 344 KB Output is correct
11 Correct 8 ms 1112 KB Output is correct
12 Correct 9 ms 1584 KB Output is correct
13 Correct 8 ms 1368 KB Output is correct
14 Correct 9 ms 1528 KB Output is correct
15 Incorrect 22 ms 2132 KB Wrong Answer [2]
16 Halted 0 ms 0 KB -

Subtask #6
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
6 Correct 1 ms 600 KB Output is correct
7 Correct 2 ms 600 KB Output is correct
8 Correct 6 ms 856 KB Output is correct
9 Correct 9 ms 1368 KB Output is correct
10 Correct 1 ms 344 KB Output is correct
11 Correct 8 ms 1112 KB Output is correct
12 Correct 9 ms 1584 KB Output is correct
13 Correct 8 ms 1368 KB Output is correct
14 Correct 9 ms 1528 KB Output is correct
15 Incorrect 22 ms 2132 KB Wrong Answer [2]
16 Halted 0 ms 0 KB -

Subtask #7
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
6 Correct 1 ms 600 KB Output is correct
7 Correct 2 ms 600 KB Output is correct
8 Correct 6 ms 856 KB Output is correct
9 Correct 9 ms 1368 KB Output is correct
10 Correct 1 ms 344 KB Output is correct
11 Correct 8 ms 1112 KB Output is correct
12 Correct 9 ms 1584 KB Output is correct
13 Correct 8 ms 1368 KB Output is correct
14 Correct 9 ms 1528 KB Output is correct
15 Incorrect 22 ms 2132 KB Wrong Answer [2]
16 Halted 0 ms 0 KB -

Subtask #8
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
6 Correct 1 ms 600 KB Output is correct
7 Correct 2 ms 600 KB Output is correct
8 Correct 6 ms 856 KB Output is correct
9 Correct 9 ms 1368 KB Output is correct
10 Correct 1 ms 344 KB Output is correct
11 Correct 8 ms 1112 KB Output is correct
12 Correct 9 ms 1584 KB Output is correct
13 Correct 8 ms 1368 KB Output is correct
14 Correct 9 ms 1528 KB Output is correct
15 Incorrect 22 ms 2132 KB Wrong Answer [2]
16 Halted 0 ms 0 KB -

Subtask #9
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 344 KB Output is correct
5 Correct 1 ms 344 KB Output is correct
6 Correct 1 ms 600 KB Output is correct
7 Correct 2 ms 600 KB Output is correct
8 Correct 6 ms 856 KB Output is correct
9 Correct 9 ms 1368 KB Output is correct
10 Correct 1 ms 344 KB Output is correct
11 Correct 8 ms 1112 KB Output is correct
12 Correct 9 ms 1584 KB Output is correct
13 Correct 8 ms 1368 KB Output is correct
14 Correct 9 ms 1528 KB Output is correct
15 Incorrect 22 ms 2132 KB Wrong Answer [2]
16 Halted 0 ms 0 KB -