#include "icc.h"
#include <bits/stdc++.h>
using namespace std;
#define all(x) begin(x), end(x)
#define sz(x) ((int)size(x))
#define trace(x) cout << #x << ": " << (x) << endl;
typedef long long ll;
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
int ask(vector<int> a, vector<int> b) {
for (int &x: a) ++x;
for (int &y: b) ++y;
return query(sz(a), sz(b), a.data(), b.data());
}
vector<int> p;
int get(int a) {
return a == p[a] ? a : p[a] = get(p[a]);
}
bool mrg(int a, int b) {
a = get(a), b = get(b);
if (a == b) {
return false;
}
p[b] = a;
return true;
}
void run(int n) {
p.resize(n);
iota(all(p), 0);
for (int cnt = 1; cnt < n; ++cnt) {
vector<int> v;
for (int i = 0; i < n; ++i) {
if (p[i] == i) v.push_back(i);
}
while (true) {
vector<int> aa, bb, a, b;
for (int i = 0; i < sz(v); ++i) {
if ((rnd() & 1) == (i & 1)) aa.push_back(v[i]);
else bb.push_back(v[i]);
}
for (int x: aa) {
for (int i = 0; i < n; ++i)
if (get(i) == x)
a.push_back(i);
}
for (int x: bb) {
for (int i = 0; i < n; ++i)
if (get(i) == x)
b.push_back(i);
}
if (!a.empty() && !b.empty() && (abs(sz(a) - sz(b)) <= 5 && ask(a, b)) {
int l = -1, r = sz(a) - 1;
while (l + 1 < r) {
vector<int> cc;
int m = (l + r) >> 1;
for (int i = 0; i <= m; ++i) cc.push_back(a[i]);
if (!ask(cc, b)) l = m;
else r = m;
}
int v1 = a[r];
l = -1, r = sz(b) - 1;
while (l + 1 < r) {
vector<int> cc;
int m = (l + r) >> 1;
for (int i = 0; i <= m; ++i) cc.push_back(b[i]);
if (!ask(a, cc)) l = m;
else r = m;
}
int v2 = b[r];
mrg(v1, v2);
setRoad(v1 + 1, v2 + 1);
break;
}
}
}
}
Compilation message
icc.cpp: In function 'void run(int)':
icc.cpp:61:83: error: expected ')' before '{' token
61 | if (!a.empty() && !b.empty() && (abs(sz(a) - sz(b)) <= 5 && ask(a, b)) {
| ~ ^~
| )
icc.cpp:84:9: error: expected primary-expression before '}' token
84 | }
| ^
