結果

問題 No.1194 Replace
ユーザー tktk_snsn
提出日時 2021-02-08 22:12:33
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 879 bytes
コンパイル時間 67 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 137,924 KB
最終ジャッジ日時 2024-07-06 01:39:20
合計ジャッジ時間 26,963 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other WA * 27
権限があれば一括ダウンロードができます

ソースコード

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, val):
    vis[s] = 1
    for t in edge[s]:
        if vis[t]:
            val = max(val, goal[t])
            continue
        if t == p:
            continue
        dfs(t, s, val)
        val = max(val, goal[t])
    goal[s] = max(goal[s], val)


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

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