結果

問題 No.92 逃走経路
ユーザー tshig
提出日時 2016-08-09 12:14:16
言語 Ruby
(3.4.1)
結果
AC  
実行時間 1,087 ms / 5,000 ms
コード長 615 bytes
コンパイル時間 49 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 46,592 KB
最終ジャッジ日時 2024-12-26 06:15:41
合計ジャッジ時間 7,016 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

class Calc0092
  def initialize(args)
    args = args.map { |l| l.chomp.split(/\s+/) }
    @n, @m, @k = args.shift.map(&:to_i)
    @roads = args.shift(@m).map { |l| l.map(&:to_i) }
    @ds = args.shift.map(&:to_i)
  end

  def run
    h = @roads.group_by { |r| r[2] }.map { |k, v|
      [k, v.flat_map { |w| [[w[0], w[1]], [w[1], w[0]]] }]
    }.to_h

    cands = h[@ds.shift].map(&:last).uniq
    @ds.each do |d|
      cands = h[d].select { |w| cands.include?(w.first) }.map(&:last).uniq
    end
    [cands.size, cands.sort.join(' ')].join("\n")
  end
end

puts Calc0092.new(STDIN.readlines).run if __FILE__ == $0
0