AtCoder Typical Contest 001の深さ優先探索の問題を解いています.
最初に私が作ったプログラムは以下の通りです.
h, w = gets.split.map(&:to_i) seen = h.times.map { w.times.map { false } } #到達しているかどうかの変数 # スタート地点の座標 sh = 0 sw = 0 # ゴール地点の座標 gh = 0 gw = 0 todo = [] moves = [[1, 0], [0, 1], [-1, 0], [0, -1]] // 恐らくここが誤作動している. roots = h.times.map do |i| line = gets.chomp if j = line.index('s') sh = i sw = j elsif j = line.index('g') gh = i gw = j end line.chars end //スタックを用いた深さ優先探索の実装 todo.push([sh, sw]) while !todo.empty? do sh, sw = todo.pop seen[sh][sw] = true moves.each do |di, dj| nh = sh + di nw = sw + dj next if nh < 0 || nw < 0 || nh >= h || nw >= w next if seen[nh][nw] || roots[nh][nw] == '#' todo.push([nh, nw]) end end if seen[gh][gw] puts "Yes" else puts "No" end
入力は
4 5 s#### ....# ##### #...g
のようなもので最初に1行目には南北の長さ 東西の長さ
として与えられています.
この私が作ったプログラムでは4問のミスが発生しておりテストケースが通りませんでした.
そこで,上記のプログラムにコメントアウトしている部分
格子状の街の区画を変数に代入し,スタート地点(s)ゴール地点(g)の場所を記録しているプログラムの部分
// 恐らくここが誤作動している. roots = h.times.map do |i| line = gets.chomp if j = line.index('s') sh = i sw = j elsif j = line.index('g') gh = i gw = j end line.chars end
を次のように変えました.
roots = h.times.map do |i| gets.chomp.chars end h.times do |i| w.times do |j| if roots[i][j] == 's' sh = i sw = j elsif roots[i][j] == 'g' gh = i gw = j end end end
このようにすることでテストケースは問題なく通りACになりました.
他の部分のコードは変えていないのでこの変えた部分で問題が発生していると考えました.
しかし,テストケースが通らない場合の方法ではどういった入力がされた場合にうまく動かないのかわかりませんでした.
ぜひご教授願えればと思います.
よろしくお願いします.
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。