#include "icc.h"
#include <iostream>
#include <vector>
#include <random>
#define rep(i, n) for (int i=0; i<(n); i++)
#define pb push_back
#define all(x) x.begin(), x.end()
using namespace std;
int query(vector<int> a, vector<int> b) {
//cout<<"query({";for(int x:a)cout<<x<<",";cout<<"}, {";for(int x:b)cout<<x<<",";cout<<"})\n";
int ar[100] = {}, br[100] = {};
rep(i, a.size()) ar[i] = a[i]+1;
rep(i, b.size()) br[i] = b[i]+1;
return query(a.size(), b.size(), ar, br);
}
int N;
int U[100];
vector<int> R[100];
int find(int x) {
if (U[x] == x) return x;
return U[x] = find(U[x]);
}
void unite(int x, int y) {
x = find(x), y = find(y);
if (x == y) return;
U[y] = x;
for (int t : R[y]) R[x].pb(t);
R[y].clear();
}
vector<int> roots() {
vector<int> ret;
rep(i, N) if (find(i) == i) ret.pb(i);
return ret;
}
pair<vector<int>, vector<int>> divide2() {
vector<int> vs = roots();
int n = vs.size();
while (true) {
random_shuffle(all(vs));
vector<int> a, b;
rep(i, n) {
if (i < n/2) for (int x : R[vs[i]]) a.pb(x);
else for (int x : R[vs[i]]) b.pb(x);
}
if (query(a, b)) return make_pair(a, b);
}
}
int find(vector<int> X, vector<int> other) {
if (X.size() == 1) return X[0];
vector<int> a, b;
int n = X.size();
rep(i, n) {
if (i < n/2) a.pb(X[i]);
else b.pb(X[i]);
}
if (query(a, other)) return find(a, other);
else return find(b, other);
}
void run(int n) {
N = n;
rep(i, N) U[i] = i, R[i].pb(i);
rep(_, N-1) {
vector<int> a, b;
tie(a, b) = divide2();
int u = find(a, b), v = find(b, a);
//cout<<u<<"<->"<<v<<"\n";
setRoad(u+1, v+1);
unite(u, v);
}
}
Compilation message
icc.cpp: In function 'int query(std::vector<int>, std::vector<int>)':
icc.cpp:5:34: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
#define rep(i, n) for (int i=0; i<(n); i++)
^
icc.cpp:13:3: note: in expansion of macro 'rep'
rep(i, a.size()) ar[i] = a[i]+1;
^
icc.cpp:5:34: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
#define rep(i, n) for (int i=0; i<(n); i++)
^
icc.cpp:14:3: note: in expansion of macro 'rep'
rep(i, b.size()) br[i] = b[i]+1;
^
icc.cpp: In function 'std::pair<std::vector<int>, std::vector<int> > divide2()':
icc.cpp:40:27: error: 'random_shuffle' was not declared in this scope
random_shuffle(all(vs));
^
icc.cpp: In function 'void run(int)':
icc.cpp:67:13: error: 'tie' was not declared in this scope
tie(a, b) = divide2();
^
