题意:给定一个连通的带边权的无向图,每次操作可以给一条边权值+1,问最少多少次操作后,使得图满足最小生成树是唯一的。
(2<=n<=2e5)
思路:
最小生成树在算法导论中有这个性质:
把一个连通无向图的生成树边按权值递增排序,称排好序的边权列表为有序边权列表,则任意两棵最小生成树的有序边权列表是相同的。(算法导论23.1-8)
所以如果MST不唯一,那么肯定是考虑到了相同边权的若干条边。
那么我们写kruskal算法的时候加一些判断就可以了。如果不需要放进MST,就让这条边的权值+1
#include<bits/stdc++.h>
using namespace std;
#define Please return
#define AC 0;
#define LL long long
#define ULL unsigned long long
const LL INF=2e18+7;
const int maxn=2e5+7;
struct Edge
{
int v,u,w;
friend bool operator <(const Edge &a,const Edge &b)
{
return a.w<b.w;
}
}edge[maxn];
int n,m;
int st[maxn];//并查集
int ans=0;
int Find(int rt)
{
if(st[rt]<0) return rt;
st[rt]=Find(st[rt]);
return st[rt];
}
void Union(int rt1,int rt2)
{
if(st[rt1]<st[rt2])
{
st[rt1]+=st[rt2];
st[rt2]=rt1;
}
else
{
st[rt2]+=st[rt1];
st[rt1]=rt2;
}
}
int main()
{
fill(st,st+maxn,-1);
scanf("%d %d",&n,&m);
for(int i=1;i<=m;i++)
{
scanf("%d %d %d",&edge[i].u,&edge[i].v,&edge[i].w);
}
sort(edge+1,edge+m+1);
for(int i=1;i<=m; )
{
int j=i;
while(j<=m && edge[j].w==edge[i].w) j++;
for(int k=i;k<j;k++)
{
int rt1=Find(edge[k].v),rt2=Find(edge[k].u);
if(rt1!=rt2) ans++;//先把所有相同的边都算进去
}
for(int k=i;k<j;k++)//MST要收录的边,
//由于每一个MST的边权值列表都是一样的,
//所以对于权值相同的边,不用考虑加入并查集的顺序
{
int rt1=Find(edge[k].v),rt2=Find(edge[k].u);
if(rt1!=rt2)
{
ans--;
Union(rt1,rt2);
}
}
i=j;
}
printf("%d\n",ans);
Please AC
}