题意:
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/4285d9c8872f04b85d635e4e5b305683.png)
解法:
将边分成<=k和>k两类,
因为>k的边加入肯定要花费代价减少边权,所以先放着,
将<=k的边用并查集合并.
如果已经联通,那么为了达到边权k,有两种做法:
1.将已经加入的最大的边权增加到k.
2.选择一条>k的边替换进来,将边权减少到k.
如果不联通,那么必须添加>k的边权,
按照MST的做法,对边排序,
设边权为val,那么将每条边的代价视为val-k,
继续做最小生成树即可.
code:
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
struct Node{
int a,b,c;
}e[maxm];
int pre[maxm];
int n,m,k;
bool cmp(Node a,Node b){
return a.c<b.c;
}
int ffind(int x){
return pre[x]==x?x:pre[x]=ffind(pre[x]);
}
void solve(){
cin>>n>>m>>k;
int num=0;
for(int i=1;i<=n;i++)pre[i]=i;
for(int i=1;i<=m;i++){
int a,b,c;cin>>a>>b>>c;
e[++num]={a,b,c};
}
sort(e+1,e+1+m,cmp);
int ans=1e18;
int cnt=n;
int pos=1;
int ma=-1;
for(;pos<=m;pos++){
if(e[pos].c>k)break;
int fa=ffind(e[pos].a);
int fb=ffind(e[pos].b);
if(fa!=fb){
pre[fa]=fb;
cnt--;
}
ma=max(ma,e[pos].c);
}
if(cnt==1){
ans=min(ans,k-ma);
for(int i=pos;i<=m;i++){
ans=min(ans,e[i].c-k);
}
}else{
int sum=0;
for(int i=pos;i<=m;i++){
int fa=ffind(e[i].a);
int fb=ffind(e[i].b);
if(fa!=fb){
pre[fa]=fb;
sum+=e[i].c-k;
cnt--;
if(cnt==1)break;
}
}
if(cnt==1)ans=min(ans,sum);
}
if(ans==1e18)ans=-1;
cout<<ans<<endl;
}
signed main(){
ios::sync_with_stdio(0);
int T;cin>>T;
while(T--){
solve();
}
return 0;
}