結果

問題 No.1767 BLUE to RED
ユーザー H3PO4H3PO4
提出日時 2021-11-26 22:54:59
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 635 bytes
コンパイル時間 271 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 229,556 KB
最終ジャッジ日時 2024-06-29 18:22:44
合計ジャッジ時間 42,814 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 933 ms
56,524 KB
testcase_01 AC 934 ms
56,532 KB
testcase_02 AC 941 ms
56,780 KB
testcase_03 AC 945 ms
56,776 KB
testcase_04 AC 935 ms
56,784 KB
testcase_05 AC 941 ms
56,656 KB
testcase_06 AC 930 ms
56,648 KB
testcase_07 AC 932 ms
56,788 KB
testcase_08 AC 931 ms
56,780 KB
testcase_09 AC 1,592 ms
117,484 KB
testcase_10 AC 1,802 ms
139,572 KB
testcase_11 AC 1,658 ms
123,048 KB
testcase_12 AC 1,602 ms
117,468 KB
testcase_13 AC 1,912 ms
149,108 KB
testcase_14 TLE -
testcase_15 TLE -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 TLE -
testcase_20 TLE -
testcase_21 TLE -
testcase_22 TLE -
testcase_23 TLE -
権限があれば一括ダウンロードができます

ソースコード

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