tyvj1391 走廊泼水节

背景

话说，中中带领的OIER们打算举行一次冬季泼水节，当然这是要秘密进行的，绝对不可以让中中知道。不过中中可是老江湖了，当然很快就发现了我们的小阴谋，于是他准备好水枪迫不及待的想要加入我们了。

描述

我们一共有N个OIER打算参加这个泼水节，同时很凑巧的是正好有N个水龙头(至于为什么，我不解释)。N个水龙头之间正好有N-1条小道，并且每个水龙头都可以经过小道到达其他水龙头(这是一棵树，你应该懂的..)。但是OIER门为了迎接中中的挑战，决定修建一些个道路(至于怎么修，秘密~),使得每个水龙头到每个水龙头之间都有一条直接的道路连接(也就是构成一个完全图呗~)。但是OIER门很懒得，并且记性也不好，他们只会去走那N-1条小道，并且希望所有水龙头之间修建的道路，都要大于两个水龙头之前连接的所有小道(小道当然要是最短的了)。所以神COW们，帮那些OIER们计算一下吧，修建的那些道路总长度最短是多少，毕竟修建道路是要破费的~~

输入格式 本题为多组数据~ 第一行t，表示有t组测试数据 对于每组数据 第一行N，表示水龙头的个数（当然也是OIER的个数）；

2到N行，每行三个整数X,Y,Z；表示水龙头X和水龙头Y有一条长度为Z的小道 输出格式

对于每组数据，输出一个整数，表示修建的所有道路总长度的最短值。

模拟kruskal的过程，按边权排序之后，对于一条边(u,v)=w，补成完全图的话需要size[u]*size[v]-1条边，为了保证最小生成树唯一边权定为w+1，然后把两个并查集合并。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define LL long long
struct edge
{
int f,t,l;
bool operator < (const edge & ee) const
{
return l<ee.l;
}
}g[6010];
int fa[6010],size[6010],n;
LL ans;
int find(int x)
{
return fa[x]==x?x:fa[x]=find(fa[x]);
}
void join(int x,int y)
{
size[y]+=size[x];
fa[x]=y;
}
int main()
{
int i,j,k,p,q,x,y,z,T;
scanf("%d",&T);
while (T--)
{
scanf("%d",&n);
for (i=1;i<n;i++)
scanf("%d%d%d",&g[i].f,&g[i].t,&g[i].l);
for (i=1;i<=n;i++)
fa[i]=i,size[i]=1;
sort(g+1,g+n);
ans=0;
for (i=1;i<n;i++)
{
x=find(fa[g[i].f]);
y=find(fa[g[i].t]);
ans+=((LL)g[i].l+1)*(size[x]*size[y]-1);
join(x,y);
}
printf("%lld\n",ans);
}
}