oj.uz

답안 #384588

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
384588 2021-04-01T21:16:25 Z BlancaHM Easter Eggs (info1cup17_eastereggs) C++14
0 / 100
 3 ms 620 KB 
#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
#include <queue>
#include "grader.h"
using namespace std;

void reset(set<int> & islandsLeft, vector <pair<int,int>> & bridges, vector<vector<int>> & adj) {
	int N = (int) bridges.size()+1;
	islandsLeft = set<int>();
	for (int i = 0; i < N; i++) {
		islandsLeft.insert(i);
		if (i) {
			adj[bridges[i - 1].first-1].push_back(bridges[i - 1].second-1);
			adj[bridges[i - 1].second-1].push_back(bridges[i - 1].first-1);
		}
	}
}

vector<int> process(set<int> & curQ, set<int> & islandsLeft) {
	vector<int> queryVec = vector<int>(), nextIs = vector<int>();
	for (auto it = curQ.begin(); it != curQ.end(); it++)
		queryVec.push_back(*it);
	if (query(queryVec)) {
		// islands = intersection of islands and curQ
		for (int is : islandsLeft) {
			if (curQ.find(is) != curQ.end())
				nextIs.push_back(is);
		}
	} else {
		// islands = intersection of islands and NOT curQ
		for (int is : islandsLeft) {
			if (curQ.find(is) == curQ.end())
				nextIs.push_back(is);
		}
	}
	return nextIs;
}

int findEgg(int N, vector <pair<int,int>> bridges) {
	vector<vector<int>> adj(N, vector<int>());
	set<int> islandsLeft, curQ;
	reset(islandsLeft, bridges, adj);
	queue<int> q;
	while ((int) islandsLeft.size() > 1) {
		// hacemos un árbol que contenga la primera mitad de islas restantes y ninguna de la segunda mitad
		curQ = set<int>();
		curQ.insert(*islandsLeft.begin());
		q = queue<int>();
		q.push(*islandsLeft.begin());
		int islands = 1;
		while(q.size()) {
			if (islands*2 >= (int) islandsLeft.size())
				break;
			int curIs = q.front();
			q.pop();
			for (int v: adj[curIs]) {
				if (islands*2 >= (int) islandsLeft.size()) break;
				if (curQ.find(v) == curQ.end()) {
					curQ.insert(v);
					q.push(v);
					if (islandsLeft.find(v) != islandsLeft.end())
						islands++;
				}
				if (islands*2 >= (int) islandsLeft.size()) break;
			}
		}
		// process islandsLeft
		vector<int> nextIs = process(curQ, islandsLeft);
		islandsLeft.clear();
		for (int is: nextIs)
			islandsLeft.insert(is);
	}
	return *islandsLeft.begin() + 1;
}

Subtask #1
0 / 40.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 2 ms 620 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #2
0 / 40.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 2 ms 544 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -

Subtask #3
0 / 20.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 3 ms 620 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -