結果

問題 No.1060 素敵な宝箱
ユーザー 👑 obakyanobakyan
提出日時 2021-07-27 10:22:23
言語 Lua
(LuaJit 2.1.1696795921)
結果
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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 1 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 8 ms
5,376 KB
testcase_13 AC 11 ms
5,376 KB
testcase_14 AC 12 ms
5,376 KB
testcase_15 AC 18 ms
5,376 KB
testcase_16 AC 14 ms
5,376 KB
testcase_17 AC 7 ms
5,376 KB
testcase_18 AC 10 ms
5,376 KB
testcase_19 AC 8 ms
5,376 KB
testcase_20 AC 16 ms
5,376 KB
testcase_21 AC 14 ms
5,376 KB
testcase_22 AC 15 ms
5,376 KB
testcase_23 AC 9 ms
5,376 KB
testcase_24 AC 23 ms
5,376 KB
testcase_25 AC 23 ms
5,376 KB
testcase_26 AC 23 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

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