#include<bits/stdc++.h>
#include "icc.h"
#define I inline void
#define S struct
#define vi vector<int>
#define vii vector<pair<int,int>>
#define pii pair<int,int>
#define pll pair<ll,ll>
using namespace std ;
using ll = long long ;
using ld = long double ;
const int N = 1e5 + 7 , mod = 1e9 + 7 ;
const int inf = N ;
// How interesting!
int n ;
int fat[N] ;
vector<pair<int,int> > ed ;
int U , V ;
inline int find(int x){return fat[x] = (x == fat[x] ? x : find(fat[x])) ; }
void link(int u , int v){
u = find(u) ;
v = find(v) ;
if(u!=v)
fat[u] = v;
}
bool ok(int u , int v){
if( u == U && v == V)
return 1 ;
if(u == V && v == U)
return 1;
return 0 ;
}
/*
void setRoad(int u , int v){
if(ok(u , v))
link(u , v) ;
else{
cout<< u <<" "<< v <<">>" ;
cout<<"bad" ;
exit(0) ;
}
}
int query(int x , int y , vector<int> v1 , vector<int> v2){
set<int> st ;
for(auto u : v1)
st.insert(find(u)) ;
for(auto u : v2){
if(st.find(find(u)) != st.end())
return 1 ;
}
int m1 = 0 , m2 = 0 ;
for(auto u : v1)
if(u == U || u == V)
m1 = 1;
for(auto u : v2)
if(u == U || u == V)
m2 = 1 ;
if( m1 == 1 && m2 == 1 ){
return 1 ;
}
return 0 ;
}*/
int query1(int x , int y , vector<int> v1 , vector<int> v2){
int *v11;
int *v22 ;
v11 = new int[x] ;
v22 = new int [y] ;
for(int i = 0 ;i < x ; ++ i)
v11[i] = v1[i] ;
for(int i = 0 ;i < y ; ++ i)
v22[i] = v2[i] ;
return query(x , y , v11 , v22) ;
}
void divide(vector<vector<int> > &vec){
vector<vector<int> > ret ;
int sz = (int) vec.size() ;
for(int i = 0 ;i < sz ; ++ i){
int sz2 = (int) vec[i].size() ;
//if(sz2 < 2)continue ;
vector<int> lhs, rhs ;
for(int j = 0 ;j < sz2 ;++ j){
if(j < sz2 / 2 )
lhs.push_back(vec[i][j]) ;
else
rhs.push_back(vec[i][j]) ;
}
if(lhs.size())
ret.push_back(lhs) ;
if(rhs.size())
ret.push_back(rhs) ;
}
vec = ret ;
}
int conquer(vector<vector<int> > &vec, int lmt = 1000000){
int ret = 0 ;
vector<int> lhs, rhs ;
for(int i = 0 ; i < min(lmt , (int) vec.size() ) ; ++ i){
for(auto u : vec[i]){
if(i&1)rhs.push_back(u) ;
else lhs.push_back(u) ;
}
}
return query1(lhs.size() , rhs.size() , lhs , rhs ) ;
}
void print(vector<vector<int> > &vec){
for(auto u : vec){
for(auto v : u )
cout<< v <<" " ;
cout<<"\n" ;
}
cout<<"----------\n" ;
}
void divide_or_conquer(vector<int> hs){
vector<vector<int> > dac { hs } ;
divide(dac) ;
int k = 0 ;
while(!conquer(dac)){
divide(dac) ;
break ;
}
int lo = 1, hi = (int) dac.size() ;
int lst = (int) dac.size() ;
while(lo<=hi){
int mid = (lo+hi) >> 1;
if(conquer(dac, mid)){
lst = mid ;
hi = mid - 1;
}else lo = mid + 1;
}
vector<int> qlhs = dac[lst-2] ,qrhs = dac[lst-1] ;
int llhs = 0 , rrhs = 0 ;
while(qlhs.size() && query1(qlhs.size() , qrhs.size() , qlhs , qrhs )){
llhs = qlhs.back() ;
qlhs.pop_back() ;
}
if(llhs)qlhs.push_back(llhs) ;
while(qrhs.size() && query1(qlhs.size() , qrhs.size() , qlhs , qrhs )){
rrhs = qrhs.back() ;
qrhs.pop_back() ;
}
if(rrhs)qrhs.push_back(rrhs) ;
int a = qlhs.back() , b = qrhs.back() ;
for(int i = 1;i <= n; ++ i){
for(int j = 1;j <= n ;++ j){
if(find(i) == a && find(j) == b && query1(1 ,1 , {i} , {j})){
setRoad(i , j) ;
//cout<< i <<" " << j <<"()\n" ;
return ;
}
}
}
}
void bad_solve(int n ){
for(int r = 0 ; r < n -1 ; ++ r){
for(int i =1 ; i <= n; ++ i){
for(int j = i + 1; j<= n; ++ j){
if(find(i) != find(j) && query(1 ,1 , {i} , {j})){
link(i , j) ;
setRoad(i ,j ) ;
// cout<< i <<" "<< j <<"\n" ;
i = j = n ;
}
}
}
}
}
void run(int n){
for(int i = 1;i <= n; ++ i)
fat[i] = i ;
bad_solve(n) ;
return ;
for(int r = 0 ; r < n - 1; ++ r){
U = ed[r].first ;
V = ed[r].second ;
vector<int> v ;
for(int i = 1;i <= n; ++ i){
if(i == fat[i])
v.push_back(i) ;
}
divide_or_conquer(v) ;
}
}
Compilation message
icc.cpp: In function 'int conquer(std::vector<std::vector<int> >&, int)':
icc.cpp:107:13: warning: unused variable 'ret' [-Wunused-variable]
107 | int ret = 0 ;
| ^~~
icc.cpp: In function 'void divide_or_conquer(std::vector<int>)':
icc.cpp:131:13: warning: unused variable 'k' [-Wunused-variable]
131 | int k = 0 ;
| ^
icc.cpp: In function 'void bad_solve(int)':
icc.cpp:180:80: error: invalid conversion from 'int' to 'int*' [-fpermissive]
180 | if(find(i) != find(j) && query(1 ,1 , {i} , {j})){
| ^
| |
| int
In file included from icc.cpp:2:
icc.h:10:30: note: initializing argument 3 of 'int query(int, int, int*, int*)'
10 | int query(int a, int b, int *A, int *B);
| ~~~~~^
icc.cpp:180:80: error: invalid conversion from 'int' to 'int*' [-fpermissive]
180 | if(find(i) != find(j) && query(1 ,1 , {i} , {j})){
| ^
| |
| int
In file included from icc.cpp:2:
icc.h:10:38: note: initializing argument 4 of 'int query(int, int, int*, int*)'
10 | int query(int a, int b, int *A, int *B);
| ~~~~~^
