#include<bits/stdc++.h> using namespace std; struct node{ int u,v,w; }e[6100]; int s[6100],fa[6100]; int sum=0; bool cmp(node a,node b){ return a.w<b.w; } int find(int x){ return x==fa[x]?x:fa[x]=find(fa[x]); } void solve(){ int n,m; cin>>n; sum=0; m=n-1; for(int i=1;i<=n;i++){ s[i]=1; fa[i]=i; } for(int i=1;i<=m;i++){ int u,v,w; cin>>u>>v>>w; e[i]={u,v,w}; } sort(e+1,e+1+m,cmp); for(int i=1;i<=m;i++){ int u=e[i].u,v=e[i].v,w=e[i].w; int a=find(u),b=find(v); if(a!=b){ sum+=(s[a]*s[b]-1)*(w+1); fa[a]=b; s[b]+=s[a]; } } cout<<sum<<endl; } int main(){ int tt; cin>>tt; while(tt--){ solve(); } return 0; }
时间: 2024-04-10 11:32:23 浏览: 164
万能头文件#include&amp;lt;bits/stdc++.h&amp;gt; using namespace std;
这段代码是一个求解最小生成树权值和的程序。它使用了Kruskal算法来构建最小生成树。算法的主要思路是首先将边按照权值从小到大进行排序,然后逐条添加边到最小生成树中,保证不形成环。在添加每条边的时候,需要判断该边的两个顶点是否属于同一个连通分量,如果不是,则将两个连通分量合并,并且更新权值和。
程序中的solve函数用于解决单个测试样例,它首先读入顶点数n,然后初始化并查集和计算最小生成树权值和的变量sum。接下来读入边的信息,并按照权值进行排序。然后使用并查集来判断是否形成环,并更新sum的值。最后输出sum。
主函数main用于循环读入测试样例的个数tt,并调用solve函数解决每个样例。
请问有什么问题我可以帮助你解答的吗?
阅读全文