Kruskal - 走廊泼水节 - AcWing 346

Kruskal - 走廊泼水节 - AcWing 346

给定一棵N个节点的树，要求增加若干条边，把这棵树扩充为完全图，并满足图的唯一最小生成树仍然是这棵树。

求增加的边的权值总和最小是多少。

注意： 树中的所有边权均为整数，且新加的所有边权也必须为整数。

输入格式

第一行包含整数t，表示共有t组测试数据。

对于每组测试数据，第一行包含整数N。

接下来N-1行，每行三个整数X,Y,Z，表示X节点与Y节点之间存在一条边，长度为Z。

输出格式

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

每个结果占一行。

数据范围

$1\le N\le 6000,1\le Z\le 100$

输入样例：

2
3
1 2 2
1 3 3
4
1 2 3
2 3 4
3 4 5 

输出样例：

4
17 

---

分析：

$完全图：任意两点间均有边。$

$由题意，每次合并两个连通块时，都要在这两个连通块中的任意两点之间建立一条边。$

$因此，每个连通块内一定是一个完全图。合并时，只需要在两个连通块之间建边。$

$现需要一颗最小生成树扩充为完全图，我们假设树中两个节点为i和j，i和j之间的边权为w_{ij}。$

$现合并节点i和节点j所在的两个连通块，假设新增的边权为w。$

$①、w<w_{ij}，那我们用边w来替换w_{ij}，就能得到一颗更小的生成树，矛盾。$

$②、w=w_{ij}，若我们用w来替换w_{ij}，就能得到一颗同样的最小生成树，与题意最小生成树唯一矛盾。$

$③、w>w_{ij}，满足题意。$

$要使得新增边权最小，那么所有新增的边权取w+1即可。$

$这样每次合并两个连通块a和b，代价为Size[a]\times Size[b]\times (w+1)。$

$其中Size[a]表示节点a所在连通块中节点的数量。$

$可见，我们还需用并查集维护一个Size数组。$

代码：

#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;

const int N=6010;

int T,n,m;
int p[N],Size[N];

struct edge
{
    int u,v,w;
    bool operator < (const edge &t) const
    {
        return w<t.w;
    }
}E[N];

int Find(int x)
{
    if(p[x]!=x) return p[x]=Find(p[x]);
    return p[x];
}

int kruskal()
{
    sort(E,E+m);
    
    int res=0;
    for(int i=0;i<m;i++)
    {
        int a=Find(E[i].u),b=Find(E[i].v),w=E[i].w;
        if(a!=b)
        {
            res+=(Size[a]*Size[b]-1)*(w+1);
            Size[b]+=Size[a];
            p[a]=b;
        }
    }

    return res;
}

int main()
{
    cin>>T;
    while(T--)
    {
        cin>>n;
        m=n-1;
        for(int i=0;i<=n;i++) p[i]=i,Size[i]=1;
        for(int i=0;i<m;i++) cin>>E[i].u>>E[i].v>>E[i].w;
            
        cout<<kruskal()<<endl;
    }
    
    return 0;
}