제출 #654384

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
654384		mousey	Easter Eggs (info1cup17_eastereggs)	C++14	0 / 100	22 ms	500 KiB

eastereggs

#include <bits/stdc++.h>
#include "grader.h"
#define ll long long
#define vll vector<ll>
#define vllp vector<pair<ll, ll> >
#define vi vector <int>
#define vip vector <pair <int, int> >
#define db double
#define ld long double
#define pdb pair <double, double> 
#define YES cout<<"YES"
#define NO cout<<"NO"
#define nl cout<<"\n"
#define vv vector <vector <ll> >
#define pll pair <ll, ll> 
#define pi pair <int, int>
#define pb push_back
#define f first
#define s second
using namespace std;
 
const ll mod=1e9+7;
const ll modx=998244353;
const double eps=1e-9;
const ll INF=INT_MAX;
const ll INFINF=LLONG_MAX;
const int M=512;
vi vc[M+5], nodes;
int sz[M+5], pa[M+5];
bool bad[M+5];
 
void dfs(int u, int par)
{
	sz[u]=1;
	for(auto i: vc[u])
	{
		if(i==par) continue;
		if(bad[i]) continue;
		pa[i]=u;
		dfs(i, u);
		sz[u]+=sz[i];
	}
}
 
int centroid_decom(int u, int size)
{
	pi p={0, 0};
	for(auto i: vc[u])
	{
		if(i==pa[u]) continue;
		if(bad[i]) continue;
		p=max(p, {sz[i], i});
	}
	if(p.f > size/2) return centroid_decom(p.s, size);
	else return u;
}
 
void find(int u, int pa)
{
	nodes.pb(u);
	for(auto i: vc[u])
	{
		if(i==pa) continue;
		find(i, u);
	}
}
 
int findEgg(int N, vip bridges)
{
	for(int i = 1; i <= N; i++) bad[i]=false, vc[i].clear();
	for(auto i: bridges)
	{
		vc[i.f].pb(i.s);
		vc[i.s].pb(i.f);
	}
	int u=1;
	while(true)
	{
		nodes.clear();
		if(N==2)
		{
			nodes.pb(u);
			int report=query(nodes);
			if(report) return u;
			else
			{
				for(auto i: vc[u])
				{
					if(!bad[i]) return i;
				}
			}
		}
		dfs(u, 0);
		pa[u]=0;
		u=centroid_decom(u, N);
		int sum=1;
		vi node;
		nodes.pb(u);
		for(auto i: vc[u])
		{
			if(i==pa[u]) continue;
			if(bad[i]) continue;
			bad[i]=true;
			sum+=sz[i];
			find(i, u);
			node.pb(i);
			if(sum>=2*N/3) break;
		}
		int report=query(nodes);
		N-=sum;
		N++;
		if(report)
		{
			for(auto i: vc[u])
			{
				if(!bad[i]) bad[i]^=1;
			}
			for(auto i: node)
			{
				bad[i]^=1;
			}
			N=sum;
		} 
//		for(auto i: nodes) cout << i << " ";
//		nl;
//		cout << N << " " << sum << " " << u << "\n" << flush;
//		for(int i = 1; i <= 5; i++) cout << bad[i] << " ";
//		nl;
//		cout << flush;
	}
}

• Subtask 1: 0 / 40
• Subtask 2: 0 / 40
• Subtask 3: 0 / 20