走廊泼水节—— MST *

2022-02-03 16:03:44 阅读：172 来源： 互联网

标签：泼水节 const int MST long fa edge 走廊 define

思路

感觉方法很妙的一道题，用类似kruskal的方法，每次合并两个联通块x, y时，假设这条边长度为 w w w，若要构成完全图，则我们需要额外添加 S x S y − 1 S_xS_y-1 SxSy−1条边，其中 S i S_i Si表示第 i i i 个连通块中的点数量，由于要保证长为 w w w 的仍为MST中边，所以我们新加的这些边长度最小为w+1。

对于每次合并都累加 ( w + 1 ) ( S x S y − 1 ) (w+1)(S_xS_y-1) (w+1)(SxSy−1) ，即可得到最终答案。容易证明这样恰好构成了完全图，无重边也无缺边。

//
//
#include <bits/stdc++.h>
using namespace std;
//#define mp make_pair
#define pii pair<int,int>
#define pb push_back
#define ll long long
#define LL long long
#define ld long double
#define endl '\n'
#define RE0 return 0
#define For(i,j,k) for (int i=(int)(j);i<=(int)(k);i++)
#define Rep(i,j,k) for (int i=(int)(j);i>=(int)(k);i--)

//#define int long long
#define db double
const int maxn = 6e3 + 10;
const int maxm = 6e3 + 10;
const int P = 1e9 + 7;    //998244353
const int INF = 0x3f3f3f3f;
const double eps=1e-7;
struct Edge {
    int u, v, w;
    bool operator < (const Edge&x) const {
        return w < x.w;
    }
}edge[maxm];
int fa[maxn];
int s[maxn];
int find(int x) {
    if(x == fa[x]) return x;
    return fa[x] = find(fa[x]);
}
int n;
void init() {
    for(int i = 1;  i<= n; i++) {
        fa[i] = i;
        s[i] = 1;
    }
    
}
void solve() {
    cin >> n;
    init();
    for(int i = 1; i < n; i++) {
        cin >> edge[i].u >> edge[i].v >> edge[i].w;
    }
    ll ans = 0;
    sort(edge + 1, edge + n );
    for(int i = 1; i < n; i++) {
        int ev = find(edge[i].v);
        int eu = find(edge[i].u);
        if(ev == eu) continue;
        ans = ans + (edge[i].w + 1) * (s[ev] * s[eu] - 1);
        fa[ev] = eu;
        s[eu] += s[ev];
    }
    cout << ans << endl;
}
signed main() {
     ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
//    int T; scanf("%d", &T); while(T--)
//    freopen("1channel00.txt","r",stdin);
//    freopen("2.txt","w",stdout);
    int T; cin >> T; while(T--)
        solve();
    RE0;
}

标签：泼水节,const,int,MST,long,fa,edge,走廊,define
来源： https://blog.csdn.net/m0_59273843/article/details/122776448

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

走廊泼水节—— MST *

思路