#include "icc.h"
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
// #define int long long
// #define double long double
// #define endl '\n'
#define fastIO ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
#define setmin(x, y) x = min((x), (y))
#define setmax(x, y) x = max((x), (y))
#define sqr(x) ((x) * (x))
#define fi first
#define se second
#define all(x) x.begin(), x.end()
mt19937 hdp(chrono::high_resolution_clock::now().time_since_epoch().count());
int rand(int l, int r){return l + ((hdp() % (r - l + 1)) + r - l + 1) % (r - l + 1);}
const int NM = 105;
// void setRoad(int x, int y)
// {
// cout << '!' << endl;
// cout << x << ' ' << y << endl;
// }
// int query(int n, int m, int a[], int b[])
// {
// cout << '?' << endl;
// // cout << n << ' ' << m << endl;
// for (int i = 0; i < n; i++)
// cout << a[i] << ' ';
// cout << endl;
// for (int i = 0; i < m; i++)
// cout << b[i] << ' ';
// cout << endl;
// int res;
// cin >> res;
// return res;
// }
int query(vector<int>& a, vector<int>& b)
{
int A[a.size()], B[b.size()];
for (int i = 0; i < a.size(); i++)
A[i] = a[i];
for (int i = 0; i < b.size(); i++)
B[i] = b[i];
return query((int)a.size(), (int)b.size(), A, B);
}
int BS(vector<int> a, vector<int>& b)
{
shuffle(a.begin(), a.end(), hdp);
int l = 0, r = a.size();
while (l < r - 1)
{
int m = l + r >> 1;
if (query(vector<int>(a.begin(), a.begin() + m), b))
r = m;
else
l = m;
}
return a[r - 1];
}
int id[NM], p[NM];
vector<int> v[NM];
set<int> s, did;
vector<int> a, b;
int find(int x)
{
return p[x] == x ? x : find(p[x]);
}
void merge(int x, int y)
{
x = find(x), y = find(y);
if (v[x].size() < v[y].size())
swap(x, y);
p[y] = x;
s.erase(y);
for (int t : v[y])
v[x].push_back(t);
}
void run(int N)
{
iota(p + 1, p + 1 + N, 1);
for (int i = 1; i <= N; i++)
{
v[i].push_back(i);
s.insert(i);
}
for (int n = N; n >= 2; n--)
{
auto it = s.begin();
for (int i = 1; i <= n; i++, it = next(it))
id[i] = *it;
shuffle(id + 1, id + 1 + n, hdp);
for (int i = 0; i < 7; i++)
{
a.clear(); b.clear();
for (int j = 0; j < n; j++)
for (auto t : v[id[j + 1]])
{
if (j >> i & 1)
a.push_back(t);
else
b.push_back(t);
}
if (a.size() && b.size() && query(a, b))
break;
}
int x = BS(a, b), y = BS(b, a);
setRoad(x, y);
merge(x, y);
}
}
// int main()
// {
// int N;
// cin >> N;
// run(N);
// }
Compilation message
icc.cpp: In function 'int query(std::vector<int>&, std::vector<int>&)':
icc.cpp:48:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
48 | for (int i = 0; i < a.size(); i++)
| ~~^~~~~~~~~~
icc.cpp:50:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
50 | for (int i = 0; i < b.size(); i++)
| ~~^~~~~~~~~~
icc.cpp: In function 'int BS(std::vector<int>, std::vector<int>&)':
icc.cpp:61:19: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
61 | int m = l + r >> 1;
| ~~^~~
icc.cpp:62:19: error: cannot bind non-const lvalue reference of type 'std::vector<int>&' to an rvalue of type 'std::vector<int>'
62 | if (query(vector<int>(a.begin(), a.begin() + m), b))
| ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
icc.cpp:45:24: note: initializing argument 1 of 'int query(std::vector<int>&, std::vector<int>&)'
45 | int query(vector<int>& a, vector<int>& b)
| ~~~~~~~~~~~~~^
