Tietoverkko

input/code.cpp: In function 'int main()':
input/code.cpp:35:30: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for (size_t i = 0; i < n+1; i++)
                            ~~^~~~~

#include <vector>
#include <iostream>
#include <algorithm>
 
using namespace std;
 
vector<pair<int,int>> puu[5001];
int kayty[5001];
 
long long total = 0;
 
void count(int index, int min) {
    vector<pair<int,int>> naapurit = puu[index];
    kayty[index] = 1;
    for (pair<int, int> pair : naapurit) {
        if (!kayty[pair.first]) {
            int val = std::min(min, pair.second);
            total += val;
            count(pair.first, val);
        }
    }
}
int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n-1; i++)
    {
        int a;
        int b;
        int x;
        cin >> a >> b >> x;
        puu[a].push_back(pair<int,int>(b, x));
        puu[b].push_back(pair<int,int>(a, x));
        for (size_t i = 0; i < n+1; i++)
        {
            kayty[i] = 0;
        }
        count(a, x);
    }
    cout << total;
}
 

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: RUNTIME ERROR