MST Unification CodeForces - 1108F （MST+思维）

题目链接

题意：给定一个连通的带边权的无向图，每次操作可以给一条边权值+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
}