結果

問題 No.19 ステージの選択
ユーザー dangodango
提出日時 2023-07-08 18:01:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 52 ms / 5,000 ms
コード長 895 bytes
コンパイル時間 715 ms
コンパイル使用メモリ 82,236 KB
実行使用メモリ 55,680 KB
最終ジャッジ日時 2024-07-22 13:28:54
合計ジャッジ時間 2,293 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 48 ms
54,784 KB
testcase_01 AC 48 ms
55,168 KB
testcase_02 AC 50 ms
55,040 KB
testcase_03 AC 48 ms
54,656 KB
testcase_04 AC 51 ms
55,680 KB
testcase_05 AC 49 ms
55,680 KB
testcase_06 AC 51 ms
55,168 KB
testcase_07 AC 50 ms
54,912 KB
testcase_08 AC 49 ms
55,296 KB
testcase_09 AC 52 ms
55,552 KB
testcase_10 AC 48 ms
55,040 KB
testcase_11 AC 48 ms
54,528 KB
testcase_12 AC 48 ms
54,784 KB
testcase_13 AC 48 ms
54,784 KB
testcase_14 AC 48 ms
55,168 KB
testcase_15 AC 48 ms
54,784 KB
testcase_16 AC 50 ms
55,040 KB
testcase_17 AC 49 ms
55,040 KB
testcase_18 AC 49 ms
55,680 KB
testcase_19 AC 48 ms
54,912 KB
testcase_20 AC 50 ms
55,168 KB
testcase_21 AC 50 ms
55,040 KB
testcase_22 AC 49 ms
55,296 KB
testcase_23 AC 49 ms
55,296 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

from functools import reduce

n = int(input())
l, s = [], []
g = [[] for _ in range(n)]
for i in range(n):
    li, si = map(int, input().split())
    si -= 1
    l.append(li)
    s.append(si)
    g[si].append(i)

ans = reduce(lambda x, y: x + y, l) / 2.0
used = [False] * n
vs = []

def dfs(v):
    used[v] = True
    for i in g[v]:
        if not used[i]:
            dfs(i)
    vs.append(v)

Min = 0

def rdfs(v):
    global Min
    used[v] = True
    Min = min(Min, l[v])
    if not used[s[v]]:
        rdfs(s[v])

for i in range(n):
    if not used[i]:
        dfs(i)

used = [False] * n
used2 = [False] * n

def dfs2(v):
    used2[v] = True
    for i in g[v]:
        if not used2[i]:
            dfs2(i)

for i in reversed(range(len(vs))):
    if used[vs[i]] or used2[vs[i]]:
        continue
    Min = 100
    rdfs(vs[i])
    ans += Min / 2.0
    dfs2(vs[i])

print("{:.1f}".format(ans))
0