# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
42950 | MatheusLealV | Simurgh (IOI17_simurgh) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "simurgh.h"
using namespace std;
int n, m;
int ok[N], pai[N];
vector<int> grafo[N];
int Find(int x)
{
if(x == pai[x]) return x;
return pai[x] = Find(pai[x]);
}
void join(int a, int b)
{
a = Find(a), b = Find(b);
if(a == b) return;
pai[a] = b;
}
bool check(vector<int> vet, vector<int> &u, vector<int> &v)
{
if(vet.size() != n - 1) return false;
for(int i = 0; i < n; i++) pai[i] = i;
for(auto x: vet)
{
join(u[x], v[x]);
}
set<int> ss;
for(int i = 0; i < n; i++) ss.insert(Find(i));
if(ss.size() != 1) return false;
if(count_common_roads(vet) == n - 1) return true;
else return false;
}
vector<int> find_roads(int n_, vector<int> u, vector<int> v)
{
n = n_;
m = u.size();
for(int i = 0; i < m; i++) grafo[u[i]].push_back(v[i]), grafo[v[i]].push_back(u[i]);
for(int mask = 0; mask < (1<<m); mask ++)
{
vector<int> aux;
for(int i = 0; i < m; i++)
{
if(mask & (1<<i))
{
aux.push_back(i);
}
}
if(check(aux, u, v)) return aux;
}
return u;
}