Submission #245573

#	Time	Username	Problem	Language	Result	Execution time	Memory
245573		TadijaSebez	Chameleon's Love (JOI20_chameleon)	C++14	100 / 100	56 ms	504 KiB

chameleon

#include "chameleon.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define pii pair<int,int>

void Solve(int n) {
	vector<int> groups[4];
	vector<bool> was(n*2+1,false);
	vector<vector<int>> cand(n*2+1,vector<int>());
	int sz=0;
	for(int i=1;i<=n*2;i++){
		bool ok=false;int z;
		for(z=0;z<sz&&cand[i].size()<3;z++){
			function<bool(vector<int>)> work=[&](vector<int> tmp){
				tmp.pb(i);
				if(Query(tmp)==tmp.size())return false;
				tmp.pop_back();
				vector<int> mem=tmp;
				while(tmp.size()>1){
					vector<int> v[2];
					for(int j=0;j<tmp.size();j++)v[j&1].pb(tmp[j]);
					v[0].pb(i);
					int B=Query(v[0]);
					v[0].pop_back();
					if(B<=v[0].size())tmp=v[0];
					else tmp=v[1];
				}
				cand[i].pb(tmp[0]);
				cand[tmp[0]].pb(i);
				vector<int> nxt;
				for(int j:mem)if(j!=tmp[0])nxt.pb(j);
				work(nxt);
				return true;
			};
			bool now=work(groups[z]);
			if(!now&&!ok){
				groups[z].pb(i);
				ok=true;
			}
		}
		if(!ok&&z==sz)groups[sz++].pb(i);
		else if(!ok)groups[z].pb(i);
	}
	set<pii> bad;
	for(int j=1;j<=n*2;j++){
		assert(cand[j].size()==1||cand[j].size()==3);
		if(cand[j].size()==3){
			int A=Query({j,cand[j][0],cand[j][1]});
			int B=Query({j,cand[j][1],cand[j][2]});
			int C=Query({j,cand[j][2],cand[j][0]});
			if(A==1)bad.insert({j,cand[j][2]});
			if(B==1)bad.insert({j,cand[j][0]});
			if(C==1)bad.insert({j,cand[j][1]});
		}
	}
	for(int j=1;j<=n*2;j++){
		if(was[j])continue;
		for(int i:cand[j]){
			if(!bad.count({i,j})&&!bad.count({j,i})){
				Answer(i,j);
				was[i]=was[j]=true;
				break;
			}
		}
	}
}

Compilation message (stderr)

chameleon.cpp: In lambda function:
chameleon.cpp:17:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(Query(tmp)==tmp.size())return false;
        ~~~~~~~~~~^~~~~~~~~~~~
chameleon.cpp:22:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
      for(int j=0;j<tmp.size();j++)v[j&1].pb(tmp[j]);
                  ~^~~~~~~~~~~
chameleon.cpp:26:10: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
      if(B<=v[0].size())tmp=v[0];
         ~^~~~~~~~~~~~~

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