#include <bits/stdc++.h>
using namespace std;
bool in[513] = {0, 1, 0, 0, 0, 0};
#ifdef TEST
int query(vector<int> islands){
cout << "Query: ";
for(auto i : islands){
cout << i << ' ';
}
bool y = false;
for(auto i : islands){
y |= in[i];
}
cout << (y ? "YES" : "NO") << endl;
return y;
}
#else
int query(vector<int> islands);
#endif
vector<int> G[513];
int c = 0, tot = 512;
int u[513], elim[513];
void dfs(int t, int v, int par = -1){
if(c == tot) return;
if(u[v] == 0) c++, u[v] = t;
for(auto i : G[v]){
if(i == par || elim[i])
continue;
dfs(t, i, v);
}
}
int findEgg(int N, vector<pair<int, int>> bridges){
if(N == 1) return 1;
for(int i = 1; i <= N; i++){
G[i].clear();
}
for(int i = 0; i < N - 1; i++){
G[bridges[i].first].push_back(bridges[i].second);
G[bridges[i].second].push_back(bridges[i].first);
}
for(int i = (N + 1) / 2, t = 1; i != 1; t++){
c = 0; tot = i; dfs(t, 1);
vector<int> islands;
for(int j = 1; j <= N; j++){
if(u[j]) islands.push_back(j);
}
int r = query(islands);
if(!r){
i = i - (i + 1) / 2;
if(i == 1){
for(int j = 1; j <= N; j++){
if(u[j] == 0) return j;
}
}
}else{
i = (i + 1) / 2;
for(int j = 1; j <= N; j++){
if(u[j] == 0) elim[j] = true;
if(u[j] == t) u[j] = 0;
}
if(i == 1){
for(int j = 1; j <= N; j++){
if(u[j] == 0) return j;
}
}
}
}
}
#ifdef TEST
int main(){
cout << findEgg(2, {{1, 2}}) << endl;
}
#endif
Compilation message
eastereggs.cpp: In function 'int findEgg(int, std::vector<std::pair<int, int> >)':
eastereggs.cpp:74:1: warning: control reaches end of non-void function [-Wreturn-type]
74 | }
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
1 ms |
464 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
1 ms |
464 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
1 ms |
592 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |