結果

問題 No.357 品物の並び替え (Middle)
ユーザー 👑 obakyanobakyan
提出日時 2020-03-28 13:07:17
言語 Lua
(LuaJit 2.1.1696795921)
結果
AC  
実行時間 37 ms / 5,000 ms
コード長 1,877 bytes
コンパイル時間 235 ms
コンパイル使用メモリ 5,308 KB
実行使用メモリ 4,500 KB
最終ジャッジ日時 2023-08-30 17:35:02
合計ジャッジ時間 1,108 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
4,376 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 3 ms
4,380 KB
testcase_03 AC 3 ms
4,380 KB
testcase_04 AC 3 ms
4,376 KB
testcase_05 AC 4 ms
4,380 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 1 ms
4,376 KB
testcase_08 AC 3 ms
4,376 KB
testcase_09 AC 26 ms
4,376 KB
testcase_10 AC 8 ms
4,380 KB
testcase_11 AC 11 ms
4,380 KB
testcase_12 AC 37 ms
4,500 KB
testcase_13 AC 19 ms
4,376 KB
testcase_14 AC 25 ms
4,376 KB
testcase_15 AC 8 ms
4,380 KB
testcase_16 AC 14 ms
4,376 KB
testcase_17 AC 8 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

local mmi, mma = math.min, math.max
local bls, brs = bit.lshift, bit.rshift
local n, m = io.read("*n", "*n")
local tot = bls(1, n)
local cost = {}
for i = 1, n do
  cost[i] = {}
end
for i = 1, m do
  local a, b, s = io.read("*n", "*n", "*n")
  a, b = a + 1, b + 1
  cost[a][b] = s
end

local function bdp_prepare()
  local alltask = {}
  local stagetask = {}
  for i = 1, n do
    stagetask[i] = {}
  end
  for i = 1, tot - 1 do
    -- set empty state
    alltask[i] = 0
  end
  for i = 1, tot - 1 do
    local ti = i
    local cnt = 0
    for j = 1, n do
      if ti % 2 == 1 then
        cnt = cnt + 1
        ti = ti - 1
      end
      ti = ti / 2
    end
    table.insert(stagetask[cnt], i)
  end
  -- set first state
  local tmp = 1
  for i = 1, n do
    alltask[tmp] = 0
    tmp = tmp * 2
  end
  return alltask, stagetask
end

local function bdp_doall(alltask, stagetask)
  for stage = 1, n - 1 do
    local stlen = #stagetask[stage]
    for i_stagetask = 1, stlen do
      local used = stagetask[stage][i_stagetask]
      local usedlist = {}
      do
        local tmp = used
        for j = 1, n do
          if tmp % 2 == 1 then
            table.insert(usedlist, j)
            tmp = tmp - 1
          end
          tmp = tmp / 2
        end
      end
      local mul = 1
      local tmp = used
      for j = 1, n do
        if tmp % 2 == 0 then
          local point = 0
          for k = 1, #usedlist do
            local v = cost[usedlist[k]][j]
            if v then point = point + v end
          end
          alltask[used + mul] = mma(alltask[used + mul], alltask[used] + point)
        else
          tmp = tmp - 1
        end
        tmp = tmp / 2
        mul = mul * 2
      end
    end
  end
end

local function bdp_getresult(alltask)
  return alltask[tot - 1]
end
local at, st = bdp_prepare()
bdp_doall(at, st)
local ret = bdp_getresult(at)
print(ret)
0