結果

問題 No.1060 素敵な宝箱
ユーザー None
提出日時 2021-05-01 04:42:08
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 70 ms / 2,000 ms
コード長 554 bytes
コンパイル時間 318 ms
コンパイル使用メモリ 82,072 KB
実行使用メモリ 77,028 KB
最終ジャッジ日時 2024-07-19 07:28:52
合計ジャッジ時間 3,001 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

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