#include "swap.h"
#include<bits/stdc++.h>
using namespace std;
struct yal{
int u,v,w,vas;
bool operator <(yal b){
return w<b.w;
}
int getad(int fu){
return (fu^u^v);
}
};
const int maxn=100000+10,lg=17;
int n,m,wlca[lg][maxn],inf=1e9+5,mainres[maxn];
vector<yal>alle;
vector<int>adjt[maxn];
struct dsu{
int par[maxn],val[maxn];
vector<int>allv[maxn];
dsu(){
for(int i=0;i<maxn;i++){
par[i]=i;
val[i]=inf;
allv[i].push_back(i);
}
};
int root(int u){
if(u==par[u]){
return u;
}
return par[u]=root(par[u]);
}
int con(int u,int v,int w){
int rootu=root(u),rootv=root(v);
if(rootu!=rootv){
par[rootv]=rootu;
if((int)allv[rootu].size()<(int)allv[rootv].size()){
swap(allv[rootu],allv[rootv]);
}
for(auto x:allv[rootv]){
allv[rootu].push_back(x);
}
if(val[rootu]<inf||val[rootv]<inf){
val[rootu]=w;
}
if(val[rootu]<inf){
for(auto x:allv[rootu]){
mainres[x]=min(mainres[x],val[rootu]);
}
allv[rootu].clear();
}
return 1;
}else{
return 0;
}
}
void tagh(int u,int w){
u=root(u);
for(auto x:allv[u]){
mainres[x]=min(mainres[x],w);
}
allv[u].clear();
val[u]=min(w,val[u]);
}
}ds;
int av=inf;
void init(int N, int M,std::vector<int> U, std::vector<int> V, std::vector<int> W) {
n=N;
m=M;
alle.resize(m);
for(int i=0;i<maxn;i++){
mainres[i]=inf;
}
for(int i=0;i<m;i++){
alle[i].u=U[i];
alle[i].v=V[i];
alle[i].w=W[i];
}
sort(alle.begin(),alle.end());
for(int i=0;i<m;i++){
if(ds.con(alle[i].u,alle[i].v)){
alle[i].vas=1;
adjt[alle[i].u].push_back(i);
adjt[alle[i].v].push_back(i);
if((int)adjt[alle[i].u].size()>2||(int)adjt[alle[i].v].size()>2){
ds.tagh(alle[i].u,alle[i].w);
}
}else{
ds.tagh(alle[i].u,alle[i].w);
}
}
}
int mx;
void dfsres(int u,int had,int par=-1,int len=0){
if(had==u){
mx=len;
return ;
}
for(auto x:adjt[u]){
int v=alle[x].getad(u);
if(v==par){
continue;
}
dfsres(v,had,u,max(len,alle[x].w));
}
}
int getMinimumFuelCapacity(int X, int Y) {
mx=inf;
dfsres(X,Y);
long long ret=max(mx,min(mainres[X],mainres[Y]));
if(ret>=inf){
return -1;
}
return ret;
}
Compilation message
swap.cpp: In function 'void init(int, int, std::vector<int>, std::vector<int>, std::vector<int>)':
swap.cpp:83:32: error: no matching function for call to 'dsu::con(int&, int&)'
83 | if(ds.con(alle[i].u,alle[i].v)){
| ^
swap.cpp:34:6: note: candidate: 'int dsu::con(int, int, int)'
34 | int con(int u,int v,int w){
| ^~~
swap.cpp:34:6: note: candidate expects 3 arguments, 2 provided
