結果

問題 No.19 ステージの選択
コンテスト
ユーザー norioc
提出日時 2026-10-04 20:39:42
言語 PyPy3
(7.3.23 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 45 ms / 5,000 ms
+ 755µs
コード長 1,148 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 63 ms
コンパイル使用メモリ 81,676 KB
実行使用メモリ 73,096 KB
最終ジャッジ日時 2026-10-04 20:40:09
合計ジャッジ時間 3,206 ms
ジャッジサーバーID
（参考情報） 		judge2_1 / judge4_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
discounted = set()  # 難易度が半分の頂点集合
for g in cc:  # グループ g に属するステージをクリアする
    ds = [difficulties[v] for v in g]
    if any(v in discounted for v in g):
        ans += sum(ds) / 2
    else:
        mi = min(ds)
        ans += sum(ds) / 2 + mi / 2

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

print(ans)
0