oj.uz

Submission #169482

# Submission time Handle Problem Language Result Execution time Memory
169482 2019-12-20T15:30:52 Z dolphingarlic ICC (CEOI16_icc) C++14
100 / 100
 146 ms 636 KB 
/*
CEOI 2016 ICC
- We want to binary search for the answer somehow
- We should always group cities in the same component together or else
  query always returns 1
- Consider the binary representations of the components
- If we split the components based on their i-th bit (from the right), then
  query(Split from i-th bit) == 1 for the first time iff all bits to the right of
  i are the same for the 2 newly connected components
- We can fill in the rest of the bits by fixing a particular bit and seeing if the
  query still returns 1
- After finding the 2 connected components, we can just binary search within those components
*/

#include "icc.h"
#include <bits/stdc++.h>
#define FOR(i, x, y) for (int i = x; i < y; i++)
using namespace std;

vector<int> a, b, cmp[101], graph[101];
bool visited[101];

void dfs(int node, int idx) {
    visited[node] = true;
    cmp[idx].push_back(node);
    for (int i : graph[node]) if (!visited[i]) dfs(i, idx);
}

void run (int n) {
    while (true) {
        int idx = 0;
        for (int i = 1; i <= n; i++)
            if (!visited[i]) dfs(i, idx++);

        for (int i = 0; ; i++) {
            // Check if the current split has a path between stuff
            a.clear(); b.clear();
            FOR(j, 0, idx) {
                for (int k : cmp[j]) {
                    if ((j >> i) & 1) a.push_back(k);
                    else b.push_back(k);
                }
            }

            if (!query(a.size(), b.size(), a.data(), b.data())) continue;

            // Find the exact bits
            int exact = 0;
            FOR(k, 0, i) {
                a.clear(); b.clear();
                FOR(j, 0, idx) {
                    if ((j & ((1 << (k + 1)) - 1)) == exact) {
                        for (int l : cmp[j]) {
                            if ((j >> i) & 1) a.push_back(l);
                            else b.push_back(l);
                        }
                    }
                }

                if (!query(a.size(), b.size(), a.data(), b.data())) exact |= (1 << k);
            }

            // Binary search in the component
            a.clear(); b.clear();
            FOR(j, 0, idx) {
                if ((j & ((1 << i) - 1)) == exact) {
                    for (int k : cmp[j]) {
                        if ((j >> i) & 1) a.push_back(k);
                        else b.push_back(k);
                    }
                }
            }

            while (a.size() > 1) {
                if (query(a.size() / 2, b.size(), a.data(), b.data())) a.erase(a.begin() + a.size() / 2, a.end());
                else a.erase(a.begin(), a.begin() + a.size() / 2);
            }
            while (b.size() > 1) {
                if (query(b.size() / 2, a.size(), b.data(), a.data())) b.erase(b.begin() + b.size() / 2, b.end());
                else b.erase(b.begin(), b.begin() + b.size() / 2);
            }

            int u = a[0], v = b[0];
            graph[u].push_back(v);
            graph[v].push_back(u);
            setRoad(u, v);
            break;
        }

        memset(visited, 0, sizeof(visited));
        FOR(i, 0, idx) cmp[i].clear();
    }
}

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 504 KB Ok! 97 queries used.
2 Correct 8 ms 504 KB Ok! 98 queries used.

Subtask #2
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 39 ms 504 KB Ok! 520 queries used.
2 Correct 43 ms 504 KB Ok! 571 queries used.
3 Correct 46 ms 604 KB Ok! 613 queries used.

Subtask #3
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 132 ms 632 KB Ok! 1430 queries used.
2 Correct 136 ms 600 KB Ok! 1385 queries used.
3 Correct 135 ms 636 KB Ok! 1464 queries used.
4 Correct 144 ms 568 KB Ok! 1481 queries used.

Subtask #4
21.0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 137 ms 504 KB Ok! 1450 queries used.
2 Correct 140 ms 632 KB Ok! 1467 queries used.
3 Correct 135 ms 504 KB Ok! 1468 queries used.
4 Correct 136 ms 504 KB Ok! 1463 queries used.

Subtask #5
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 146 ms 568 KB Ok! 1550 queries used.
2 Correct 134 ms 504 KB Ok! 1468 queries used.
3 Correct 136 ms 600 KB Ok! 1446 queries used.
4 Correct 135 ms 632 KB Ok! 1468 queries used.
5 Correct 135 ms 600 KB Ok! 1455 queries used.
6 Correct 136 ms 504 KB Ok! 1462 queries used.

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 135 ms 504 KB Ok! 1468 queries used.
2 Correct 135 ms 564 KB Ok! 1468 queries used.