#include"icc.h"
#include<bits/stdc++.h>
using namespace std;
mt19937 rng(69420);
#define f first
#define s second
typedef long long ll;
typedef long double ld;
const int maxn = 105;
const int mod = 998244353;
int t, n;
map<int, int> mp;
/*int query(int sza, int szb, int a[], int b[]){
cout << sza << endl;
for(int i = 0; i < sza; i++)cout << a[i] << " ";
cout << endl;
cout << szb << endl;
for(int i = 0; i < szb; i++)cout << b[i] << " ";
cout << endl;
int x;
cin >> x;
return x;
}
void setRoad(int a, int b){
cout << "FOUND " << a << " " << b << endl;
}*/
int up[maxn];
int Find(int x){
if(!up[x])return x;
else return up[x] = Find(up[x]);
}
void Union(int a, int b){
a = Find(a);
b = Find(b);
up[a] = b;
}
int arr1[maxn], arr2[maxn];
int gt(vector<int> a, vector<int> b){
//cout << a.size() << " " << b.size() << "\n";
if(a.empty() || b.empty())return 0;
for(int i = 0; i < a.size(); i++)arr1[i] = a[i];
for(int i = 0; i < b.size(); i++)arr2[i] = b[i];
return query(a.size(), b.size(), arr1, arr2);
}
int bs(vector<int> a, vector<int> b){
int l = 0, r = a.size()-1;
shuffle(a.begin(), a.end(), rng);
shuffle(b.begin(), b.end(), rng);
while(l < r){
int mid = (l+r+1)>>1;
if(gt(vector<int>(a.begin(), a.begin()+mid), b))r = mid-1;
else l = mid;
}
return a[l];
}
void findedge(vector<int> a, vector<int> b){
assert(!a.empty() && !b.empty());
int h1 = bs(a, b), h2 = bs(b, a);
Union(h1, h2);
setRoad(h1, h2);
}
void run(int N){
n = N;
for(int i = 0; i < n-1; i++){
mp.clear();
int crr = 0;
vector<int> ord;
for(int i = 1; i <= n; i++)ord.push_back(i);
shuffle(ord.begin(), ord.end(), rng);
for(auto i: ord)if(mp.find(Find(i)) == mp.end())mp[Find(i)] = crr++;
ord.clear();
int tot = 0;
while((1<<tot) < crr-1)tot++;
for(int j = 0; j <= tot; j++){
//cout << i << " " << j << endl;
vector<int> v[2];
for(int k = 1; k <= n; k++)v[(mp[Find(k)]>>j)&1].push_back(k);
//cout << v[0].size() << " " << v[1].size() << endl;
if((j == tot) || gt(v[0], v[1])){
findedge(v[0], v[1]);
break;
}
}
}
}
Compilation message
icc.cpp: In function 'int gt(std::vector<int>, std::vector<int>)':
icc.cpp:54:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
54 | for(int i = 0; i < a.size(); i++)arr1[i] = a[i];
| ~~^~~~~~~~~~
icc.cpp:55:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
55 | for(int i = 0; i < b.size(); i++)arr2[i] = b[i];
| ~~^~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
7 ms |
468 KB |
Ok! 92 queries used. |
2 |
Correct |
7 ms |
468 KB |
Ok! 98 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
30 ms |
468 KB |
Ok! 544 queries used. |
2 |
Correct |
34 ms |
468 KB |
Ok! 657 queries used. |
3 |
Correct |
33 ms |
468 KB |
Ok! 652 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
109 ms |
496 KB |
Ok! 1378 queries used. |
2 |
Correct |
138 ms |
512 KB |
Ok! 1616 queries used. |
3 |
Correct |
114 ms |
500 KB |
Ok! 1478 queries used. |
4 |
Correct |
120 ms |
504 KB |
Ok! 1513 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
137 ms |
504 KB |
Ok! 1510 queries used. |
2 |
Correct |
142 ms |
500 KB |
Ok! 1491 queries used. |
3 |
Correct |
126 ms |
500 KB |
Ok! 1603 queries used. |
4 |
Correct |
110 ms |
512 KB |
Ok! 1453 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
119 ms |
508 KB |
Ok! 1590 queries used. |
2 |
Correct |
136 ms |
512 KB |
Ok! 1614 queries used. |
3 |
Correct |
143 ms |
508 KB |
Ok! 1622 queries used. |
4 |
Correct |
119 ms |
508 KB |
Ok! 1601 queries used. |
5 |
Correct |
110 ms |
508 KB |
Ok! 1505 queries used. |
6 |
Correct |
125 ms |
756 KB |
Ok! 1547 queries used. |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
124 ms |
512 KB |
Ok! 1615 queries used. |
2 |
Correct |
121 ms |
504 KB |
Ok! 1613 queries used. |