Submission #444631

#	Time	Username	Problem	Language	Result	Execution time	Memory
444631		cheetose	Chameleon's Love (JOI20_chameleon)	C++17	100 / 100	83 ms	1472 KiB

chameleon

#include <bits/stdc++.h>
#define mp make_pair
#define pb push_back
#define X first
#define Y second
#define y0 y12
#define y1 y22
#define INF 987654321
#define PI 3.141592653589793238462643383279502884
#define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
#define fdn(i,a,b,c) for(int (i)=(a);(i)>=(b);(i)-=(c))
#define MEM0(a) memset((a),0,sizeof(a))
#define MEM_1(a) memset((a),-1,sizeof(a))
#define ALL(a) a.begin(),a.end()
#define COMPRESS(a) sort(ALL(a));a.resize(unique(ALL(a))-a.begin())
#define SYNC ios_base::sync_with_stdio(false);cin.tie(0)
using namespace std;
typedef long long ll;
typedef long double ld;
typedef double db;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int, int> Pi;
typedef pair<ll, ll> Pll;
typedef pair<ld, ld> Pd;
typedef vector<int> Vi;
typedef vector<ll> Vll;
typedef vector<ld> Vd;
typedef vector<Pi> VPi;
typedef vector<Pll> VPll;
typedef vector<Pd> VPd;
typedef tuple<int, int, int> iii;
typedef tuple<int, int, int, int> iiii;
typedef tuple<ll, ll, ll> LLL;
typedef vector<iii> Viii;
typedef vector<LLL> VLLL;
typedef complex<double> base;
const int MOD = 998244353;
ll POW(ll a, ll b, ll MMM = MOD) { ll ret = 1; for (; b; b >>= 1, a = (a*a) % MMM)if (b & 1)ret = (ret*a) % MMM; return ret; }
int dx[] = { 0,1,0,-1,1,1,-1,-1 }, dy[] = { 1,0,-1,0,1,-1,1,-1 };
int ddx[] = { -1,-2,1,-2,2,-1,2,1 }, ddy[] = { -2,-1,-2,1,-1,2,1,2 };
 
#include "chameleon.h"
Vi v[1001],tmp[2];
bool chk[1001],e[1001][1001];
void dfs(int N,int c){
	chk[N]=1;
	tmp[c].pb(N);
	for(int next:v[N]){
		if(!chk[next])dfs(next,c^1);
	}
}
void go(Vi &vv){
	int N=vv.size();
	int f=0;
	fup(kk,0,2,1){
		int l=f,r=N-2;
		while(l<=r){
			int m=l+r>>1;
			Vi tt={vv.back()};
			fup(k,f,m,1)tt.pb(vv[k]);
			if(Query(tt)!=tt.size())r=m-1;
			else l=m+1;
		}
		if(l>N-2)return;
		v[vv.back()].pb(vv[l]);
		v[vv[l]].pb(vv.back());
		f=l+1;
	}
}
void Solve(int N){
	fup(i,1,2*N,1){
		tmp[0].clear();
		tmp[1].clear();
		MEM0(chk);
		fup(j,1,i-1,1)if(!chk[j])dfs(j,0);
		fup(j,0,1,1){
			tmp[j].pb(i);
			if(tmp[j].size()>1 && Query(tmp[j])!=tmp[j].size())go(tmp[j]);
		}
	}
	MEM0(chk);
	fup(i,1,2*N,1){
		if(v[i].size()==1){
			if(chk[i] || chk[v[i][0]])continue;
			chk[i]=chk[v[i][0]]=1;
			Answer(i,v[i][0]);
			continue;
		}
		int a=v[i][0],b=v[i][1],c=v[i][2];
		int t=Query({i,a,b});
		if(t==1){
			e[i][c]=1;
			continue;
		}
		t=Query({i,a,c});
		if(t==1){
			e[i][b]=1;
			continue;
		}
		t=Query({i,c,b});
		if(t==1){
			e[i][a]=1;
			continue;
		}
	}
	fup(i,1,2*N,1){
		if(chk[i])continue;
		if(v[i].size()==3){
			chk[i]=1;
			int a=v[i][0],b=v[i][1],c=v[i][2];
			if(!e[i][a] && !e[a][i]){
				chk[a]=1;
				Answer(i,a);
				continue;
			}
			if(!e[i][b] && !e[b][i]){
				chk[b]=1;
				Answer(i,b);
				continue;
			}
			if(!e[i][c] && !e[c][i]){
				chk[c]=1;
				Answer(i,c);
				continue;
			}
		}
	}
}
/*
int main(){
}*/

Compilation message (stderr)

chameleon.cpp: In function 'void go(Vi&)':
chameleon.cpp:10:30: warning: unnecessary parentheses in declaration of 'kk' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
chameleon.cpp:56:2: note: in expansion of macro 'fup'
   56 |  fup(kk,0,2,1){
      |  ^~~
chameleon.cpp:59:11: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   59 |    int m=l+r>>1;
      |          ~^~
chameleon.cpp:10:30: warning: unnecessary parentheses in declaration of 'k' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
chameleon.cpp:61:4: note: in expansion of macro 'fup'
   61 |    fup(k,f,m,1)tt.pb(vv[k]);
      |    ^~~
chameleon.cpp:62:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |    if(Query(tt)!=tt.size())r=m-1;
      |       ~~~~~~~~~^~~~~~~~~~~
chameleon.cpp: In function 'void Solve(int)':
chameleon.cpp:10:30: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
chameleon.cpp:72:2: note: in expansion of macro 'fup'
   72 |  fup(i,1,2*N,1){
      |  ^~~
chameleon.cpp:10:30: warning: unnecessary parentheses in declaration of 'j' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
chameleon.cpp:76:3: note: in expansion of macro 'fup'
   76 |   fup(j,1,i-1,1)if(!chk[j])dfs(j,0);
      |   ^~~
chameleon.cpp:10:30: warning: unnecessary parentheses in declaration of 'j' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
chameleon.cpp:77:3: note: in expansion of macro 'fup'
   77 |   fup(j,0,1,1){
      |   ^~~
chameleon.cpp:79:39: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   79 |    if(tmp[j].size()>1 && Query(tmp[j])!=tmp[j].size())go(tmp[j]);
      |                          ~~~~~~~~~~~~~^~~~~~~~~~~~~~~
chameleon.cpp:10:30: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
chameleon.cpp:83:2: note: in expansion of macro 'fup'
   83 |  fup(i,1,2*N,1){
      |  ^~~
chameleon.cpp:10:30: warning: unnecessary parentheses in declaration of 'i' [-Wparentheses]
   10 | #define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
      |                              ^
chameleon.cpp:107:2: note: in expansion of macro 'fup'
  107 |  fup(i,1,2*N,1){
      |  ^~~

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