#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define pb emplace_back
#define AI(i) begin(i), end(i)
template<class T> bool chmin(T &a, T b) { return b < a && (a = b, true); }
template<class T> bool chmax(T &a, T b) { return a < b && (a = b, true); }
#ifdef KEV
#define DE(args...) kout("[ " + string(#args) + " ] = ", args)
void kout() { cerr << endl; }
template<class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ', kout(b...); }
template<class T> void debug(T l, T r) { while (l != r) cerr << *l << " \n"[next(l)==r], ++l; }
#else
#define DE(...) 0
#define debug(...) 0
#endif
#include "chameleon.h"
#include <vector>
namespace {
const int MAX_N = 1010;
int qcnt;
int qry(vector<int> p) { for (int &i : p) ++i; ++qcnt; return Query(p); }
int n, love[MAX_N];
vector<int> edge[MAX_N];
void ans(int a, int b) { ++a, ++b; if (a > b) return; Answer(a, b); }
void add_edge(int a, int b) {
edge[a].pb(b), edge[b].pb(a);
}
void find(int x, vector<int> g) {
auto gen_vec = [&](int id) {
vector<int> ret(begin(g), begin(g)+id+1);
ret.pb(x);
return ret;
};
for (int i = 0;i < 3;++i) {
if (i == 1) {
auto vec = gen_vec(g.size()-1);
if (qry(vec) == vec.size()) return;
}
int l = 0, r = g.size()-1, mid;
while (l < r) {
mid = l + r >> 1;
auto vec = gen_vec(mid);
if (qry(vec) == vec.size())
l = mid+1;
else
r = mid;
}
int y = g[l];
add_edge(x, y);
g.erase(begin(g)+l);
}
// if (g.empty()) return;
// vector<int> q = g;
// q.pb(x);
// if (qry(q) == q.size()) return;
//
// if (g.size() == 1) {
// edge[x].pb(g[0]), edge[g[0]].pb(x);
// return;
// }
//
// int mid = g.size()/2;
// vector<int> l(begin(g), begin(g)+mid), r(begin(g)+mid, end(g));
// DE(l.size(), r.size());
// find(x, l), find(x, r);
}
void make_edge() {
vector<int> one, two;
for (int i = 0;i < n;++i) one.pb(i), two.pb(i + n);
// for (int i = 0;i < n + n;++i) {
// one.pb(i);
// if (qry(one) != one.size()) {
// one.pop_back();
// two.pb(i);
// }
// }
assert(one.size() == n && two.size() == n);
for (int i = 0;i < n;++i) {
int oc = qcnt;
find(one[i], two);
int use = qcnt - oc;
DE(i, use, edge[ one[i] ].size());
debug(AI( edge[ one[i] ]));
}
}
}
void Solve(int N) {
n = N;
fill(love, love + n + n, -1);
make_edge();
for (int i = 0;i < n + n;++i) {
assert(edge[i].size() == 1 || edge[i].size() == 3);
if (edge[i].size() == 1) continue;
int cnt = 0;
for (int j = 0;j < 3;++j) {
vector<int> p {i};
for (int k = 0;k < 3;++k) if (j != k) p.pb( edge[i][k] );
if (qry(p) == 1) {
love[ i ] = edge[i][j];
++cnt;
}
}
assert(cnt == 1);
}
for (int i = 0;i < n + n;++i) {
for (int j = 0;j < edge[i].size();++j) {
if (love[i] != edge[i][j] && love[ edge[i][j] ] != i)
ans(i, edge[i][j]);
}
}
}
Compilation message
chameleon.cpp: In function 'void {anonymous}::find(int, std::vector<int>)':
chameleon.cpp:45:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
45 | if (qry(vec) == vec.size()) return;
| ~~~~~~~~~^~~~~~~~~~~~~
chameleon.cpp:50:12: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
50 | mid = l + r >> 1;
| ~~^~~
chameleon.cpp:52:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
52 | if (qry(vec) == vec.size())
| ~~~~~~~~~^~~~~~~~~~~~~
In file included from /usr/include/c++/10/cassert:44,
from /usr/include/x86_64-linux-gnu/c++/10/bits/stdc++.h:33,
from chameleon.cpp:1:
chameleon.cpp: In function 'void {anonymous}::make_edge()':
chameleon.cpp:90:20: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
90 | assert(one.size() == n && two.size() == n);
| ~~~~~~~~~~~^~~~
chameleon.cpp:90:39: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
90 | assert(one.size() == n && two.size() == n);
| ~~~~~~~~~~~^~~~
chameleon.cpp:14:17: warning: statement has no effect [-Wunused-value]
14 | #define DE(...) 0
| ^
chameleon.cpp:96:3: note: in expansion of macro 'DE'
96 | DE(i, use, edge[ one[i] ].size());
| ^~
chameleon.cpp:15:20: warning: statement has no effect [-Wunused-value]
15 | #define debug(...) 0
| ^
chameleon.cpp:97:3: note: in expansion of macro 'debug'
97 | debug(AI( edge[ one[i] ]));
| ^~~~~
chameleon.cpp:95:7: warning: unused variable 'use' [-Wunused-variable]
95 | int use = qcnt - oc;
| ^~~
chameleon.cpp: In function 'void Solve(int)':
chameleon.cpp:128:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
128 | for (int j = 0;j < edge[i].size();++j) {
| ~~^~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
328 KB |
Output is correct |
| 2 |
Correct |
0 ms |
328 KB |
Output is correct |
| 3 |
Runtime error |
31 ms |
620 KB |
Execution killed with signal 6 |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
328 KB |
Output is correct |
| 2 |
Runtime error |
1 ms |
456 KB |
Execution killed with signal 6 |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
328 KB |
Output is correct |
| 2 |
Runtime error |
1 ms |
456 KB |
Execution killed with signal 6 |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
328 KB |
Output is correct |
| 2 |
Correct |
1 ms |
328 KB |
Output is correct |
| 3 |
Correct |
58 ms |
392 KB |
Output is correct |
| 4 |
Correct |
68 ms |
392 KB |
Output is correct |
| 5 |
Correct |
57 ms |
392 KB |
Output is correct |
| 6 |
Correct |
56 ms |
400 KB |
Output is correct |
| 7 |
Correct |
63 ms |
392 KB |
Output is correct |
| 8 |
Correct |
69 ms |
392 KB |
Output is correct |
| 9 |
Correct |
66 ms |
328 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
0 ms |
328 KB |
Output is correct |
| 2 |
Correct |
0 ms |
328 KB |
Output is correct |
| 3 |
Runtime error |
31 ms |
620 KB |
Execution killed with signal 6 |
| 4 |
Halted |
0 ms |
0 KB |
- |
