Submission #604556

#	Time	Username	Problem	Language	Result	Execution time	Memory
604556		generic_placeholder_name	ICC (CEOI16_icc)	C++17	100 / 100	533 ms	624 KiB

icc

#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("avx,avx2,fma,popcnt")

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/trie_policy.hpp>
#include <ext/rope>

using namespace std;
using namespace __gnu_pbds;
using namespace __gnu_cxx;

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

#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define mp make_pair
#define gcd __gcd
#define fastio ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
#define rep(i, n) for (int i=0; i<(n); i++)
#define rep1(i, n) for (int i=1; i<=(n); i++)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define endl "\n"

typedef long long ll;
typedef unsigned long long ull;
typedef unsigned uint;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;
typedef vector<vector<int>> vvi;
typedef vector<ll> vll;
typedef vector<vector<ll>> vvll;
typedef vector<bool> vb;
typedef vector<vector<bool>> vvb;
template<typename T, typename cmp = less<T>>
using ordered_set = tree<T, null_type, cmp, rb_tree_tag, tree_order_statistics_node_update>;
typedef trie<string, null_type, trie_string_access_traits<>, pat_trie_tag, trie_prefix_search_node_update> pref_trie;

constexpr int N = 105;
int color[N];
bool ban[N][N];
vi adj[N];

void dfs(int u) {
    for(int v: adj[u]) if(!~color[v]) {
        color[v] = color[u];
        dfs(v);
    }
}

#include "icc.h"

int query(vi& a, vi& b) {
    return query(a.size(), b.size(), a.data(), b.data());
}

void run(int N) {
    int& n = N;
    rep(_, n - 1) {
        vi l, r;
        memset(ban, 0, sizeof(ban));
        while(1) {
            int mx = 0;
            rep(_, 200) {
                memset(color, -1, sizeof(color));
                rep1(i, n) if(!~color[i]) {
                    color[i] = rng() & 1;
                    dfs(i);
                }
                vi cl, cr;
                rep1(i, n) (color[i] ? cl : cr).pb(i);
                int cnt = 0;
                for(int x: cl) for(int y: cr) cnt += !ban[x][y];
                if(cnt > mx) mx = cnt, l = cl, r = cr;
            }
            if(query(l, r)) break;
            else {
                for(int x: l) for(int y: r) ban[x][y] = ban[y][x] = 1;
            }
        }
        while(l.size() > 1 || r.size() > 1) {
            shuffle(all(l), rng);
            shuffle(all(r), rng);
            vi a, b;
            if(l.size() > 1) {
                rep(i, l.size()) (i & 1 ? a : b).pb(l[i]);
                if(query(a, r)) l = a;
                else l = b;
                a.clear(), b.clear();
            }
            vi rr;
            for(int x: r) {
                bool ok = 0;
                for(int y: l) if(!ban[x][y]) {ok = 1; break;}
                if(ok) rr.pb(x);
            }
            r = rr;
            if(r.size() > 1) {
                rep(i, r.size()) (i & 1 ? a : b).pb(r[i]);
                if(query(l, a)) r = a;
                else r = b;
            }
            vi ll;
            for(int x: l) {
                bool ok = 0;
                for(int y: r) if(!ban[x][y]) {ok = 1; break;}
                if(ok) ll.pb(x);
            }
            l = ll;
        }
        int u = l[0], v = r[0];
        adj[u].pb(v);
        adj[v].pb(u);
        setRoad(u, v);
    }
}

Compilation message (stderr)

icc.cpp: In function 'void run(int)':
icc.cpp:23:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 | #define rep(i, n) for (int i=0; i<(n); i++)
      |                                  ^
icc.cpp:92:17: note: in expansion of macro 'rep'
   92 |                 rep(i, l.size()) (i & 1 ? a : b).pb(l[i]);
      |                 ^~~
icc.cpp:23:34: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   23 | #define rep(i, n) for (int i=0; i<(n); i++)
      |                                  ^
icc.cpp:105:17: note: in expansion of macro 'rep'
  105 |                 rep(i, r.size()) (i & 1 ? a : b).pb(r[i]);
      |                 ^~~

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