解题报告:这题一开始没看懂,其实意思挺简单的,当然是看了题解以后 还挺容易的,主要是让你把一个树形图,变成一个完全图,(完全图就是边长数量=(n*(n-1)/2)n为点数),同时满足之前的那棵树还是唯一的最小生成树,通过图我们可以发现,在合并两个集合的时候,如果枚举的边小于等于该条边的长度w,是不行的,不满足唯一生成树,这样我们每次只需要加上(pq-1)(w+1)就是正解了,pq分别代表两个集合中的点数,减去1是因为一条边已经在了。
#include<bits/stdc++.h>
using namespace std;
const int N=6010 , M = N*N/2;
typedef long long ll;
int s[N];
int p[N];
struct node{
int a;
int b;
int w;
bool operator <(const node&T)const
{
return w<T.w;
}
}q[N];
int find(int x)
{
if(x!=p[x])
p[x]=find(p[x]);
return p[x];
}
int main()
{
int t;
cin>>t;
while(t--)
{
int n;
cin >> n;
for(int i=1;i<=n;i++)
p[i]=i,s[i]=1;
for(int i=0;i<n-1;i++)
{
int a,b,c;
cin >> a >> b >>c;
q[i]={
a,b,c};
}
ll res =0 ;
sort(q,q+n-1);
for(int i=0;i<n-1;i++)
{
int a = find(q[i].a) , b=find(q[i].b) , c = q[i].w;
if(a!=b)
{
p[a] = b;
res += (c+1)*(s[a]*s[b]-1);
s[b]+=s[a];
}
}
cout<< res << endl;
}
}