Codeforces1468 J. Road Reform（MST变形）

题意：

解法：

将边分成<=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;
    //将<=k的边全部加入
    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);//第一种方法是将ma加到k.
        for(int i=pos;i<=m;i++){//第二种方法是选择一条>k的边减到k.
            ans=min(ans,e[i].c-k);
        }
    }else{//如果不连通,那么必须加>k的边
        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;
}

---