結果

問題 No.1060 素敵な宝箱
ユーザー ntuda
提出日時 2025-06-30 12:56:29
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 744 bytes
コンパイル時間 1,338 ms
コンパイル使用メモリ 82,340 KB
実行使用メモリ 77,316 KB
最終ジャッジ日時 2025-06-30 12:56:35
合計ジャッジ時間 5,103 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 1 WA * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

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