Tietoverkko

#include <bits/stdc++.h>
using namespace std;

long n, a, b, c, vastaus;
vector<pair<long, pair<long, long>>> v;
long koko[200001], edustaja[200001];

int main() {
    cin>>n;
    for(long i=0;i<n-1;i++){
        cin>>a>>b>>c;
        v.push_back({c, {a,b}});
        
        edustaja[i+1]=i+1;
        koko[i+1]=1;
    }
    edustaja[n]=n;
    koko[n]=1;
    sort(v.begin(), v.end());
    
    for(long i=n-2;i>=0;i--){
        a=v[i].second.first;
        b=v[i].second.second;
        while(a!=edustaja[a]){
            a = edustaja[a];
        }
        while(b!=edustaja[b]){
            b = edustaja[b];
        }
        vastaus += v[i].first*koko[a]*koko[b];
        if(koko[a]>koko[b]){
            edustaja[b]=a;
            koko[a] += koko[b];
        }else{
            edustaja[a]=b;
            koko[b] += koko[a];
        }
    }
    cout<<vastaus;
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: ACCEPTED