結果
問題 |
No.1060 素敵な宝箱
|
ユーザー |
![]() |
提出日時 | 2021-05-11 08:16:40 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 86 ms / 2,000 ms |
コード長 | 415 bytes |
コンパイル時間 | 187 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 76,928 KB |
最終ジャッジ日時 | 2024-09-21 14:02:11 |
合計ジャッジ時間 | 2,863 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
from functools import reduce add = lambda x,y: [v+w for v,w in zip(x,y)] n,m = map(int,input().split()) a = [list(map(int,input().split())) for _ in range(n)] r = reduce(add,a) p = [sum(rj*aij for rj,aij in zip(r,ai)) for i,ai in enumerate(a)] order = sorted(range(n),key = lambda i:p[i],reverse=1) x = reduce(add,[a[idx] for idx in order[::2]]) print(sum(xi**2 for xi in x) - sum((ri-xi)**2 for ri,xi in zip(r,x)))