結果

問題 No.1060 素敵な宝箱
ユーザー 👑 obakyan
提出日時 2021-07-27 10:22:23
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 23 ms / 2,000 ms
コード長 751 bytes
コンパイル時間 395 ms
コンパイル使用メモリ 5,376 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-23 05:57:37
合計ジャッジ時間 1,715 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

local n, m = io.read("*n", "*n")
local a = {}
for i = 1, n do
  a[i] = {}
  for j = 1, m do
    a[i][j] = io.read("*n")
  end
end
local asum = {}
for i = 1, m do
  local v = 0
  for j = 1, n do
    v = v + a[j][i]
  end
  asum[i] = v
end
local scoreidx = {}
local score = {}
for i = 1, n do
  local s = 0
  for j = 1, m do
    s = s + asum[j] * a[i][j]
  end
  score[i] = s
  scoreidx[i] = i
end
table.sort(scoreidx, function(x, y) return score[x] > score[y] end)
local p1, p2 = {}, {}
for i = 1, m do
  p1[i], p2[i] = 0, 0
end
for i = 1, n do
  local t = i % 2 == 1 and p1 or p2
  local pos = scoreidx[i]
  for j = 1, m do
    t[j] = t[j] + a[pos][j]
  end
end
local ret = 0
for i = 1, m do
  ret = ret + p1[i] * p1[i] - p2[i] * p2[i]
end
print(ret)
0