結果

問題 No.1060 素敵な宝箱
コンテスト
ユーザー None
提出日時 2021-05-01 04:42:08
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 62 ms / 2,000 ms
コード長 554 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 275 ms
コンパイル使用メモリ 85,356 KB
実行使用メモリ 167,156 KB
最終ジャッジ日時 2026-04-02 15:09:34
合計ジャッジ時間 3,002 ms
ジャッジサーバーID
（参考情報） 		judge4_1 / judge5_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

N,M=map(int, input().split())
A=[]
S=[0]*M
for _ in range(N):
    tmp=[]
    for i,a in enumerate(map(int, input().split())):
        tmp.append(a)
        S[i]+=a
    A.append(tmp)
B=[]
for i in range(N):
    tmp=0
    for j,a in enumerate(A[i]):
        tmp+=S[j]*a
    B.append((tmp,A[i]))

B.sort(key=lambda x:x[0], reverse=True)

sign=0
P,Q=[0]*M,[0]*M
for _,aa in B:
    for i,a in enumerate(aa):
        if sign==0:
            P[i]+=a
        else:
            Q[i]+=a
    sign=1-sign

res=0
for i in range(M):
    res+=P[i]**2-Q[i]**2
print(res)
0