Submission #287390

---

Source Code Expand

num, pos = gets.chomp.split(" ")
pos      = pos.to_i
dia      = gets.chomp.split(" ")
num_d    = dia.count("1")
edge = Array.new
while line = gets
  a, b = line.split(" ")
  edge.push( [ a.to_i-1, b.to_i-1 ] )
end
arr = Array.new
dia.each_with_index{|e,i|
  arr[i] = Hash.new
  arr[i]["d"] = ( e == "1" )
}
edge.each{|a,b|
  if arr[a]["children"] == nil
    arr[a]["children"] = [ b ]
  else
    arr[a]["children"].push(b)
  end

  if arr[b]["parent"] == nil
    arr[b]["parent"] = a
  end
}

begin
  cut = 0
  arr.each_with_index{|n,i|
    if n["children"] == nil && n["d"] == false
      arr[i] = nil
      arr.each_with_index{|nn,ii|
        next if nn == nil
        if nn["children"] != nil && nn["children"].include?(i)
          nn["children"].delete(i)
          if nn["children"] == []
            nn.delete("children")
          end
        end
      }
      cut += 0
    end
  }
end while cut > 0

root = nil
arr.each_with_index{|e,i|
  next if e == nil
  root = i if e["parent"] == nil
}

@cost  = 0
@first = nil
@last  = nil

def search( arr, num )
  return if arr[num] == nil
  if arr[num]["d"]
    @first = @cost if @first == nil
    @last  = @cost
  end
  return if arr[num]["children"] == nil
  arr[num]["children"].each{|e|
    @cost += 1
    search( arr, e )
  }
end
search( arr, root )
# p [ @first, @last, @cost ] 
puts 2 * ( @last - @first )

Submission Info

Compile Error

./Main.rb:15: warning: shadowing outer local variable - a
./Main.rb:15: warning: shadowing outer local variable - b

Judge Result