oj.uz

Submission #604540

# Submission time Handle Problem Language Result Execution time Memory
604540 2022-07-25T07:22:38 Z generic_placeholder_name ICC (CEOI16_icc) C++17
90 / 100
 354 ms 628 KB 
#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(_, 100) {
                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) {
            vi a, b;
            rep(i, l.size()) (i & 1 ? a : b).pb(l[i]);
            if(query(a, r)) l = a;
            else l = b;
        }
        while(r.size() > 1) {
            vi a, b;
            rep(i, r.size()) (i & 1 ? a : b).pb(r[i]);
            if(query(l, a)) r = a;
            else r = b;
        }
        int u = l[0], v = r[0];
        adj[u].pb(v);
        adj[v].pb(u);
        setRoad(u, v);
    }
}

Compilation message

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:89:13: note: in expansion of macro 'rep'
   89 |             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:95:13: note: in expansion of macro 'rep'
   95 |             rep(i, r.size()) (i & 1 ? a : b).pb(r[i]);
      |             ^~~

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 468 KB Ok! 97 queries used.
2 Correct 6 ms 468 KB Ok! 97 queries used.

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 36 ms 500 KB Ok! 528 queries used.
2 Correct 74 ms 492 KB Ok! 721 queries used.
3 Correct 88 ms 500 KB Ok! 710 queries used.

Subtask #3
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 146 ms 508 KB Ok! 1350 queries used.
2 Correct 354 ms 512 KB Ok! 1812 queries used.
3 Correct 151 ms 504 KB Ok! 1352 queries used.
4 Correct 244 ms 496 KB Ok! 1520 queries used.

Subtask #4
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 188 ms 504 KB Ok! 1418 queries used.
2 Correct 218 ms 504 KB Ok! 1473 queries used.
3 Correct 320 ms 628 KB Ok! 1716 queries used.
4 Correct 147 ms 468 KB Ok! 1343 queries used.

Subtask #5
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 332 ms 512 KB Ok! 1697 queries used.
2 Correct 324 ms 500 KB Ok! 1707 queries used.
3 Correct 317 ms 504 KB Ok! 1723 queries used.
4 Correct 279 ms 624 KB Ok! 1647 queries used.
5 Correct 196 ms 500 KB Ok! 1444 queries used.
6 Correct 275 ms 500 KB Ok! 1620 queries used.

Subtask #6
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 316 ms 508 KB Too many queries! 1703 out of 1625
2 Halted 0 ms 0 KB -