結果

問題 No.2200 Weird Shortest Path
ユーザー googol_S0
提出日時 2023-01-28 19:32:33
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 637 ms / 2,000 ms
コード長 637 bytes
コンパイル時間 147 ms
コンパイル使用メモリ 82,400 KB
実行使用メモリ 120,840 KB
最終ジャッジ日時 2024-06-28 22:03:00
合計ジャッジ時間 16,253 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

belong=list()
group=list()
def init(X):
  global belong
  global group
  belong=list(range(X))
  group=[[i] for i in range(X)]

def unite(x,y):
  if len(group[belong[x]])<len(group[belong[y]]):
    x,y=y,x
  if belong[x]==belong[y]:
    return 0
  z=belong[y]
  V=len(group[z])*len(group[belong[x]])
  for i in range(len(group[z])):
    belong[group[z][-1]]=belong[x]
    group[belong[x]].append(group[z][-1])
    del group[z][-1]
  return V

N,M=map(int,input().split())
init(N)
X=[tuple(map(int,input().split())) for i in range(M)]
X.sort(key=lambda x:x[2])
ANS=0
for i in range(M):
  ANS+=X[i][2]*unite(X[i][0]-1,X[i][1]-1)
print(ANS)
0