标签:space 通过 mit using 分割 cstring oar code --
2
3
1 2 2
1 3 3
4
1 2 3
2 3 4
3 4 5
4 17
要把最小生成树扩充为完全图,且保证最小生成树唯一,不变
那么我们考虑把Kruskal做最小生成树的做法
首先把各定点分为n个集合,然后不断通过加轻量级边来扩充最小生成树
考虑得到两个分割s1,s2,若对于任意u属于s1,v属于s2,(u,v)!=(x,y)
那么在完全图中肯定要加入(u,v)这条边,
又知要保证生成树唯一,即要保证(x,y)为轻量级边
所以w(u,v)=w(x,y)+1
所以枚举每条(x,y),通过乘法原理可知ans+=(w(x,y)+1)*(|s1|*|s2|-1);
#include<cstdio> #include<algorithm> #include<cstring> using namespace std; struct my{ int x,y,w; }; const int maxn=50000+10; int fa[maxn],s[maxn]; my edge[maxn]; bool cmp(const my &a,const my &b){ return a.w<b.w; } int getfa(int x){ if(fa[x]==x) return x; return fa[x]=getfa(fa[x]); } int main(){ int t; int x,y,w,n; scanf("%d",&t); while(t--){ long long ans=0; scanf("%d",&n); for (int i=0;i<=n;i++) fa[i]=i,s[i]=1; for (int i=1;i<n;i++) scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].w); sort(edge+1,edge+n,cmp); for (int i=1;i<n;i++){ int x=getfa(edge[i].x); int y=getfa(edge[i].y); if(x==y) continue; ans+=(long long)(edge[i].w+1)*(s[x]*s[y]-1); fa[x]=y; s[y]+=s[x]; } printf("%lld\n",ans); } return 0; }
标签:space 通过 mit using 分割 cstring oar code --
原文地址:https://www.cnblogs.com/lmjer/p/9350758.html