Submission #259240

#	Time	Username	Problem	Language	Result	Execution time	Memory
259240		doowey	ICC (CEOI16_icc)	C++14	100 / 100	170 ms	632 KiB

icc

#include <bits/stdc++.h>
#include "icc.h"

using namespace std;

typedef pair<int, int> pii;

#define fi first
#define se second
#define mp make_pair

int ask(vector<int> ta, vector<int> tb){
    int n = ta.size();
    int m = tb.size();
    int f1[n];
    int f2[m];
    for(int i = 0 ; i < n; i ++ )
        f1[i]=ta[i];
    for(int i = 0 ; i < m; i ++ )
        f2[i]=tb[i];
    return query(n, m, f1, f2);
}

const int MAXN = 105;
int par[MAXN];
int sz[MAXN];

int fin(int x){
    if(x==par[x])
        return x;
    return par[x]=fin(par[x]);
}

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

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

void run(int N){
    for(int i = 1 ; i <= N ; i ++ )
        par[i]=i,sz[i]=1;
    int i1, i2;
    int qs = 0;
    int xi, yi;
    int sz;
    for(int i = 0 ; i < N - 1; i ++ ){
        vector<vector<int>> G(N + 1);
        for(int j = 1; j <= N; j ++ ){
            G[fin(j)].push_back(j);
        }
        vector<vector<int>> nw;
        for(int j = 1; j <= N; j ++ ){
            if(!G[j].empty())
                nw.push_back(G[j]);
        }
        sz = nw.size();
        vector<int> lis1, lis2;
        for(int lg = 8; lg >= 0 ; lg -- ){
            if((1 << lg) >= nw.size()) continue;
            vector<int> pi, qi;
            for(int j = 0 ; j < nw.size(); j ++ ){
                if((j & (1 << lg))){
                    for(auto x : nw[j]) pi.push_back(x);
                }
                else{
                    for(auto x : nw[j]) qi.push_back(x);
                }
            }
            if(pi.empty() || qi.empty()) continue;
            if(ask(pi, qi)){
                lis1 = pi;
                lis2 = qi;
                break;
            }
        }
        int l = 0, r = lis1.size() - 1;
        int mid;
        while(l < r){
            mid = (l + r ) / 2;
            vector<int> fsh;
            for(int j = 0 ; j <= mid; j ++){
                fsh.push_back(lis1[j]);
            }
            if(ask(fsh,lis2))
                r = mid;
            else
                l = mid + 1;
        }
        i1 = lis1[l];
        l = 0, r = lis2.size() - 1;
        while(l < r){
            mid = (l + r) / 2;
            vector<int> fsh;
            for(int j = 0 ; j <= mid ; j ++ ){
                fsh.push_back(lis2[j]);
            }
            if(ask(lis1,fsh)){
                r = mid;
            }
            else{
                l = mid + 1;
            }
        }
        i2 = lis2[l];
        setRoad(i1, i2);
        unite(i1,i2);
    }
}

Compilation message (stderr)

icc.cpp: In function 'void run(int)':
icc.cpp:66:26: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             if((1 << lg) >= nw.size()) continue;
                ~~~~~~~~~~^~~~~~~~~~~~
icc.cpp:68:31: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             for(int j = 0 ; j < nw.size(); j ++ ){
                             ~~^~~~~~~~~~~
icc.cpp:50:9: warning: unused variable 'qs' [-Wunused-variable]
     int qs = 0;
         ^~
icc.cpp:51:9: warning: unused variable 'xi' [-Wunused-variable]
     int xi, yi;
         ^~
icc.cpp:51:13: warning: unused variable 'yi' [-Wunused-variable]
     int xi, yi;
             ^~
icc.cpp:52:9: warning: variable 'sz' set but not used [-Wunused-but-set-variable]
     int sz;
         ^~

• Subtask 1: 7 / 7
• Subtask 2: 11 / 11
• Subtask 3: 22 / 22
• Subtask 4: 21 / 21
• Subtask 5: 29 / 29
• Subtask 6: 10 / 10