Submission #3777038

---

Source Code Expand

N = gets.to_i
G = Array.new(N+1){[]}
(2 .. N).each do |child|
  parent = gets.to_i
  G[parent] << child
end
class << G
  def dfs
    salary = Array.new(N+1, 0)
    used = Array.new(N+1)
    stack = [1]
    until stack.empty?
      v = stack.last
      used[v] = 0 unless used[v]
      offset = used[v]
      
      if offset < G[v].size
        u = G[v][offset]
        stack << u
        used[v] += 1
      else
        if G[v].empty?
          salary[v] = 1
        else
          salary[v] = G[v].map{|u| salary[u]}.minmax.inject(&:+) + 1
        end
        stack.pop
      end
    end
    salary[1]
  end
end

puts G.dfs

Submission Info

Compile Error

./Main.rb:25: warning: shadowing outer local variable - u

Judge Result