Tietoverkko

#include <iostream>
#include <vector>
#include <map>

using namespace std; 
int N;
vector<pair<int,int>> v[200001]; 
long long int sum;
bool tama=false; 

map<pair<int, int>, short int > juu; 
long long int globalsum=0; 
long long int valisum=0; 

int haku(int s, int e,int target){

    int ssumma=1; 

    for(auto u: v[s]){ 
        if(u.first!=e&&u.second==target){
            
            juu[{s, u.first}]=1;
            int uusi = haku(u.first, s, target);
            valisum+=globalsum*uusi; 
            globalsum+=uusi; 
            tama=true; }
      
        if(u.first!=e&&u.second>target) ssumma+=haku(u.first, s, target);
}
return ssumma; 
}


int main()
{     
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);  
    int a,b,c;
    int i; 
    sum=0; 
    cin>>N; 
    int di[N][3]; //a ja b pitää olla long long, mutta miksi?
    for(i=0; i<N-1; i++){
        cin>>a>>b>>c;
        di[i][0]=c;
        di[i][1]=a;
        di[i][2]=b;
        v[a].push_back({b,c});
        v[b].push_back({a,c}); 
     
    }
    int vasen, oikea; 
    long long int kl0, kl1, kl2; 
    for(i=0; i<N-1; i++){
        kl0 = di[i][0]; 
        kl1 = di[i][1];
        kl2 = di[i][2];
        if(juu[{kl2,kl1}]||juu[{kl1,kl2}])
            continue;
       
        vasen=haku(kl1, kl2 , kl0);
        oikea=haku(kl2, kl1, kl0);
        juu[{kl2, kl1}]=1;
        if(tama){
            valisum+=globalsum*vasen;
            globalsum+=vasen; 
            valisum+=globalsum*oikea;
            sum+=valisum*kl0;
            tama=false;
            globalsum=0; 
            valisum=0; 
            continue;
        }
       
        sum+= kl0*vasen*oikea;
      
    }


    cout<<sum;

  
    return 0;
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED