結果

問題 No.19 ステージの選択
ユーザー Navier_Boltzmann
提出日時 2023-06-14 21:06:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 53 ms / 5,000 ms
コード長 1,656 bytes
コンパイル時間 692 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 56,704 KB
最終ジャッジ日時 2024-06-23 03:11:43
合計ジャッジ時間 2,555 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

import pypyjit
pypyjit.set_param('max_unroll_recursion=-1')
from collections import *
from itertools import *
from functools import *
from heapq import *
import sys,math
input = sys.stdin.readline

N = int(input())
e = [[] for _ in range(N)]
re = [[] for _ in range(N)]
L = []
for i in range(N):
    l,s = map(int,input().split())
    L.append(2*l)
    s -= 1
    if i==s:
        continue
    e[s].append(i)
    re[i].append(s)

# 強連結成分分解(SCC): グラフGに対するSCCを行う
# 入力: <N>: 頂点サイズ, <G>: 順方向の有向グラフ, <RG>: 逆方向の有向グラフ
# 出力: (<ラベル数>, <各頂点のラベル番号>)
def scc(N, G, RG):
    order = []
    used = [0]*N
    group = [None]*N
    def dfs(s):
        used[s] = 1
        for t in G[s]:
            if not used[t]:
                dfs(t)
        order.append(s)
    def rdfs(s, col):
        group[s] = col
        used[s] = 1
        for t in RG[s]:
            if not used[t]:
                rdfs(t, col)
    for i in range(N):
        if not used[i]:
            dfs(i)
    used = [0]*N
    label = 0
    for s in reversed(order):
        if not used[s]:
            rdfs(s, label)
            label += 1
    return label, group
    
    
n,group = scc(N,e,re)
G = [set() for _ in range(n)]
for i in range(N):
    gi = group[i]
    for j in e[i]:
        gj = group[j]
        if gj==gi:
            continue
        G[gj].add(gi)
X = [[] for _ in range(n)]
for i,g in enumerate(group):
    X[g].append(L[i])

ans = 0

for i in range(n):

    if len(G[i])==0:
        ans += sum(X[i])//2 + min(X[i])//2
    else:
        ans += sum(X[i])//2
print(ans*0.5)
0