Laskettelukeskus

def calculate(current)
  children = $nodes[current]
  #puts "#{current}, #{children}"
  if !children.empty?
    t = 0
    children.each { |c| t += calculate(c) }
    if t >= $requireds[current - 1]
      return t
    end
  end
  return $requireds[current - 1] # shift
end

count = gets.chomp.to_i
 

$nodes = Hash.new { Array.new }

(count - 1).times do
  a, b = gets.chomp.split(" ").map(&:to_i)
  $nodes[a] |= [b];
end

$requireds = gets.chomp.split(" ").map(&:to_i)

# hash stores parent
# parent through child until child has no child, then take thing and return with alg
puts calculate(1)

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 2

Verdict: ACCEPTED

Test 5

Group: 2

Verdict: ACCEPTED

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:6:in `each': stack level too deep (SystemStackError)
	from input/code.rb:6:i...

Test 10

Group: 1, 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: ACCEPTED

Test 12

Group: 1, 2

Verdict: ACCEPTED

Test 13

Group: 2

Verdict: ACCEPTED