結果

問題 No.19 ステージの選択
コンテスト
ユーザー norioc
提出日時 2026-10-04 20:30:27
言語 PyPy3
(7.3.23 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 46 ms / 5,000 ms
+ 377µs
コード長 1,149 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 63 ms
コンパイル使用メモリ 82,684 KB
実行使用メモリ 71,936 KB
最終ジャッジ日時 2026-10-04 20:30:40
合計ジャッジ時間 2,969 ms
ジャッジサーバーID
（参考情報） 		judge4_0 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from collections import defaultdict
from atcoder.scc import SCCGraph


def scc(n: int, adj: dict[int, list[int]]) -> list[list[int]]:
    """グラフを強連結成分分解する。

    Args:
        n: 頂点数
        adj: 隣接リスト {u: [v...]}

    Returns:
        強連結成分のリスト (トポロジカル順)
    """

    g = SCCGraph(n)
    for u, vs in adj.items():
        for v in vs:
            g.add_edge(u, v)

    return g.scc()


N = int(input())

adj = defaultdict(list)
difficulties = []  # 各ステージの難易度
for i in range(N):
    L, S = map(int, input().split())
    S -= 1
    # S がクリア済みだと i の難易度が半分になる
    adj[S].append(i)
    difficulties.append(L)

cc = scc(N, adj)
ans = 0
vset = set()  # 難易度が半分になる頂点集合
for g in cc:  # グループ g に属するステージをクリアする
    ds = [difficulties[v] for v in g]
    if any(v in vset for v in g):
        ans += sum(ds) / 2
    else:
        ds.sort()
        ans += ds[0]
        ans += sum(ds[1:]) / 2

    for v in g:
        for to in adj[v]:
            vset.add(to)

print(ans)
0