結果

問題 No.1 道のショートカット
ユーザー shi-mo
提出日時 2016-03-22 02:35:28
言語 Ruby
(3.4.1)
結果
AC  
実行時間 97 ms / 5,000 ms
コード長 1,244 bytes
コンパイル時間 44 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 12,416 KB
最終ジャッジ日時 2024-07-20 16:21:44
合計ジャッジ時間 5,296 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 40
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:57: warning: ambiguous first argument; put parentheses or a space even after `-' operator
Main.rb:31: warning: assigned but unused variable - num_ways
Main.rb:36: warning: assigned but unused variable - pos_head
Syntax OK

ソースコード

diff #

#!ruby

def load_int(io)
  io.gets.to_i
end

def load_int_array(io)
  io.gets.chomp.split.map(&:to_i)
end

def load_problem(io)
  num_cities = load_int(io)
  cost_limit = load_int(io)
  num_ways   = load_int(io)

  s = load_int_array(io)
  t = load_int_array(io)
  y = load_int_array(io)
  m = load_int_array(io)

  ways = []
  num_ways.times do |i|
    ways << { start: s[i], terminal: t[i], cost: y[i], time: m[i] }
  end
  ways_from = ways.group_by{|way| way[:start] }

  [ num_cities, cost_limit, num_ways, ways_from ]
end

def main
  num_cities, cost_limit, num_ways, ways_from = load_problem(STDIN)

  candidates = [ { pos: 1, cost: 0, time: 0 } ]
  result = nil

pos_head = 1
  until candidates.empty?
    candidates.sort_by!{|c| c[:time] }

    current = candidates.shift
    if num_cities == current[:pos]
      result = current
      break
    end

    next unless ways_from[current[:pos]]

    ways_from[current[:pos]].each do |next_way|
      cost = current[:cost] + next_way[:cost]
      time = current[:time] + next_way[:time]
      next if cost_limit < cost
      candidates.unshift({ pos: next_way[:terminal], cost: cost, time: time })
    end
  end

  if result.nil?
    puts -1
    exit 0
  end
  puts result[:time]
end

main
0