oj.uz

Submission #1028774

# Submission time Handle Problem Language Result Execution time Memory
1028774 2024-07-20T08:30:40 Z serifefedartar ICC (CEOI16_icc) C++17
90 / 100
 148 ms 664 KB 
#include <bits/stdc++.h>
#include "icc.h"
using namespace std;

#define fast ios::sync_with_stdio(0);cin.tie(0);
typedef long long ll;
#define f first
#define s second
#define LOGN 20
const ll MOD = 1e9 + 7;
const ll MAXN = 105;

int par[MAXN], sz[MAXN], temp1[MAXN], temp2[MAXN];
int find(int node) {
    if (node == par[node])
        return node;
    return par[node] = find(par[node]);
}

void unite(int a, int b) {
    a = find(a);
    b = find(b);
    if (a == b)
        return ;
    if (sz[b] > sz[a])
        swap(a, b);
    sz[a] += sz[b];
    par[b] = a;
}

int qry(int sz1, int sz2, vector<int> A, vector<int> B) {
    for (int i = 0; i < sz1; i++)
        temp1[i] = A[i];
    for (int i = 0; i < sz2; i++)
        temp2[i] = B[i];
    return query(sz1, sz2, temp1, temp2);
}

int n;
void match(vector<vector<int>> arr) {
    while (true) {
        random_shuffle(arr.begin(), arr.end());

        for (int rep = 0; rep < 10; rep++) {
            vector<int> A, B;
            for (int i = 0; i < arr.size(); i++) {
                if (i & (1<<rep)) {
                    for (auto u : arr[i])
                        A.push_back(u);
                } else {
                    for (auto u : arr[i])
                        B.push_back(u);
                }
            }

            if (qry(A.size(), B.size(), A, B)) {
                random_shuffle(A.begin(), A.end());
                random_shuffle(B.begin(), B.end());
                while ((int)B.size() > 1) {
                    vector<int> X, Y;
                    for (int i = 0; i < (int)B.size() / 2; i++)
                        X.push_back(B[i]);
                    for (int i = (int)B.size() / 2; i < (int)B.size(); i++)
                        Y.push_back(B[i]);

                    if (qry(A.size(), X.size(), A, X))
                        B = X;
                    else
                        B = Y;
                }

                while ((int)A.size() > 1) {
                    vector<int> X, Y;
                    for (int i = 0; i < (int)A.size() / 2; i++)
                        X.push_back(A[i]);
                    for (int i = (int)A.size() / 2; i < (int)A.size(); i++)
                        Y.push_back(A[i]);

                    if (qry(X.size(), 1, X, B))
                        A = X;
                    else
                        A = Y;
                }

                setRoad(B[0], A[0]);
                unite(B[0], A[0]);

                return ;
            }
        }
    }
}

void solve() {
    map<int, vector<int>> mp;
    for (int i = 1; i <= n; i++)
        mp[find(i)].push_back(i);

    vector<vector<int>> temp;
    for (auto u : mp)
        temp.push_back(u.s);

    match(temp);
}

void run(int N) {
    n = N;
    for (int i = 1; i <= N; i++)
        sz[i] = 1, par[i] = i;
    for (int i = 1; i < N; i++)
        solve();
}

Compilation message

icc.cpp: In function 'void match(std::vector<std::vector<int> >)':
icc.cpp:46:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   46 |             for (int i = 0; i < arr.size(); i++) {
      |                             ~~^~~~~~~~~~~~

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 600 KB Ok! 104 queries used.
2 Correct 4 ms 604 KB Ok! 102 queries used.

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 48 ms 604 KB Ok! 542 queries used.
2 Correct 45 ms 600 KB Ok! 678 queries used.
3 Correct 27 ms 600 KB Ok! 671 queries used.

Subtask #3
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 71 ms 656 KB Ok! 1418 queries used.
2 Correct 119 ms 664 KB Ok! 1660 queries used.
3 Correct 79 ms 440 KB Ok! 1528 queries used.
4 Correct 84 ms 648 KB Ok! 1549 queries used.

Subtask #4
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 78 ms 648 KB Ok! 1561 queries used.
2 Correct 105 ms 604 KB Ok! 1539 queries used.
3 Correct 88 ms 604 KB Ok! 1630 queries used.
4 Correct 126 ms 660 KB Ok! 1465 queries used.

Subtask #5
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 83 ms 660 KB Ok! 1627 queries used.
2 Correct 148 ms 640 KB Ok! 1639 queries used.
3 Correct 83 ms 600 KB Ok! 1651 queries used.
4 Correct 103 ms 640 KB Ok! 1639 queries used.
5 Correct 102 ms 660 KB Ok! 1504 queries used.
6 Correct 84 ms 604 KB Ok! 1559 queries used.

Subtask #6
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 84 ms 604 KB Too many queries! 1630 out of 1625
2 Halted 0 ms 0 KB -