結果

問題 No.1194 Replace
ユーザー tktk_snsn
提出日時 2021-02-08 22:28:26
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 850 bytes
コンパイル時間 178 ms
コンパイル使用メモリ 82,352 KB
実行使用メモリ 263,332 KB
最終ジャッジ日時 2024-07-06 02:26:45
合計ジャッジ時間 14,133 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.buffer.readline
sys.setrecursionlimit(10 ** 7)

N, M = map(int, input().split())
BC = tuple(tuple(map(int, input().split())) for _ in range(M))

val = set()
for b, c in BC:
    val.add(b)
    val.add(c)

val = sorted(val)
vtoi = {v: i for i, v in enumerate(val)}
sz = len(val)
edge = [[] for _ in range(sz)]
for b, c in BC:
    s = vtoi[b]
    t = vtoi[c]
    edge[s].append(t)

vis = [0] * sz
goal = list(range(sz))


def dfs(s, p):
    vis[s] = 1
    for t in edge[s]:
        if vis[t]:
            goal[s] = max(goal[s], goal[t])
            continue
        if t == p:
            continue
        dfs(t, s)
        goal[s] = max(goal[s], goal[t])


for i in range(sz):
    if vis[i]:
        continue
    dfs(i, -1)

ans = N * (N + 1) // 2
for i, g in enumerate(goal):
    ans -= val[i]
    ans += val[g]
print(ans)
0