Acwing346. 走廊泼水节

题目链接：346. 走廊泼水节 - AcWing题库

思路：对于每条边，有两个顶点u、v，分别属于两个集合（假设集合内部点已经完成全连接），那么将这两个集合连接起来，需要连接s[u]*s[v]-1条边（s[i]代表i所在集合 点的数量），且这些边的最小值为(u--v边权)w+1。

连接顺序：将所有边按照边权大小从小到大排序。

1.为什么必须增序排序，不可以降序排序？

假设有一条小边放到了许多大边后面，则可能会通过它使原集合生成树中两个点之间形成一条更小的路径，会使最小生成树不再是数据所给的树，将违背题意。

2.为什么可以确保取到最小？

倘若有其他方案或更优方案，那么在上述连接过程中，在连接集合u和集合v时，一定会有一条边小于w+1，那么在选取最小生成树时，选这条生成的边而不选初始的边也可以获得 最优/更优解，这与题意不符，因此可以确保取到最小。

代码：

#include<iostream>
#include<algorithm>
using namespace std;

const int N = 6010;
int T,n;
int u,v,w;
int ans;
int f[N],s[N];
struct Edge
{
    int u,v,w;
    bool operator<(const Edge a)const{
        return w<a.w;
    }
}e[N];

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

void solve()
{
    cin>>n;
    
    //初始化
    ans=0;
    for(int i=1;i<=n;i++) f[i]=i,s[i]=1;
    
    for(int i=1;i<n;i++)
    {
        cin>>e[i].u>>e[i].v>>e[i].w;
    }
    
    sort(e+1,e+n);
    
    for(int i=1;i<n;i++)
    {
        //将u点所在的集合与v点所在的集合合并，需要连接su+sv-1条边(s[i]代表i点所在集合的点数)，且每条边边权至少为w+1
        u=e[i].u,v=e[i].v,w=e[i].w;
        u=Find(u),v=Find(v);
        if(u==v) continue;
        ans+=(w+1)*(s[u]*s[v]-1);
        s[u]+=s[v];
        f[v]=u;
    }
    
    cout<<ans<<endl;
}

int main()
{
    cin>>T;
    while(T--)
    {
        solve();
    }
    
    return 0;
}