AcWing 346. 走廊泼水节

解题报告：这题一开始没看懂，其实意思挺简单的，当然是看了题解以后 还挺容易的，主要是让你把一个树形图，变成一个完全图，（完全图就是边长数量=（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;
    }
}