Laskettelukeskus

class Node
  def initialize(parent)
    @parent = parent
    @leaf = true
    @c = 0
  end

  def make
    child = Node.new(self)
    @leaf = false
    return child
  end

  def leaf?
    return @leaf
  end
  
  def add_to_parents(c)
    @c += c
    if @parent.nil?
      return
    else
      @parent.add_to_parents(c)
    end
  end

  def count
    @c
  end

#  def to_s
#    return "(p=#{@parent}, l=#{@leaf}, c=#{@c})"
#  end
end

total = 0
nodes = []
nodes << Node.new(nil)
(gets.chomp.to_i - 1).times do
  a = gets.chomp.split(" ").map(&:to_i)
  nodes.insert(a[1] - 1, nodes[a[0] - 1].make)
end
notleaf = []
gets.chomp.split(" ").map(&:to_i).each_with_index do |c, index|
  node = nodes[index]
  if node.leaf?
    total += c
    node.add_to_parents(c)
  else
    notleaf.unshift([nodes[index], c])
  end
end

notleaf.each do |node, count|
  delta = count - node.count
  if delta > 0
    node.add_to_parents(delta)
    total += delta
  end
end

puts total

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:41:in `block in <main>': undefined method `make' for nil:NilClass (NoMethodE...

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:41:in `block in <main>': undefined method `make' for nil:NilClass (NoMethodE...

Test 4

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:41:in `block in <main>': undefined method `make' for nil:NilClass (NoMethodE...

Test 5

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:41:in `block in <main>': undefined method `make' for nil:NilClass (NoMethodE...

Test 6

Group: 1, 2

Verdict: RUNTIME ERROR

input/code.rb:41:in `block in <main>': undefined method `make' for nil:NilClass (NoMethodE...

Test 7

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:41:in `block in <main>': undefined method `make' for nil:NilClass (NoMethodE...

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:23:in `add_to_parents': stack level too deep (SystemStackError)
	from input/...

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