Submission #1112625

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
1112625	2024-11-14T12:49:38 Z	Ghulam_Junaid	ICC (CEOI16_icc)	C++17	40 / 100	149 ms	764 KB

#include <bits/stdc++.h>
#include "icc.h"
// #include "grader.cpp"
using namespace std;

#define all(vv) vv.begin(), vv.end() 
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

const int N = 105;
int n, par[N];
vector<int> S[N];

int root(int v){
    return (par[v] == -1 ? v : par[v] = root(par[v]));
}

void merge(int u, int v){
    if ((u = root(u)) == (v = root(v)))
        return ;

    if (S[u].size() > S[v].size())
        swap(u, v);

    for (int x : S[u]){
        par[x] = v;
        S[v].push_back(x);
    }
    S[u].clear();
}

void run(int nn){
    n = nn;
    for (int i = 1; i <= n; i ++)
        par[i] = -1, S[i] = {i};

    for (int edge = 0; edge < n - 1; edge ++){
        vector<int> roots[2], vec[2];
        while (1){
            set<int> st;
            for (int v = 1; v <= n; v ++)
                st.insert(root(v));
            
            vector<int> all_roots;
            for (int v : st)
                all_roots.push_back(v);

            for (int j : {0, 1})
                roots[j].clear(), vec[j].clear();

            for (int x : all_roots){
                int id = rng() % 2;
                roots[id].push_back(x);
                for (int v : S[x])
                    vec[id].push_back(v);
            }

            int a[(int)vec[0].size()], b[(int)vec[1].size()];
            copy(all(vec[0]), a), copy(all(vec[1]), b);

            int res = query((int)vec[0].size(), (int)vec[1].size(), a, b);
            if (res) break;
        }

        for (int j : {0, 1}){
            int l = -1;
            int r = roots[j].size() - 1;

            while (r - l > 1){
                int mid = (l + r) / 2;

                vec[j].clear();
                for (int i = 0; i <= mid; i ++)
                    for (int v : S[roots[j][i]])
                        vec[j].push_back(v);

                int a[(int)vec[0].size()], b[(int)vec[1].size()];
                copy(all(vec[0]), a), copy(all(vec[1]), b);
                int res = query((int)vec[0].size(), (int)vec[1].size(), a, b);
                
                if (res)
                    r = mid;
                else
                    l = mid;

                vec[j].clear();
                for (int i = l + 1; i <= r; i ++)
                    for (int v : S[roots[j][i]])
                        vec[j].push_back(v);

            }
            int imp = roots[j][r];
            roots[j] = {imp};
        }

        vector<int> tmp[2] = {vec[0], vec[1]};
        for (int j : {0, 1}){
            int l = -1;
            int r = vec[j].size() - 1;

            while (r - l > 1){
                int mid = (l + r) / 2;

                tmp[j].clear();
                for (int i = 0; i <= mid; i ++)
                    tmp[j].push_back(vec[j][i]);

                int a[(int)tmp[0].size()], b[(int)tmp[1].size()];
                copy(all(tmp[0]), a), copy(all(tmp[1]), b);
                int res = query((int)tmp[0].size(), (int)tmp[1].size(), a, b);
                
                if (res)
                    r = mid;
                else
                    l = mid;

                tmp[j].clear();
                for (int i = l + 1; i <= r; i ++)
                    tmp[j].push_back(vec[j][i]);
            }
            vec[j] = tmp[j];
        }

        setRoad(vec[0][0], vec[1][0]);
        merge(vec[0][0], vec[1][0]);
    }
}

Subtask #1
7.0 / 7.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	592 KB	Ok! 122 queries used.
2	Correct	5 ms	648 KB	Ok! 122 queries used.

Subtask #2
11.0 / 11.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	28 ms	592 KB	Ok! 674 queries used.
2	Correct	53 ms	760 KB	Ok! 865 queries used.
3	Correct	37 ms	592 KB	Ok! 879 queries used.

Subtask #3
22.0 / 22.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	109 ms	592 KB	Ok! 1832 queries used.
2	Correct	134 ms	644 KB	Ok! 2123 queries used.
3	Correct	149 ms	764 KB	Ok! 2096 queries used.
4	Correct	105 ms	592 KB	Ok! 2033 queries used.