結果

問題 No.1767 BLUE to RED
コンテスト
ユーザー H3PO4
提出日時 2021-11-26 22:54:59
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 635 bytes
コンパイル時間 271 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 229,556 KB
最終ジャッジ日時 2024-06-29 18:22:44
合計ジャッジ時間 42,814 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 11 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

from scipy.sparse.csgraph import minimum_spanning_tree
from scipy.sparse import csr_matrix

N, M = map(int, input().split())
A = map(int, input().split())
B = map(int, input().split())
nodes = []
for i, a in enumerate(A):
    nodes.append((a, i))
for i, b in enumerate(B, N):
    nodes.append((b, i))
nodes.sort()

length = [0] * N
frm = list(range(N))
to = [N + M] * N

for i in range(N + M - 1):
    frm.append(nodes[i][1])
    to.append(nodes[i + 1][1])
    length.append(nodes[i + 1][0] - nodes[i][0])

matr = csr_matrix((length, (frm, to)), shape=(N + M + 1, N + M + 1))
T = minimum_spanning_tree(matr).astype(int)
print(T.sum())
0