Submission #776214

#	Time	Username	Problem	Language	Result	Execution time	Memory
776214		rxlfd314	Easter Eggs (info1cup17_eastereggs)	C++17	100 / 100	16 ms	376 KiB

eastereggs

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

int findEgg(int N, vector<pair<int, int>> B) {
	vector<int> adj[N];
	for (auto &[a, b] : B) {
		adj[--a].push_back(--b);
		adj[b].push_back(a);
	}
	
	int rtin[N], timer = 0;
	function<void(int, int)> dfs = [&](int f, int pf) {
		rtin[timer++] = f;
		for (int nf : adj[f]) {
			if (nf != pf) {
				dfs(nf, f);
			}
		}
	};
	dfs(0, 0);

	int lo = 0, hi = N-1;
	while (lo < hi) {
		int mid = lo + hi >> 1;
		vector<int> vec;
		for (int i = 0; i <= mid; i++) {
			vec.push_back(rtin[i]+1);
		}
		query(vec) ? hi = mid : lo = mid + 1;
	}
	
	return rtin[lo] + 1;
}

Compilation message (stderr)

eastereggs.cpp: In function 'int findEgg(int, std::vector<std::pair<int, int> >)':
eastereggs.cpp:25:16: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   25 |   int mid = lo + hi >> 1;
      |             ~~~^~~~

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