oj.uz

Submission #922875

# Submission time Handle Problem Language Result Execution time Memory
922875 2024-02-06T08:35:36 Z boris_mihov Minerals (JOI19_minerals) C++17
80 / 100
 38 ms 4340 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;
int with[MAXN];
int perm[MAXN];
bool isIn[MAXN];
int taken[MAXN];
int taken0[MAXN];
int taken1[MAXN];
std::vector <int> unique;
std::vector <int> second;
std::mt19937 rng(76543);
int to[MAXN];

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);

    int cnt = 0;
    for (int i = 1 ; i <= 2 * n ; ++i)
    {
        int last = countDiff;
        if (unique.size() < n && (second.size() == n || query(i) > last))
        {
            unique.push_back(i);
        } else
        {
            to[i] = unique.size();
            second.push_back(i);
        }
    }

    assert(unique.size() == n);
    assert(second.size() == n);

    int lg = 0;
    while ((1 << lg + 1) < n)
    {
        lg++;
    }

    for (int bit = lg ; bit >= 0 ; --bit)
    {
        int cntIn = 0;
        int cntOut = 0;
        int min = INF;
        
        std::fill(taken, taken + n, 0);
        std::fill(taken0, taken0 + n, 0);
        std::fill(taken1, taken1 + n, 0);
        for (int i = 0 ; i < n ; ++i)
        {
            if (((i & (1 << bit)) > 0) != isIn[unique[i]])
            {
                query(unique[i]);
            }

            if (isIn[unique[i]])
            {
                min = std::min(min, i);
            }

            cntIn += isIn[unique[i]];
            cntOut += !isIn[unique[i]];
        }

        for (int i = 0 ; i < n ; ++i)
        {
            if (cntIn == 0)
            {
                continue;
            }

            if (to[second[i]] < min)
            {
                cntOut--;
                continue;
            }

            if (cntOut == 0)
            {
                with[second[i]] |= (1 << bit);
                continue;
            }

            if (taken[with[second[i]]] == (1 << bit + 1) - 1)
            {
                if (taken1[with[second[i]]] < taken0[with[second[i]]])
                {
                    with[second[i]] |= (1 << bit);
                }

                continue;
            }
            
            int curr = countDiff;
            int next = query(second[i]);
            taken[with[second[i]]]++;
            if (curr == next)
            {
                taken1[with[second[i]]]++;
                with[second[i]] |= (1 << bit);
                cntIn--;
            } else
            {
                taken0[with[second[i]]]++;
                cntOut--;
            }
        }
    }

    for (int i = 0 ; i < n ; ++i)
    {
        answer(second[i], unique[with[second[i]]]);
    }
}

Compilation message

minerals.cpp: In function 'void Solve(int)':
minerals.cpp:48:27: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   48 |         if (unique.size() < n && (second.size() == n || query(i) > last))
      |             ~~~~~~~~~~~~~~^~~
minerals.cpp:48:49: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   48 |         if (unique.size() < n && (second.size() == n || query(i) > last))
      |                                   ~~~~~~~~~~~~~~^~~~
In file included from /usr/include/c++/10/cassert:44,
                 from minerals.cpp:5:
minerals.cpp:58:26: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   58 |     assert(unique.size() == n);
      |            ~~~~~~~~~~~~~~^~~~
minerals.cpp:59:26: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   59 |     assert(second.size() == n);
      |            ~~~~~~~~~~~~~~^~~~
minerals.cpp:62:21: warning: suggest parentheses around '+' inside '<<' [-Wparentheses]
   62 |     while ((1 << lg + 1) < n)
      |                  ~~~^~~
minerals.cpp:111:53: warning: suggest parentheses around '+' inside '<<' [-Wparentheses]
  111 |             if (taken[with[second[i]]] == (1 << bit + 1) - 1)
      |                                                 ~~~~^~~
minerals.cpp:44:9: warning: unused variable 'cnt' [-Wunused-variable]
   44 |     int cnt = 0;
      |         ^~~

Subtask #1
6.0 / 6.0

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

Subtask #2
25.0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2392 KB Output is correct
2 Correct 1 ms 2392 KB Output is correct
3 Correct 2 ms 2648 KB Output is correct
4 Correct 5 ms 2812 KB Output is correct
5 Correct 9 ms 2904 KB Output is correct

Subtask #3
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2392 KB Output is correct
2 Correct 1 ms 2392 KB Output is correct
3 Correct 1 ms 2392 KB Output is correct
4 Correct 1 ms 2496 KB Output is correct
5 Correct 1 ms 2392 KB Output is correct
6 Correct 1 ms 2392 KB Output is correct
7 Correct 2 ms 2648 KB Output is correct
8 Correct 5 ms 2812 KB Output is correct
9 Correct 9 ms 2904 KB Output is correct
10 Correct 1 ms 2392 KB Output is correct
11 Correct 6 ms 2904 KB Output is correct
12 Correct 9 ms 2900 KB Output is correct
13 Correct 8 ms 2904 KB Output is correct
14 Correct 8 ms 3328 KB Output is correct

Subtask #4
30.0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2392 KB Output is correct
2 Correct 1 ms 2392 KB Output is correct
3 Correct 1 ms 2392 KB Output is correct
4 Correct 1 ms 2496 KB Output is correct
5 Correct 1 ms 2392 KB Output is correct
6 Correct 1 ms 2392 KB Output is correct
7 Correct 2 ms 2648 KB Output is correct
8 Correct 5 ms 2812 KB Output is correct
9 Correct 9 ms 2904 KB Output is correct
10 Correct 1 ms 2392 KB Output is correct
11 Correct 6 ms 2904 KB Output is correct
12 Correct 9 ms 2900 KB Output is correct
13 Correct 8 ms 2904 KB Output is correct
14 Correct 8 ms 3328 KB Output is correct
15 Correct 29 ms 4064 KB Output is correct
16 Correct 38 ms 4124 KB Output is correct
17 Correct 36 ms 4152 KB Output is correct
18 Correct 30 ms 4156 KB Output is correct

Subtask #5
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2392 KB Output is correct
2 Correct 1 ms 2392 KB Output is correct
3 Correct 1 ms 2392 KB Output is correct
4 Correct 1 ms 2496 KB Output is correct
5 Correct 1 ms 2392 KB Output is correct
6 Correct 1 ms 2392 KB Output is correct
7 Correct 2 ms 2648 KB Output is correct
8 Correct 5 ms 2812 KB Output is correct
9 Correct 9 ms 2904 KB Output is correct
10 Correct 1 ms 2392 KB Output is correct
11 Correct 6 ms 2904 KB Output is correct
12 Correct 9 ms 2900 KB Output is correct
13 Correct 8 ms 2904 KB Output is correct
14 Correct 8 ms 3328 KB Output is correct
15 Correct 29 ms 4064 KB Output is correct
16 Correct 38 ms 4124 KB Output is correct
17 Correct 36 ms 4152 KB Output is correct
18 Correct 30 ms 4156 KB Output is correct
19 Correct 37 ms 4016 KB Output is correct
20 Correct 35 ms 4068 KB Output is correct
21 Correct 30 ms 4340 KB Output is correct
22 Correct 37 ms 3916 KB Output is correct

Subtask #6
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2392 KB Output is correct
2 Correct 1 ms 2392 KB Output is correct
3 Correct 1 ms 2392 KB Output is correct
4 Correct 1 ms 2496 KB Output is correct
5 Correct 1 ms 2392 KB Output is correct
6 Correct 1 ms 2392 KB Output is correct
7 Correct 2 ms 2648 KB Output is correct
8 Correct 5 ms 2812 KB Output is correct
9 Correct 9 ms 2904 KB Output is correct
10 Correct 1 ms 2392 KB Output is correct
11 Correct 6 ms 2904 KB Output is correct
12 Correct 9 ms 2900 KB Output is correct
13 Correct 8 ms 2904 KB Output is correct
14 Correct 8 ms 3328 KB Output is correct
15 Correct 29 ms 4064 KB Output is correct
16 Correct 38 ms 4124 KB Output is correct
17 Correct 36 ms 4152 KB Output is correct
18 Correct 30 ms 4156 KB Output is correct
19 Correct 37 ms 4016 KB Output is correct
20 Correct 35 ms 4068 KB Output is correct
21 Correct 30 ms 4340 KB Output is correct
22 Correct 37 ms 3916 KB Output is correct
23 Correct 33 ms 4076 KB Output is correct
24 Correct 31 ms 4064 KB Output is correct
25 Correct 29 ms 4308 KB Output is correct
26 Correct 29 ms 4052 KB Output is correct

Subtask #7
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2392 KB Output is correct
2 Correct 1 ms 2392 KB Output is correct
3 Correct 1 ms 2392 KB Output is correct
4 Correct 1 ms 2496 KB Output is correct
5 Correct 1 ms 2392 KB Output is correct
6 Correct 1 ms 2392 KB Output is correct
7 Correct 2 ms 2648 KB Output is correct
8 Correct 5 ms 2812 KB Output is correct
9 Correct 9 ms 2904 KB Output is correct
10 Correct 1 ms 2392 KB Output is correct
11 Correct 6 ms 2904 KB Output is correct
12 Correct 9 ms 2900 KB Output is correct
13 Correct 8 ms 2904 KB Output is correct
14 Correct 8 ms 3328 KB Output is correct
15 Correct 29 ms 4064 KB Output is correct
16 Correct 38 ms 4124 KB Output is correct
17 Correct 36 ms 4152 KB Output is correct
18 Correct 30 ms 4156 KB Output is correct
19 Correct 37 ms 4016 KB Output is correct
20 Correct 35 ms 4068 KB Output is correct
21 Correct 30 ms 4340 KB Output is correct
22 Correct 37 ms 3916 KB Output is correct
23 Correct 33 ms 4076 KB Output is correct
24 Correct 31 ms 4064 KB Output is correct
25 Correct 29 ms 4308 KB Output is correct
26 Correct 29 ms 4052 KB Output is correct
27 Incorrect 35 ms 4044 KB Wrong Answer [2]
28 Halted 0 ms 0 KB -

Subtask #8
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2392 KB Output is correct
2 Correct 1 ms 2392 KB Output is correct
3 Correct 1 ms 2392 KB Output is correct
4 Correct 1 ms 2496 KB Output is correct
5 Correct 1 ms 2392 KB Output is correct
6 Correct 1 ms 2392 KB Output is correct
7 Correct 2 ms 2648 KB Output is correct
8 Correct 5 ms 2812 KB Output is correct
9 Correct 9 ms 2904 KB Output is correct
10 Correct 1 ms 2392 KB Output is correct
11 Correct 6 ms 2904 KB Output is correct
12 Correct 9 ms 2900 KB Output is correct
13 Correct 8 ms 2904 KB Output is correct
14 Correct 8 ms 3328 KB Output is correct
15 Correct 29 ms 4064 KB Output is correct
16 Correct 38 ms 4124 KB Output is correct
17 Correct 36 ms 4152 KB Output is correct
18 Correct 30 ms 4156 KB Output is correct
19 Correct 37 ms 4016 KB Output is correct
20 Correct 35 ms 4068 KB Output is correct
21 Correct 30 ms 4340 KB Output is correct
22 Correct 37 ms 3916 KB Output is correct
23 Correct 33 ms 4076 KB Output is correct
24 Correct 31 ms 4064 KB Output is correct
25 Correct 29 ms 4308 KB Output is correct
26 Correct 29 ms 4052 KB Output is correct
27 Incorrect 35 ms 4044 KB Wrong Answer [2]
28 Halted 0 ms 0 KB -

Subtask #9
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 2392 KB Output is correct
2 Correct 1 ms 2392 KB Output is correct
3 Correct 1 ms 2392 KB Output is correct
4 Correct 1 ms 2496 KB Output is correct
5 Correct 1 ms 2392 KB Output is correct
6 Correct 1 ms 2392 KB Output is correct
7 Correct 2 ms 2648 KB Output is correct
8 Correct 5 ms 2812 KB Output is correct
9 Correct 9 ms 2904 KB Output is correct
10 Correct 1 ms 2392 KB Output is correct
11 Correct 6 ms 2904 KB Output is correct
12 Correct 9 ms 2900 KB Output is correct
13 Correct 8 ms 2904 KB Output is correct
14 Correct 8 ms 3328 KB Output is correct
15 Correct 29 ms 4064 KB Output is correct
16 Correct 38 ms 4124 KB Output is correct
17 Correct 36 ms 4152 KB Output is correct
18 Correct 30 ms 4156 KB Output is correct
19 Correct 37 ms 4016 KB Output is correct
20 Correct 35 ms 4068 KB Output is correct
21 Correct 30 ms 4340 KB Output is correct
22 Correct 37 ms 3916 KB Output is correct
23 Correct 33 ms 4076 KB Output is correct
24 Correct 31 ms 4064 KB Output is correct
25 Correct 29 ms 4308 KB Output is correct
26 Correct 29 ms 4052 KB Output is correct
27 Incorrect 35 ms 4044 KB Wrong Answer [2]
28 Halted 0 ms 0 KB -