結果

問題 No.1767 BLUE to RED
コンテスト
ユーザー H3PO4
提出日時 2021-11-26 22:54:59
言語 Python3
(3.14.3 + numpy 2.4.4 + scipy 1.17.1)
コンパイル:
python3 -mpy_compile _filename_
実行:
python3 _filename_
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 635 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 572 ms
コンパイル使用メモリ 20,696 KB
実行使用メモリ 307,316 KB
最終ジャッジ日時 2026-03-19 08:26:37
合計ジャッジ時間 54,058 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other AC * 7 TLE * 14
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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