#include "icc.h"
#include <bits/stdc++.h>
using namespace std;
vector<vector<int>> components;
int Query(vector<int> A, vector<int> B) {
if (A.empty() || B.empty()) return 0;
return query(A.size(), B.size(), A.data(), B.data());
}
int QueryComponent(vector<int> A, vector<int> B) {
vector<int> P, Q;
for (auto i : A) {
for (auto j : components[i]) {
P.emplace_back(j);
}
}
for (auto i : B) {
for (auto j : components[i]) {
Q.emplace_back(j);
}
}
return Query(P, Q);
}
void Answer(int A, int B) {
return setRoad(A, B);
}
int Split(int L, int R, vector<int> &A, vector<int> &B, int depth) {
if (depth == 0) {
int mid = (L + R) / 2;
for (int i = L; i <= mid; i++) A.emplace_back(i);
for (int i = mid + 1; i <= R; i++) B.emplace_back(i);
return max(mid - L + 1, R - mid);
}
int mid = (L + R) / 2;
return max(Split(L, mid, A, B, depth - 1), Split(mid + 1, R, A, B, depth - 1));
}
pair<int, int> BinarySearch(vector<int> A, vector<int> B, bool onComponent) {
pair<int, int> res = {-1, -1};
{
int lo = 0, hi = A.size() - 1;
while (lo < hi) {
int mid = (lo + hi) / 2;
vector<int> halfA;
for (int i = 0; i <= mid; i++) {
halfA.emplace_back(A[i]);
}
int res = (onComponent ? QueryComponent(halfA, B) : Query(halfA, B));
if (res) {
hi = mid;
} else {
lo = mid + 1;
}
}
res.first = A[lo];
}
{
int lo = 0, hi = B.size() - 1;
while (lo < hi) {
int mid = (lo + hi) / 2;
vector<int> halfB;
for (int i = 0; i <= mid; i++) {
halfB.emplace_back(B[i]);
}
int res = (onComponent ? QueryComponent(halfB, A) : Query(halfB, A));
if (res) {
hi = mid;
} else {
lo = mid + 1;
}
}
res.second = B[lo];
}
return res;
}
void MergeComponent(int C1, int C2) {
vector<int> Merged;
for (auto i : components[C1]) Merged.emplace_back(i);
for (auto i : components[C2]) Merged.emplace_back(i);
components[C1].clear();
components[C2].clear();
sort(begin(components), end(components), [&](const vector<int> &a, const vector<int> &b) {
return a.size() > b.size();
});
while (components.back().empty()) components.pop_back();
components.emplace_back(Merged);
}
void run(int N) {
for (int i = 1; i <= N; i++) {
components.emplace_back(vector<int>{i});
}
for (int i = 1; i < N; i++) {
vector<int> A, B; // find two components that are connected by new edge
for (int depth = 0; ; depth++) {
int sz = Split(0, components.size() - 1, A, B, depth);
if (sz == 1 || QueryComponent(A, B)) break;
A.clear(), B.clear();
}
// Binary Search to find edge's endpoint
auto TwoComp = BinarySearch(A, B, true);
auto Edge = BinarySearch(components[TwoComp.first], components[TwoComp.second], false);
Answer(Edge.first, Edge.second);
MergeComponent(TwoComp.first, TwoComp.second);
}
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
512 KB |
Ok! 107 queries used. |
2 |
Correct |
11 ms |
512 KB |
Ok! 109 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
57 ms |
512 KB |
Ok! 638 queries used. |
2 |
Correct |
51 ms |
512 KB |
Ok! 632 queries used. |
3 |
Correct |
65 ms |
512 KB |
Ok! 666 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
163 ms |
512 KB |
Ok! 1749 queries used. |
2 |
Correct |
155 ms |
512 KB |
Ok! 1651 queries used. |
3 |
Correct |
188 ms |
632 KB |
Ok! 1919 queries used. |
4 |
Correct |
171 ms |
512 KB |
Ok! 1784 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
181 ms |
512 KB |
Ok! 1858 queries used. |
2 |
Correct |
175 ms |
512 KB |
Ok! 1851 queries used. |
3 |
Correct |
172 ms |
512 KB |
Ok! 1720 queries used. |
4 |
Correct |
171 ms |
512 KB |
Ok! 1784 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
178 ms |
488 KB |
Ok! 1741 queries used. |
2 |
Correct |
179 ms |
632 KB |
Ok! 1690 queries used. |
3 |
Correct |
168 ms |
512 KB |
Ok! 1703 queries used. |
4 |
Incorrect |
174 ms |
512 KB |
Too many queries! 1786 out of 1775 |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
209 ms |
564 KB |
Too many queries! 1708 out of 1625 |
2 |
Halted |
0 ms |
0 KB |
- |