#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], tmpS[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; ++_) {
for (int i = 0; i < num_components; ++i) swap(tmpS[i], S[i]);
vector<int> perm(num_components); iota(perm.begin(), perm.end(), 0);
shuffle(perm.begin(), perm.end(), rng);
for (int i = 0; i < num_components; ++i) swap(S[perm[i]], tmpS[i]);
vector<int> bits, same_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;
same_bits.emplace_back(bit);
}
shuffle(a, a + sza, rng);
int low = 1, high = sza, pu = -1, pv = -1, u = -1, v = -1;
while (low <= high) {
int mid = low + high >> 1;
if (query(mid, szb, a, b)) high = (pu = mid) - 1;
else low = mid + 1;
}
u = a[pu - 1];
// assert(pu != -1);
vector<int> keep;
for (int i = 0; i < szb; ++i) {
bool flag = true;
for (auto bit : same_bits) if (BIT(b[i], bit) != BIT(u, bit)) {
flag = false;
break;
}
if (flag) keep.emplace_back(b[i]);
}
szb = keep.size();
for (int i = 0; i < szb; ++i) b[i] = keep[i];
shuffle(b, b + szb, rng);
low = 1, high = szb;
while (low <= high) {
int mid = low + high >> 1;
if (query(pu, mid, a, b)) high = (pv = mid) - 1;
else low = mid + 1;
}
v = b[pv - 1];
// assert(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:139:18: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
139 | int mid = low + high >> 1;
| ~~~~^~~~~~
icc.cpp:161:18: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
161 | int mid = low + high >> 1;
| ~~~~^~~~~~
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
2 ms |
600 KB |
Wrong road! |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
1 ms |
604 KB |
Wrong road! |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
1 ms |
604 KB |
Wrong road! |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
1 ms |
604 KB |
Wrong road! |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
1 ms |
604 KB |
Wrong road! |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
1 ms |
644 KB |
Wrong road! |
| 2 |
Halted |
0 ms |
0 KB |
- |
