题目链接:http://codeforces.com/problemset/problem/1108/F
题意:给你一个图,有很多边,不会有自环和重边,可能现在这个图的mst不唯一,让你改变一些边的权值使得mst唯一,求最小操作次数。
思路:通过分析可以看出,只有边权相等的一些边,才会使mst不唯一,所以只去找这些边权相等且所连顶点又在mst中的边就行了。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,ll> P;
const int maxn=2e5+10;
const int inf=0x3f3f3f3f;
#define pb push_back
#define ft first
#define sd second
struct edge{
int u,v,w;
edge(){}
edge(int u,int v,int w):u(u),v(v),w(w){}
};
int n,m,ans;
int a[maxn];
int p[maxn];
vector<edge> g;
vector<int> vv;
int fin(int x){
return p[x]==x?x:p[x]=fin(p[x]);
}
void unite(int x,int y){
x=fin(x);
y=fin(y);
if(x!=y)p[x]=y;
}
bool same(int x,int y){
return fin(x)==fin(y);
}
bool cmp(edge x,edge y){
return x.w<y.w;
}
void kru(){
for(int i=1;i<=n;i++){
p[i]=i;
}
int ans=0;
sort(g.begin(),g.end(),cmp);
for(int i=0;i<m;i++){
vv.pb(g[i].w);
}
for(int i=0;i<(int)g.size();i++){
int w=g[i].w;
int sum=upper_bound(vv.begin(),vv.end(),w)-lower_bound(vv.begin(),vv.end(),w);//边权相等的边的个数
int k=i+sum;
for(int j=i;j<k;j++){
if(same(g[j].u,g[j].v)){//之前已经加入mst的端点所连的边不算入在内
sum--;
}
}
for(int j=i;j<k;j++){//构成mst的边不算入在内
if(!same(g[j].u,g[j].v)){
unite(g[j].u,g[j].v);
sum--;
}
}
ans+=sum;//剩下的就是要找的边
i=k-1;
}
cout<<ans<<'\n';
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
int x,y,w;
scanf("%d%d%d",&x,&y,&w);
g.pb(edge(x,y,w));
}
kru();
return 0;
}