Tietoverkko

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

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

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

    long long 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);
            kerro.push_back(uusi);
            tama=true; }
      
        if(u.first!=e&&u.second>target) ssumma+=haku(u.first, s, target);
}
return ssumma; 
}


int main()
{     
    long long int i,a,b,c;
    sum=0; 
    cin>>N; 
    vector<vector<long long int>> di; //a ja b pitää olla long long, mutta miksi?
    for(i=0; i<N-1; i++){
        cin>>a>>b>>c;
        di.push_back({c,a,b});
        v[a].push_back({b,c});
        v[b].push_back({a,c}); 
     
    }
    long long int vasen, oikea; 
    for(auto kl: di){
        if(juu[{kl[2],kl[1]}]||juu[{kl[1],kl[2]}])
            continue;
       
        vasen=haku(kl[1], kl[2] , kl[0]);
        oikea=haku(kl[2], kl[1], kl[0]);
        juu[{kl[2], kl[1]}]=1;
        if(tama){
            kerro.push_back(vasen);
            kerro.push_back(oikea);
            long long  int maara=kerro.size();
            for(int ri=0; ri<maara; ri++){
         
                for( int rj=ri; rj<maara; rj++){
                    if(ri==rj) continue;
                  
                    sum+=kl[0]*kerro[ri]*kerro[rj];  }
                     }
            tama=false;
            kerro.clear();
            continue;
        }
       
        sum+= kl[0]*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