#ifndef hwe
#include "icc.h"
#endif
#include <bits/stdc++.h>
using namespace std;
#define popcount(n) (__builtin_popcountll((n)))
#define clz(n) (__builtin_clzll((n)))
#define ctz(n) (__builtin_ctzll((n)))
#define lg(n) (63 - __builtin_clzll((n)))
#define BIT(n, i) (((n) >> (i)) & 1ll)
#define MASK(i) (1ll << (i))
#define FLIP(n, i) ((n) ^ (1ll << (i)))
#define ON(n, i) ((n) | MASK(i))
#define OFF(n, i) ((n) & ~MASK(i))
#define Int __int128
#define fi first
#define se second
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<long long, long long> pll;
typedef pair<long long, int> pli;
typedef pair<int, long long> pil;
typedef vector<pair<int, int>> vii;
typedef vector<pair<long long, long long>> vll;
typedef vector<pair<long long, int>> vli;
typedef vector<pair<int, long long>> vil;
template <class T1, class T2>
bool maximize(T1 &x, T2 y) {
if (x < y) {
x = y;
return true;
}
return false;
}
template <class T1, class T2>
bool minimize(T1 &x, T2 y) {
if (x > y) {
x = y;
return true;
}
return false;
}
template <class T>
void remove_duplicate(vector<T> &ve) {
sort (ve.begin(), ve.end());
ve.resize(unique(ve.begin(), ve.end()) - ve.begin());
}
mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count());
template <class T> T random(T l, T r) {
return uniform_int_distribution<T>(l, r)(rng);
}
template <class T> T random(T r) {
return rng() % r;
}
const int N = 100 + 5;
const int MOD = 1e9 + 7;
const int inf = 1e9;
const ll INF = 1e18;
#ifdef hwe
int n, current_road = 0;
map<int, int> mp[N];
pii roads[N];
int query(int sza, int szb, int a[], int b[]) {
// cerr << "Query a: ";
// for (int i = 0; i < sza; ++i) cerr << a[i] << ' ';
// cerr << '\n';
// cerr << "Query b: ";
// for (int i = 0; i < szb; ++i) cerr << b[i] << ' ';
// cerr << '\n';
for (int i = 0; i < sza; ++i) for (int j = 0; j < szb; ++j) {
int u = a[i], v = b[j];
if (u > v) swap(u, v);
if (mp[u][v] == 1) return 1;
}
return 0;
}
void setRoad(int u, int v) {
if (u > v) swap(u, v);
cerr << "Set road: " << u << ' ' << v << '\n';
++current_road;
mp[roads[current_road].fi][roads[current_road].se] = 1;
}
#endif
int a[N], b[N];
vector<int> S[N];
void run(int n) {
int num_components = n;
for (int i = 0; i < num_components; ++i) S[i] = {i + 1};
int sza = 0, szb = 0;
for (int _ = 1; _ < n; ++_) {
vector<int> bits;
for (int bit = 0; MASK(bit) < num_components; ++bit) bits.emplace_back(bit);
shuffle(bits.begin(), bits.end(), rng);
for (auto bit : bits) {
vector<int> ones, zeros;
for (int i = 0; i < num_components; ++i) {
if (BIT(i, bit)) ones.emplace_back(i);
else zeros.emplace_back(i);
}
sza = 0, szb = 0;
for (auto &x : zeros) for (auto &y : S[x]) {
a[sza++] = y;
}
for (auto &x : ones) for (auto &y : S[x]) {
b[szb++] = y;
}
int tmp = query(sza, szb, a, b);
if (tmp == 1) break;
}
int pu = sza, pv = szb;
if (sza <= 10) {
vector<int> order(sza - 1); iota(order.begin(), order.end(), 1);
shuffle(order.begin(), order.end(), rng);
for (auto p : order) {
if (p > pu) continue;
if (query(p, szb, a, b)) pu = p;
}
}
else {
int low = 1, high = sza - 1;
while (low <= high) {
int mid = low + high >> 1;
if (query(mid, szb, a, b)) high = (pu = mid) - 1;
else low = mid + 1;
}
}
// assert(pu != -1);
if (szb <= 10) {
vector<int> order(szb - 1); iota(order.begin(), order.end(), 1);
shuffle(order.begin(), order.end(), rng);
for (auto p : order) {
if (p > pv) continue;
if (query(pu, p, a, b)) pv = p;
}
}
else {
int low = 1, high = szb - 1;
while (low <= high) {
int mid = low + high >> 1;
if (query(pu, mid, a, b)) high = (pv = mid) - 1;
else low = mid + 1;
}
}
// assert(pv != -1);
int u = a[pu - 1], v = b[pv - 1];
setRoad(u, v);
int p = -1, q = -1;
for (int i = 0; i < num_components; ++i) for (auto &x : S[i]) {
if (x == u) p = i;
if (x == v) q = i;
}
for (auto &x : S[q]) S[p].emplace_back(x);
swap(S[q], S[num_components - 1]); --num_components;
}
}
#ifdef hwe
signed main() {
cin >> n;
for (int i = 0; i < n - 1; ++i) {
int u, v; cin >> u >> v;
if (u > v) swap(u, v);
roads[i] = make_pair(u, v);
}
current_road = 0; mp[roads[0].fi][roads[0].se] = 1;
run(n);
cerr << '\n'; return 0;
}
#endif
Compilation message
icc.cpp: In function 'void run(int)':
icc.cpp:141:19: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
141 | int mid = low + high >> 1;
| ~~~~^~~~~~
icc.cpp:159:19: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
159 | int mid = low + high >> 1;
| ~~~~^~~~~~
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
5 ms |
604 KB |
Ok! 133 queries used. |
| 2 |
Correct |
4 ms |
636 KB |
Ok! 118 queries used. |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
20 ms |
604 KB |
Ok! 580 queries used. |
| 2 |
Correct |
27 ms |
604 KB |
Ok! 718 queries used. |
| 3 |
Correct |
27 ms |
600 KB |
Ok! 713 queries used. |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
63 ms |
624 KB |
Ok! 1477 queries used. |
| 2 |
Correct |
77 ms |
600 KB |
Ok! 1734 queries used. |
| 3 |
Correct |
66 ms |
600 KB |
Ok! 1597 queries used. |
| 4 |
Correct |
71 ms |
620 KB |
Ok! 1595 queries used. |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
64 ms |
600 KB |
Ok! 1542 queries used. |
| 2 |
Correct |
69 ms |
604 KB |
Ok! 1581 queries used. |
| 3 |
Correct |
72 ms |
604 KB |
Ok! 1702 queries used. |
| 4 |
Correct |
64 ms |
620 KB |
Ok! 1545 queries used. |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
72 ms |
604 KB |
Ok! 1704 queries used. |
| 2 |
Correct |
72 ms |
856 KB |
Ok! 1704 queries used. |
| 3 |
Correct |
72 ms |
604 KB |
Ok! 1707 queries used. |
| 4 |
Correct |
71 ms |
620 KB |
Ok! 1696 queries used. |
| 5 |
Correct |
65 ms |
600 KB |
Ok! 1562 queries used. |
| 6 |
Correct |
82 ms |
600 KB |
Ok! 1590 queries used. |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
73 ms |
620 KB |
Too many queries! 1724 out of 1625 |
| 2 |
Halted |
0 ms |
0 KB |
- |
