Submission #230621

#	Time	Username	Problem	Language	Result	Execution time	Memory
230621		wilwxk	Chameleon's Love (JOI20_chameleon)	C++14	40 / 100	74 ms	504 KiB

chameleon

#include <bits/stdc++.h>
#include "chameleon.h"
using namespace std;

const int MAXN = 1005;
const int LOGN = 9;
int color[MAXN];
vector<int> bad[MAXN], ans[MAXN];
vector<int> comp[2];
int n;

void dfs(int u) {
	for(auto v : ans[u]) {
		if(color[v] != -1) continue;
		color[v] = color[u]^1;
		dfs(v);
	}
}

void separate(int k) {
	for(int i = 1; i <= k; i++) color[i] = -1;
	for(int i = 1; i <= k; i++) {
		if(color[i] != -1) continue;
		color[i] = 0;
		dfs(i);
	}

	comp[0].clear();
	comp[1].clear();
	for(int i = 1; i <= k; i++) comp[color[i]].push_back(i);
}

void Solve(int N) {
	n = 2*N;

	comp[0].push_back(1);

	for(int u = 2; u <= n; u++) {

		separate(u-1);
		// for(int a : comp[0]) printf("%d ", a); cout << endl;
		// for(int a : comp[1]) printf("%d ", a); cout << endl;

		for(int k = 0; k < 2; k++) {

			int aa = -1, bb = comp[k].size(), cc = -1;

			vector<int> qv;
			for(int j = LOGN; j >= 0; j--) {
				int ind = aa + (1<<j);
				if(ind >= comp[k].size()) continue;
				qv.clear();
				for(int i = 0; i <= ind; i++) qv.push_back(comp[k][i]);
				qv.push_back(u);
				if(Query(qv) == qv.size()) aa = ind;
			}

			for(int j = LOGN; j >= 0; j--) {
				int ind = bb - (1<<j);
				if(ind < 0) continue;
				qv.clear();
				for(int i = ind; i < comp[k].size(); i++) qv.push_back(comp[k][i]);
				qv.push_back(u);
				if(Query(qv) == qv.size()) bb = ind;
			}


			cc = aa+1;
			for(int j = LOGN; j >= 0; j--) {
				int ind = cc + (1<<j);
				if(ind >= bb-1) continue;
				qv.clear();
				for(int i = aa+2; i <= ind; i++) qv.push_back(comp[k][i]);
				qv.push_back(u);
				if(Query(qv) == qv.size()) cc = ind;
			}

			// printf("%d: abc %d %d %d\n", u, aa, bb, cc);

			if(aa+1 < cc+1 && cc+1 < bb-1) {
				cc = comp[k][cc+1];
				ans[cc].push_back(u);
				ans[u].push_back(cc);
				// printf("connect %d %d\n", u, cc);
			}
			if(aa+1 < comp[k].size() && aa+1 != bb-1) {
				aa = comp[k][aa+1];
				ans[aa].push_back(u);
				ans[u].push_back(aa);
				// printf("connect %d %d\n", u, aa);
			}
			if(bb-1 >= 0) {
				bb = comp[k][bb-1];
				ans[bb].push_back(u);
				ans[u].push_back(bb);
				// printf("connect %d %d\n", u, bb);
			}
		}
	}

	for(int u = 1; u <= n; u++) {
		if(ans[u].size() == 1) continue;
		else {
			if(Query({u, ans[u][0], ans[u][1]}) == 1) {
				bad[ans[u][2]].push_back(u);
				bad[u].push_back(ans[u][2]);
			}
			else if(Query({u, ans[u][1], ans[u][2]}) == 1) {
				bad[ans[u][0]].push_back(u);
				bad[u].push_back(ans[u][0]);
			}
			else {
				bad[ans[u][1]].push_back(u);
				bad[u].push_back(ans[u][1]);
			}
		}
	}

	for(int u = 1; u <= n; u++) {
		if(color[u] == 2) continue;

		int x;
		if(ans[u].size() == 1) {
			x = ans[u][0];
		}
		else {
			for(int k = 0; k < 3; k++) {
				if(ans[u][k] != bad[u][0] && ans[u][k] != bad[u][1]) {
					x = ans[u][k];
				}
			}
		}
		
		Answer(u, x);
		color[u] = color[x] = 2;
	}
}

Compilation message (stderr)

chameleon.cpp: In function 'void Solve(int)':
chameleon.cpp:51:12: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(ind >= comp[k].size()) continue;
        ~~~~^~~~~~~~~~~~~~~~~
chameleon.cpp:55:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(Query(qv) == qv.size()) aa = ind;
        ~~~~~~~~~~^~~~~~~~~~~~
chameleon.cpp:62:24: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = ind; i < comp[k].size(); i++) qv.push_back(comp[k][i]);
                      ~~^~~~~~~~~~~~~~~~
chameleon.cpp:64:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(Query(qv) == qv.size()) bb = ind;
        ~~~~~~~~~~^~~~~~~~~~~~
chameleon.cpp:75:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(Query(qv) == qv.size()) cc = ind;
        ~~~~~~~~~~^~~~~~~~~~~~
chameleon.cpp:86:12: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    if(aa+1 < comp[k].size() && aa+1 != bb-1) {
       ~~~~~^~~~~~~~~~~~~~~~
chameleon.cpp:135:23: warning: 'x' may be used uninitialized in this function [-Wmaybe-uninitialized]
   color[u] = color[x] = 2;
              ~~~~~~~~~^~~

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