結果
問題 | No.748 yuki国のお財布事情 |
ユーザー | H3PO4 |
提出日時 | 2021-01-05 09:43:29 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,891 ms / 2,000 ms |
コード長 | 1,117 bytes |
コンパイル時間 | 133 ms |
コンパイル使用メモリ | 12,928 KB |
実行使用メモリ | 84,356 KB |
最終ジャッジ日時 | 2024-10-15 16:31:56 |
合計ジャッジ時間 | 38,942 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
外部呼び出し有り |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1,017 ms
56,952 KB |
testcase_01 | AC | 957 ms
56,828 KB |
testcase_02 | AC | 960 ms
56,828 KB |
testcase_03 | AC | 965 ms
56,312 KB |
testcase_04 | AC | 965 ms
56,956 KB |
testcase_05 | AC | 968 ms
56,824 KB |
testcase_06 | AC | 957 ms
56,960 KB |
testcase_07 | AC | 951 ms
57,088 KB |
testcase_08 | AC | 966 ms
56,568 KB |
testcase_09 | AC | 966 ms
56,948 KB |
testcase_10 | AC | 964 ms
56,568 KB |
testcase_11 | AC | 973 ms
57,208 KB |
testcase_12 | AC | 963 ms
57,088 KB |
testcase_13 | AC | 1,043 ms
59,380 KB |
testcase_14 | AC | 1,078 ms
58,828 KB |
testcase_15 | AC | 1,045 ms
58,912 KB |
testcase_16 | AC | 1,177 ms
62,596 KB |
testcase_17 | AC | 1,709 ms
76,808 KB |
testcase_18 | AC | 1,577 ms
74,492 KB |
testcase_19 | AC | 1,542 ms
75,892 KB |
testcase_20 | AC | 1,465 ms
73,208 KB |
testcase_21 | AC | 1,652 ms
74,488 KB |
testcase_22 | AC | 961 ms
55,812 KB |
testcase_23 | AC | 973 ms
56,316 KB |
testcase_24 | AC | 974 ms
56,444 KB |
testcase_25 | AC | 1,853 ms
80,532 KB |
testcase_26 | AC | 1,891 ms
84,000 KB |
testcase_27 | AC | 1,793 ms
84,356 KB |
testcase_28 | AC | 1,346 ms
67,324 KB |
ソースコード
import sys from collections import defaultdict import numpy as np from scipy.sparse.csgraph import connected_components, minimum_spanning_tree from scipy.sparse import csr_matrix input = sys.stdin.buffer.readline MAX = 10 ** 10 N, M, K = map(int, input().split()) if not M: print(0);exit() ABC = np.array([tuple(map(int, input().split())) for _ in range(M)], dtype=np.int64) ABC[:, :2] -= 1 ans = ABC[:, 2].sum() if K: E = np.array([int(input()) - 1 for _ in range(K)]) frm = ABC[E, 0] to = ABC[E, 1] ans -= ABC[E, 2].sum() connection = csr_matrix(([1] * K, (frm, to)), shape=(N, N)) compress_N, labels = connected_components(connection) ABC[:, :2] = labels[ABC[:, :2]] else: compress_N = N d = defaultdict(lambda: MAX) for a, b, c in ABC: if a == b: continue if a > b: a, b = b, a d[(a, b)] = min(d[(a, b)], c) length, frm, to = [], [], [] for (a, b), c in d.items(): length.append(c) frm.append(a) to.append(b) matr = csr_matrix((length, (frm, to)), shape=(compress_N, compress_N)) T = minimum_spanning_tree(matr) ans -= int(T.sum()) print(ans)