| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 992504 | MuntherCarrot | Simurgh (IOI17_simurgh) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
// #include "simurgh.h"
using namespace std;
#define ll long long
#define all(x) x.begin(), x.end()
struct dsu{
vector<int> p, sz;
dsu(int n){
p.resize(n);
iota(all(p), 0);
sz.resize(n, 1);
}
int find(int x){
return x == p[x] ? x : p[x] = find(p[x]);
}
void merge(int a, int b){
a = find(a);
b = find(b);
p[b] = a;
sz[a] += sz[b];
}
int size(){
return *max_element(all(sz));
}
};
int count_common_roads(const std::vector<int>& r){
for(int i : r) cout << i << ' ';
cout << endl;
int x; cin >> x; return x;
}
vector<int> find_roads(int n, vector<int> u, vector<int> v){
int m = u.size();
vector<int> vec(m);
iota(all(vec), 0);
while(true){
for(int i = 0; i < m; i++){
swap(vec[i], vec[rand() % m]);
}
dsu A(n);
vector<int> V;
for(int i : vec){
if(A.find(v[i]) == A.find(u[i])) continue;
A.merge(v[i], u[i]);
V.push_back(i);
}
sort(all(V));
if(count_common_roads(V) == n - 1){
return V;
}
}
}
int main(){
ios_base::sync_with_stdio(0); cin.tie(0), cout.tie(0);
find_roads(4, {0, 0, 0, 1, 1, 2}, {1, 2, 3, 2, 3, 3});
return 0;
}