提出 #66579651

---

ソースコード 拡げる

# frozen_string_literal: true

n = gets.to_i
x = gets.split.map(&:to_i)
uvw = Array.new(n - 1) { gets.split.map(&:to_i) }

g = Array.new(n) { [] }

uvw.each do |u, v, w|
  u -= 1
  v -= 1
  g[u] << [v, w]
  g[v] << [u, w]
end
@ans = 0
def dfs(p, v, x, g)
  g[v].each do |e|
    nv, w = e
    next if nv == p

    diff = dfs(v, nv, x, g)
    x[v] += diff
    @ans += diff.abs * w
  end
  x[v]
end
dfs(-1, 0, x, g)
puts @ans

提出情報

ジャッジ結果