#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;
}
Compilation message
/usr/bin/ld: /tmp/ccvEW2Mt.o: in function `count_common_roads(std::vector<int, std::allocator<int> > const&)':
grader.cpp:(.text+0x370): multiple definition of `count_common_roads(std::vector<int, std::allocator<int> > const&)'; /tmp/ccmRxk1s.o:simurgh.cpp:(.text+0x380): first defined here
/usr/bin/ld: /tmp/ccvEW2Mt.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccmRxk1s.o:simurgh.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status
