Tietoverkko

def määritä(a, b, x, pienimmät: dict, n):
   # print("määritys") 
    i = 0
    store = pienimmät.copy()
  #  print(store)
    if a in store:
        for i in store[a]:
        #     print(store[i])
            if store[a][i] < x:
                # print("kolme") 
                if not b in pienimmät:
                    pienimmät.update({b: {i: store[a][i]}})
                else:
                    pienimmät[b].update({i: store[a][i]})
                pienimmät[i][b] = store[i][a]            
            else:
                # print("neljä") 
                if not b in pienimmät:
                    pienimmät.update({b: {i: x}})
                else:
                    pienimmät[b].update({i: x})
                pienimmät[i][b] = x
    if not a in pienimmät:
        pienimmät.update({a : {b : x}})
    else:
        pienimmät[a].update({b : x})
    if not b in pienimmät:
        pienimmät.update({b : {a : x}})
    else:
        pienimmät[b].update({a : x})

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

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

Test 3

Group: 3

Verdict: TIME LIMIT EXCEEDED