結果

問題 No.845 最長の切符
ユーザー 👑 obakyan
提出日時 2019-07-06 20:24:04
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 205 ms / 3,000 ms
コード長 2,154 bytes
コンパイル時間 342 ms
コンパイル使用メモリ 6,812 KB
実行使用メモリ 11,776 KB
最終ジャッジ日時 2024-10-01 08:44:54
合計ジャッジ時間 2,237 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

local mce, mfl, msq, mmi, mma = math.ceil, math.floor, math.sqrt, math.min, math.max

local function heldkarp_prepare(n)
  local tot = 1
  local alltask = {}
  local stagetask = {}
  for i = 1, n do
    tot = tot * 2
    alltask[i] = {}
    stagetask[i] = {}
  end
  for i = 1, n do
    for j = 1, tot - 1 do
      alltask[i][j] = 0
    end
  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[i][tmp] = 0
    tmp = tmp * 2
  end
  return tot, alltask, stagetask
end

local function heldkarp_doall(n, alltask, stagetask, cost)
  for stage = 1, n - 1 do
    local stlen = #stagetask[stage]
    for i_stagetask = 1, stlen do
      local used = stagetask[stage][i_stagetask]
      local t_used = used
      for i = 1, n do
        if t_used % 2 == 1 then
          local val = alltask[i][used]
          local mul = 1
          local tmp = used
          for j = 1, n do
            if tmp % 2 == 0 then
              if cost[i][j] then
                alltask[j][used + mul] = mma(alltask[j][used + mul], val + cost[i][j])
              else
                alltask[i][used + mul] = mma(alltask[i][used + mul], val)
              end
            else
              tmp = tmp - 1
            end
            tmp = tmp / 2
            mul = mul * 2
          end
          t_used = t_used - 1
        end
        t_used = t_used / 2
      end
    end
  end
end

local function heldkarp_getresult(n, tot, alltask)
  local ret = 0
  for i = 1, n do
    ret = mma(ret, alltask[i][tot - 1])
  end
  return ret
end

local n, m = io.read("*n", "*n")
local cost = {}
for i = 1, n do
  cost[i] = {}
end
for i = 1, m do
  local a, b, c = io.read("*n", "*n", "*n")
  if cost[a][b] then
    cost[a][b] = mma(cost[a][b], c)
  else
    cost[a][b] = c
  end
  cost[b][a] = cost[a][b]
end
local tot, alltask, stagetask = heldkarp_prepare(n)
heldkarp_doall(n, alltask, stagetask, cost)
print(heldkarp_getresult(n, tot, alltask))
0