Submission #1028747

# Submission time Handle Problem Language Result Execution time Memory
1028747 2024-07-20T08:06:27 Z serifefedartar ICC (CEOI16_icc) C++17
90 / 100
103 ms 856 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 < 7; 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)) {
                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++) {
      |                             ~~^~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 3 ms 600 KB Ok! 97 queries used.
2 Correct 4 ms 604 KB Ok! 105 queries used.
# Verdict Execution time Memory Grader output
1 Correct 19 ms 604 KB Ok! 532 queries used.
2 Correct 28 ms 604 KB Ok! 683 queries used.
3 Correct 24 ms 604 KB Ok! 680 queries used.
# Verdict Execution time Memory Grader output
1 Correct 63 ms 856 KB Ok! 1437 queries used.
2 Correct 84 ms 604 KB Ok! 1689 queries used.
3 Correct 87 ms 644 KB Ok! 1597 queries used.
4 Correct 77 ms 652 KB Ok! 1514 queries used.
# Verdict Execution time Memory Grader output
1 Correct 64 ms 640 KB Ok! 1468 queries used.
2 Correct 69 ms 664 KB Ok! 1501 queries used.
3 Correct 93 ms 660 KB Ok! 1633 queries used.
4 Correct 69 ms 648 KB Ok! 1477 queries used.
# Verdict Execution time Memory Grader output
1 Correct 79 ms 640 KB Ok! 1633 queries used.
2 Correct 103 ms 600 KB Ok! 1659 queries used.
3 Correct 80 ms 656 KB Ok! 1669 queries used.
4 Correct 80 ms 600 KB Ok! 1621 queries used.
5 Correct 71 ms 604 KB Ok! 1501 queries used.
6 Correct 72 ms 600 KB Ok! 1591 queries used.
# Verdict Execution time Memory Grader output
1 Incorrect 74 ms 640 KB Too many queries! 1636 out of 1625
2 Halted 0 ms 0 KB -