Tietoverkko

 
#include <bits/stdc++.h>

using namespace std; 
int N;
vector<pair<int,int>> v[200000];
long long int sum;
int vasen=0; 
int oikea= 0; 

void haku(int s, int e, int target){
    vasen++; 
    oikea++;
 
    for(auto u: v[s]){

        if(u.first!=e&&u.second>=target) haku(u.first, s, target);
    }


}
   
int main()
{
    long long int i, a,b,c;
    sum=0; 

    cin>>N; 
    vector<vector<long long int>> di; 
    for(i=0; i<N-1; i++){
        cin>>a>>b>>c;
        di.push_back({a,b,c});
        v[a].push_back({b,c});
        v[b].push_back({a,c});
    }
    for(auto kl: di){
        vasen=0;
        haku(kl[0], kl[1] , kl[2]);
        int juu=vasen;
        oikea=0; 
        haku(kl[1], kl[0], kl[2]);
        //cout<<kl[2]*(juu*oikea)<<endl;
        sum+= kl[2]*(juu*oikea);
    }

    cout<<sum;

  
    return 0;
}

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 2, 3

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: RUNTIME ERROR

terminate called after throwing an instance of 'std::bad_alloc'
  what():  std::bad_alloc