Tietoverkko

def laske(kaikki: list, n):
    luku = 0
    i = 0
    käydyt = 0
    while i < len(kaikki):
        kaikki[i].sort(reverse=True)
        for num in kaikki[i]:
            luku += num * käydyt
            käydyt += 1
        i += 1
    return luku
            

def sijoita(a, b, x, pienimmät: list, kerrokset: dict, kerta):

    kerros = 0
    if kerta == 0:
        kerrokset.update({a: 0, b: 1})
        pienimmät.append([0])
        pienimmät.append([x])
    else: 
        kerros = kerrokset[a] + 1
        if kerros >= len(pienimmät):
            pienimmät.append([x])
        else:
            pienimmät[kerros].append(x)
        kerrokset.update({b: kerros})

 #   print(pienimmät)
    return pienimmät, kerrokset

def main():
    n = int(input())
  #  print(n) 
    i = 0
    a = []
    b = []
    x = []
    pienimmät = []
    kerrokset = {}

    while i < n - 1:
        abx = input().split(" ")
        a = int(abx[0])
        b = int(abx[1])
        x = int(abx[2])
        pienimmät, kerrokset = sijoita(a, b, x, pienimmät, kerrokset, i) 
      #  print(a, b, x) 
        
        i += 1
   # print("moi") 
    print(laske(pienimmät, n))
			
main() 

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 2, 3

Verdict: WRONG ANSWER

Test 3

Group: 3

Verdict: WRONG ANSWER