Laskettelukeskus

class Node
  attr_accessor :leaf, :parent, :count, :total
  def initialize(parent, count)
    @parent = parent
    @count = count
    @total = 0
    @leaf = true
    @children = []
  end

  def make(count)
    n = Node.new(self, count)
    @children << n
    @leaf = false
    return n
  end
  
  def add(num)
    @total += num
    if !parent.nil?
      parent.add(num)
    end
  end

  def go_down_and_inc_as_needed
    @children.each { |child| child.go_down_and_inc_as_needed }
    if count > total
      add(count - total)
    end
  end
end

count = gets.chomp.to_i
rows = [[1, 1]]
(count-1).times do
  rows << gets.chomp.split(" ").map(&:to_i)
end
c = gets.chomp.split(" ").map(&:to_i)
c.each_with_index do |x, i|
  rows[i] << x
end

#puts rows.to_s
nodes = {}
nodes[1] = Node.new(nil, rows[0][2])
rows[1..-1].each do |row|
  nodes[row[1]] = nodes[row[0]].make(row[2])
end
#puts nodes
nodes[1].go_down_and_inc_as_needed
puts nodes[1].total
# walk from 1, dfs, calc running total, when going up if running total < c then self.add_recursive

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: WRONG ANSWER

Test 3

Group: 1, 2

Verdict: WRONG ANSWER

Test 4

Group: 2

Verdict: WRONG ANSWER

Test 5

Group: 2

Verdict: WRONG ANSWER

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:26:in `each': stack level too deep (SystemStackError)
	from input/code.rb:26...

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