Submission #433552

#	Time	Username	Problem	Language	Result	Execution time	Memory
433552		Kevin_Zhang_TW	Chameleon's Love (JOI20_chameleon)	C++17	20 / 100	69 ms	620 KiB

chameleon

#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 (stderr)

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) {
      |                  ~~^~~~~~~~~~~~~~~~

• Subtask 1: 0 / 4
• Subtask 2: 0 / 20
• Subtask 3: 0 / 20
• Subtask 4: 20 / 20
• Subtask 5: 0 / 36