Task: | Tietoverkko |
Sender: | Kirill_Akimov |
Submission time: | 2021-10-17 20:17:39 +0300 |
Language: | Python3 (PyPy3) |
Status: | READY |
Result: | 25 |
group | verdict | score |
---|---|---|
#1 | ACCEPTED | 10 |
#2 | ACCEPTED | 15 |
#3 | TIME LIMIT EXCEEDED | 0 |
test | verdict | time | group | |
---|---|---|---|---|
#1 | ACCEPTED | 0.06 s | 1, 2, 3 | details |
#2 | ACCEPTED | 0.56 s | 2, 3 | details |
#3 | TIME LIMIT EXCEEDED | -- | 3 | details |
Code
from array import array n = int(input()) graph = {} s = 0 for _ in range(n-1): a, b, x = [int(e) for e in input().split()] if a in graph: graph[a][b] = x else: graph[a] = {b: x} if b in graph: graph[b][a] = x else: graph[b] = {a: x} l = array('I', []) def count(v, p): global s global graph global l length = len(l) d = {} for i in range(length): if l[i] > graph[v][p]: d[i] = l[i] l[i] = graph[v][p] if p > 0: l.append(graph[v][p]) s += sum(l) for key in graph[v]: if key != p: count(key, v) for key in d: l[key] = d[key] current = len(l) if p > 0: for i in range(length + 1, len(l)): if l[i] > graph[v][p]: l[i] = graph[v][p] count(1, 0) print(s)
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: ACCEPTED
input |
---|
5000 1 2 613084013 1 3 832364259 2 4 411999902 3 5 989696303 ... |
correct output |
---|
1103702320243776 |
user output |
---|
1103702320243776 |
Test 3
Group: 3
Verdict: TIME LIMIT EXCEEDED
input |
---|
200000 1 2 613084013 1 3 832364259 2 4 411999902 3 5 989696303 ... |
correct output |
---|
1080549209850010931 |
user output |
---|
(empty) |