#include "lokahia.h"
#include <bits/stdc++.h>
using namespace std;
vector<int> thing[205];
int qLim = 400;
int p[205];
bool vis[205];
int root(int a){
return p[a]==a?a:p[a]=root(p[a]);
}
priority_queue<pair<int,int>,vector<pair<int,int> >,greater<pair<int,int> > > pq;
int sz[205];
int queryCount = 0;
int query(int a, int b){
if (queryCount>=qLim){
return -2;
}
if (a==b) return a;
int res = CollectRelics(a,b);
queryCount++;
if (res!=-1){
thing[res].push_back(a);
thing[res].push_back(b);
//printf("add edge %d %d, %d %d\n",res,a,res,b);
}
return res;
}
int dfs(int node){
vis[node] = true;
int res = 1;
for (auto x : thing[node]){
if (!vis[x]){
res += dfs(x);
}
}
return res;
}
int FindBase(int N){
if (N==1) return 0;
for (int x = 0; x<N; x++){
pq.push({1,x});
p[x] = x;
}
int mn = -1;
while (!pq.empty()){
if (pq.size()<3){
if (pq.size()==1) {mn = pq.top().second;pq.pop();}
else{
auto a = pq.top();
pq.pop();
auto b = pq.top();
pq.pop();
int r = query(root(a.second),root(b.second));
if (r==-2) return -1;
if (r==-1){
mn = sz[a.second]>sz[b.second]?a.second:b.second;
}
}
continue;
}
auto a = pq.top();
pq.pop();
auto b = pq.top();
pq.pop();
auto c = pq.top();
pq.pop();
int res = query(root(a.second),root(b.second));
if (res==-2) return -1;
if (res==-1){
int r2 = query(root(a.second),root(c.second));
if (r2==-2) return -1;
if (r2==-1) {
int r3 = query(root(b.second),root(c.second));
if (r3==-2) return -1;
if (r3!=-1){
if (root(b.second)!=root(r3)){
sz[root(r3)] += sz[root(b.second)];
p[root(b.second)] = r3;
}
if (root(c.second)!=root(r3)){
sz[root(r3)] += sz[root(c.second)];
p[root(c.second)] = r3;
}
pq.push({sz[root(r3)],r3});
}
else{
if (pq.empty()){
mn = sz[a.second]>sz[b.second]?a.second:b.second;
mn = sz[mn]>sz[c.second]?mn:c.second;
}
}
}
else{
if (root(a.second)!=root(r2)){
sz[root(r2)] += sz[root(a.second)];
p[root(a.second)] = r2;
}
if (root(c.second)!=root(r2)){
sz[root(r2)] += sz[root(c.second)];
p[root(c.second)] = r2;
}
pq.push({sz[root(r2)],r2});
}
}
else{
if (root(b.second)!=root(res)){
sz[root(res)] += sz[root(b.second)];
p[root(b.second)] = res;
}
if (root(a.second)!=root(res)){
sz[root(res)] += sz[root(a.second)];
p[root(a.second)] = res;
}
pq.push({sz[root(res)],res});
}
}
for (int x = 0; x<N; x++){
if (root(x)!=root(mn)){
query(x,root(mn));
}
}
for (int x = 0; x<N; x++){
for (int y = 0; y<N; y++){
vis[y] = false;
}
if (dfs(x)>N/2){
return x;
}
}
return -1;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
5 ms |
640 KB |
Correct : C = 181 |
2 |
Correct |
5 ms |
640 KB |
Correct : C = 225 |
3 |
Incorrect |
6 ms |
640 KB |
Wrong |
4 |
Partially correct |
5 ms |
640 KB |
Partially correct : C = 395 |
5 |
Partially correct |
6 ms |
640 KB |
Partially correct : C = 370 |
6 |
Correct |
5 ms |
512 KB |
Correct : C = 4 |
7 |
Correct |
5 ms |
640 KB |
Correct : C = 299 |
8 |
Correct |
5 ms |
512 KB |
Correct : C = 119 |
9 |
Partially correct |
5 ms |
640 KB |
Partially correct : C = 398 |
10 |
Correct |
6 ms |
640 KB |
Correct : C = 199 |
11 |
Correct |
5 ms |
640 KB |
Correct : C = 178 |
12 |
Correct |
5 ms |
640 KB |
Correct : C = 296 |
13 |
Partially correct |
6 ms |
640 KB |
Partially correct : C = 384 |
14 |
Incorrect |
6 ms |
640 KB |
Wrong |
15 |
Correct |
5 ms |
640 KB |
Correct : C = 197 |
16 |
Correct |
5 ms |
640 KB |
Correct : C = 239 |
17 |
Partially correct |
5 ms |
640 KB |
Partially correct : C = 400 |
18 |
Correct |
5 ms |
640 KB |
Correct : C = 270 |
19 |
Partially correct |
5 ms |
640 KB |
Partially correct : C = 303 |
20 |
Correct |
5 ms |
640 KB |
Correct : C = 209 |
21 |
Partially correct |
5 ms |
640 KB |
Partially correct : C = 396 |
22 |
Partially correct |
5 ms |
640 KB |
Partially correct : C = 350 |
23 |
Correct |
5 ms |
640 KB |
Correct : C = 218 |
24 |
Correct |
5 ms |
640 KB |
Correct : C = 117 |
25 |
Correct |
5 ms |
640 KB |
Correct : C = 218 |
26 |
Correct |
6 ms |
560 KB |
Correct : C = 236 |
27 |
Correct |
5 ms |
512 KB |
Correct : C = 0 |