結果
問題 |
No.1060 素敵な宝箱
|
ユーザー |
|
提出日時 | 2020-05-31 21:51:39 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 461 bytes |
コンパイル時間 | 361 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 44,600 KB |
最終ジャッジ日時 | 2024-11-18 21:14:17 |
合計ジャッジ時間 | 16,919 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 1 WA * 23 |
ソースコード
import numpy as np N, M = map(int, input().split()) A = [list(map(int, input().split())) for _ in range(N)] A = np.array(A) me = np.zeros(M) E = np.zeros(M) a = np.sum(A, axis=1) for i in range(int(N/2)): max_A = np.argmax(a) me += A[max_A] A[max_A] = 0 a[max_A] = 0 max_A = np.argmax(a) E += A[max_A] A[max_A] = 0 a[max_A] = 0 me = np.power(me, 2) E = np.power(E, 2) result = np.sum(me) - np.sum(E) print(int(result))