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
  nodes << nodes[gets.split(" ")[0].to_i - 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:40:in `block in <main>': undefined method `make' for nil:NilClass (NoMethodE...

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

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

Test 4

Group: 2

Verdict: RUNTIME ERROR

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

Test 5

Group: 2

Verdict: RUNTIME ERROR

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

Test 6

Group: 1, 2

Verdict: RUNTIME ERROR

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

Test 7

Group: 2

Verdict: RUNTIME ERROR

input/code.rb:40: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