CSES - Datatähti 2022 alku - Results
Submission details
Task:Tietoverkko
Sender:Jellu
Submission time:2021-10-17 11:07:22 +0300
Language:Python3 (CPython3)
Status:READY
Result:10
Feedback
groupverdictscore
#1ACCEPTED10
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.04 s1, 2, 3details
#2--2, 3details
#3--3details

Code

etaisyydet = {1:0}
yhteydet = {}
syvyydet = {1:0}
summa = 0

n = int(input())
for i in range(n-1):
    a,b,x = [int(x) for x in input().split()]
    etaisyydet[max(a,b)] = x
    if a not in yhteydet:
        yhteydet[a] = [b]
    else:
        yhteydet[a].append(b)
    if b not in yhteydet:
        yhteydet[b] = [a]
    else:
        yhteydet[b].append(a)
    syvyydet[b] = syvyydet[a] + 1 

for i in range(1, n + 1):
    for j in range(i + 1, n + 1):
        a = i
        b = j
        lista = []
        while a != b:
            if syvyydet[a] > syvyydet[b]:
                lista.append(etaisyydet[a])
                a = yhteydet[a][0]
            elif syvyydet[b] > syvyydet[a]:
                lista.append(etaisyydet[b])
                b = yhteydet[b][0]
            else:
                lista.append(etaisyydet[a])
                a = yhteydet[a][0]
                lista.append(etaisyydet[b])
                b = yhteydet[b][0]
        summa += min(lista)

print(summa)

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
100
1 2 74
1 3 100
2 4 50
3 5 40
...

correct output
88687

user output
88687

Test 2

Group: 2, 3

Verdict:

input
5000
1 2 613084013
1 3 832364259
2 4 411999902
3 5 989696303
...

correct output
1103702320243776

user output
(empty)

Test 3

Group: 3

Verdict:

input
200000
1 2 613084013
1 3 832364259
2 4 411999902
3 5 989696303
...

correct output
1080549209850010931

user output
(empty)