Tietoverkko

def määritä(a, b, x, pienimmät):
   # print("määritys") 
    i = 0
    store = [row[:] for row in pienimmät]    
    while i < len(pienimmät):
        if not store[i][a] == 0:
            if store[i][a] < x:
          #      print("yksi") 
                pienimmät[i][b] = pienimmät[i][a]
            else:
                pienimmät[i][b] = x
        #        print("kaksi", i) 
        if not store[i][b] == 0:
            if store[i][b] < x:
                pienimmät[i][a] = pienimmät[i][a]
         #       print("kolme") 
            else:
                pienimmät[i][a] = x						
            #    print("neljä", i) 
        i += 1
    pienimmät[a][b] = x
    pienimmät[b][a] = x
   # print(pienimmät)
    return pienimmät

def laske(verkko):
 #   print(verkko) 
    i = 1
    luku = 0
    for kone in verkko[:-1]:
        saveI = i
        while i < len(kone) - 1:
     #       print(kone[i],kone, i) 
            luku += kone[i]
            i += 1
        i = saveI + 1
    return luku	
	
def main():
    n = int(input())
  #  print(n) 
    i = 0
    a = []
    b = []
    x = []
    pienimmät = [[0 for x in range(n+1)] for y in range(n+1)] 
    while i < n - 1:
        abx = input().split(" ")
        a = int(abx[0])
        b = int(abx[1])
        x = int(abx[2])
      #  print(a, b, x) 
        pienimmät = määritä(a - 1, b - 1, x, pienimmät)
        i += 1
    print(laske(pienimmät))
   # print("moi") 
			
main() 
		
		
	
		

Test 1

Group: 1, 2, 3

Verdict: WRONG ANSWER

Test 2

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED