走廊泼水节

题目描述

给定一棵N个节点的树，要求增加若干条边，把这棵树扩充为完全图，并满足图的唯一最小生成树仍然是这棵树。
求增加的边的权值总和最小是多少。
注意： 树中的所有边权均为整数，且新加的所有边权也必须为整数。

输入格式

第一行包含整数t，表示共有t组测试数据。
对于每组测试数据，第一行包含整数N。
接下来N-1行，每行三个整数X,Y,Z，表示X节点与Y节点之间存在一条边，长度为Z。

输出格式

每组数据输出一个整数，表示权值总和最小值。
每个结果占一行。

数据范围

1≤N≤6000
1≤Z≤100

输入样例
2
3
1 2 2
1 3 3
4
1 2 3
2 3 4
3 4 5
输出样例
4
17

题目分析

本题有两个条件：1.将一颗树补全为一个完全图。2.该完全图的最小生成树还是题中一开始给的树。

我们首先要考虑怎么补全边才能是该图的最小生成树不变。关于这个问题我们可以根据kruskal算法的原理来考虑：kruskal算法是将所有边从小到大进行排序，每次取出最小的边。如果这条边能够连接当前两个不连通的连通块，那么就将该边加入。在这道题中，为了不影响这条边且保证权值和最小，我们需要给这两个连通块加的边的长度全部等于这个最小边的长度w+1。

另一个问题是合并两个连通块时，要加多少边？设两个连通块中分别有numa和numb个点，如果要将两个连通块合并并补成完全图，我们需要加numa*numb条边，去掉一条最小生成树的边之外，一共要加numa*numb-1条边。这个操作我们可以在写并查集时额外维护一共num[]数组来记录每个连通块中点的数量。

代码如下

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#define LL long long
#define ULL unsigned long long
#define PII pair<int,int>
#define x first
#define y second
using namespace std;
const int N=6e3+5,INF=0x3f3f3f3f;
struct Edge{
    int u,v,w;
    bool operator<(const Edge &a)const
    { return w<a.w; }
}h[N];
int p[N],num[N];
int find(int x)
{
    if(p[x]!=x)  p[x]=find(p[x]);
    return p[x];
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        for(int i=0;i<N;i++) p[i]=i,num[i]=1;
        int n;
        scanf("%d",&n);
        for(int i=1;i<n;i++)
        {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            h[i]={u,v,w};
        }
        int ans=0;
        sort(h+1,h+n);
        for(int i=1;i<n;i++)
        {
            int a=find(h[i].u),b=find(h[i].v),w=h[i].w+1;
            if(a==b) continue;
            p[b]=a;
            ans+=(num[a]*num[b]-1)*w;
            num[a]+=num[b];
        }
        printf("%d\n",ans);
    }
    return 0;
}