# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
436523 | PiejanVDC | Keys (IOI21_keys) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "keys.h"
#include <bits/stdc++.h>
using namespace std;
vector<int> find_reachable(vector<int> r, vector<int> u, vector<int> v, vector<int> c) {
int n = r.size(), m = u.size();
vector<int>adj[n];
map<pair<int,int>,bool>mp;
for(int i = 0 ; i < m ; i++) {
if(mp[{u[i],v[i]}] || mp[{v[i],u[i]}]) continue;
adj[u[i]].push_back(v[i]);
adj[v[i]].push_back(u[i]);
mp[{u[i],v[i]}]=true;
mp[{v[i],u[i]}]=true;
}
vector<int>ans(n,0);
bool all=true;
for(int i = 0 ; i < n ; i++) {
if(r[i] != 0 || (int)adj[i].size()==0) {
all=false;
ans[i]=1;
}
}
if(all) {
vector<int>v; int mx=0;
for(int i = 0 ; i < n ; i++) {
s.push(i);
vector<bool>vis(n,false);
int cnt=0;
while(!s.empty()) {
int node = s.top();
s.pop();
cnt++;
for(auto z : adj[node]) {
if(!vis[z]) s.push(z);
}
}
if(cnt > mx) {
v.clear();
v.push_back(i);
mx=cnt;
} else if(mx == cnt) {
v.push_back(i);
}
}
for(auto z : v) {
ans[z] = 1;
}
}
return ans;
}