結果

問題 No.19 ステージの選択
ユーザー dangodango
提出日時 2023-07-08 18:01:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 108 ms / 5,000 ms
コード長 895 bytes
コンパイル時間 1,587 ms
コンパイル使用メモリ 86,356 KB
実行使用メモリ 72,232 KB
最終ジャッジ日時 2023-09-29 19:23:57
合計ジャッジ時間 4,207 ms
ジャッジサーバーID
(参考情報)
judge15 / judge12
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 106 ms
72,220 KB
testcase_01 AC 107 ms
72,148 KB
testcase_02 AC 107 ms
72,124 KB
testcase_03 AC 108 ms
71,736 KB
testcase_04 AC 106 ms
72,004 KB
testcase_05 AC 107 ms
72,112 KB
testcase_06 AC 106 ms
72,212 KB
testcase_07 AC 106 ms
71,992 KB
testcase_08 AC 108 ms
71,924 KB
testcase_09 AC 107 ms
72,124 KB
testcase_10 AC 108 ms
71,940 KB
testcase_11 AC 107 ms
72,104 KB
testcase_12 AC 105 ms
72,232 KB
testcase_13 AC 104 ms
72,164 KB
testcase_14 AC 106 ms
72,156 KB
testcase_15 AC 106 ms
71,996 KB
testcase_16 AC 107 ms
72,068 KB
testcase_17 AC 105 ms
71,972 KB
testcase_18 AC 106 ms
72,120 KB
testcase_19 AC 108 ms
72,100 KB
testcase_20 AC 105 ms
72,212 KB
testcase_21 AC 108 ms
71,776 KB
testcase_22 AC 106 ms
71,764 KB
testcase_23 AC 108 ms
72,000 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