結果
問題 |
No.2200 Weird Shortest Path
|
ユーザー |
|
提出日時 | 2023-01-29 16:15:05 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 855 bytes |
コンパイル時間 | 327 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 93,340 KB |
最終ジャッジ日時 | 2024-06-29 13:08:38 |
合計ジャッジ時間 | 21,165 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | WA * 2 |
other | WA * 42 RE * 2 |
ソースコード
import sys sys.setrecursionlimit(10 ** 8) N,M = map(int,input().split()) edge = [] C = 15 mask = 1 << C for _ in range(M): a,b,w = map(int,input().split()) edge.append((w,(a << C) + b)) edge.sort() ans = 0 parent = [-1] * N def find(i): if parent[i] < 0:return i parent[i] = find(parent[i]) return parent[i] def unite(i,j): I = find(i) J = find(j) if I == J:return False if parent[I] < parent[J]: I,J = J,I parent[J] += parent[I] parent[I] = J #parent[i] = J #parent[j] = J return True def isSame(i,j): return find(i) == find(j) for w,k in edge: a = k // mask b = k % mask #print(w,a,b) a -= 1 b -= 1 A = find(a) B = find(b) if A == B:continue na = -parent[A] nb = -parent[B] ans += w * na * nb unite(a,b) print(na,nb,w) print(ans)