結果

問題 No.1060 素敵な宝箱
コンテスト
ユーザー ntuda
提出日時 2025-06-30 12:56:29
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
WA  
実行時間 -
コード長 744 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 226 ms
コンパイル使用メモリ 95,860 KB
実行使用メモリ 85,536 KB
最終ジャッジ日時 2026-07-12 23:36:05
合計ジャッジ時間 4,345 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 1 WA * 23
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

N,M = map(int,input().split())
AA = [list(map(int,input().split())) for _ in range(N)]
used = [0] * N
now = [[0] * M for _ in range(2)]
t1 = 0
maxid = -1
for i,A in enumerate(AA):
    t2 = 0
    for a in A:
        t2 += a * a
    if t1 < t2:
        t1 = t2
        maxid = i
now[0] = AA[maxid][:]
used[maxid] = 1
for j in range(1,N):
    t1 = 0
    for i, A in enumerate(AA):
        if used[i]:
            continue
        t2 = 0
        for k,a in enumerate(A):
            t2 += (now[j%2][k] + a) ** 2
        if t1 < t2:
            t1 = t2
            maxid = i
    for k in range(M):
        now[j % 2][k] += AA[maxid][k]
    used[maxid] = 1

X = 0
Y = 0
for i in range(M):
    X += now[0][i] ** 2
    Y += now[1][i] ** 2
print(X - Y)
0