結果

問題 No.872 All Tree Path
ユーザー simansiman
提出日時 2021-08-02 18:20:00
言語 Ruby
(3.3.0)
結果
RE  
実行時間 -
コード長 787 bytes
コンパイル時間 173 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 12,416 KB
最終ジャッジ日時 2024-09-16 14:41:02
合計ジャッジ時間 2,992 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 RE -
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

if !ENV['RUBY_THREAD_VM_STACK_SIZE']
  exec({'RUBY_THREAD_VM_STACK_SIZE'=>'10000000'}, '/usr/bin/ruby', $0)
else
  N = gets.to_i
  E = Array.new(N + 1) { [] }
  V = Array.new(N + 1) { [] }

  (N - 1).times do
    u, v, w = gets.split.map(&:to_i)

    E[u] << [v, w]
    E[v] << [u, w]
  end

  def dfs1(u, cnt = 0, parent = -1)
    cnt = 0

    E[u].each do |v, w|
      next if v == parent

      v_cnt = dfs1(v, cnt, u)
      cnt += v_cnt
      V[u] << v_cnt
    end

    cnt + 1
  end

  $ans = 0

  def dfs2(u, v_cnt = 0, parent = -1)
    t_cnt = V[u].sum + v_cnt
    i = 0

    E[u].each do |v, w|
      next if v == parent

      cnt = V[u][i]
      $ans += 2 * w * (N - cnt) * cnt
      dfs2(v, t_cnt - cnt + 1, u)
      i += 1
    end
  end

  dfs1(1)
  dfs2(1)

  puts $ans
end
0