Tietoverkko

n = int(input())
yhteydet = []
naapurit = []

def maara(tietokone, pois):
    m = 1
    for naapuri in naapurit[tietokone]:
        if naapuri != pois:
            m += maara(naapuri, tietokone)

    return m

for i in range(n+1):
    naapurit.append([])

for i in range(n - 1):
    arr = list(map(int, input().split()))
    yhteys = {"a": arr[0], "b": arr[1], "nopeus": arr[2]}
    yhteydet.append(yhteys)
    naapurit[arr[0]].append(arr[1])
    naapurit[arr[1]].append(arr[0])

yhteydet = sorted(yhteydet, key=lambda k: k['nopeus']) 

summa = 0
for yhteys in yhteydet:
    a = yhteys["a"]
    b = yhteys["b"]

    m1 = maara(b, a)

    naapurit[b].remove(a)
    naapurit[a].remove(b)

    m2 = maara(a, b) - 1
    r = m1 * (m2 + 1) * yhteys["nopeus"]
    summa += r

print(summa)

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED