結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,349 ms
55,424 KB
testcase_01 AC 1,389 ms
55,552 KB
testcase_02 AC 1,396 ms
55,680 KB
testcase_03 RE -
testcase_04 AC 89 ms
12,288 KB
testcase_05 AC 1,402 ms
55,424 KB
testcase_06 AC 1,402 ms
55,680 KB
testcase_07 AC 1,364 ms
55,552 KB
testcase_08 AC 180 ms
17,024 KB
testcase_09 AC 182 ms
17,280 KB
testcase_10 AC 178 ms
17,152 KB
testcase_11 AC 178 ms
17,152 KB
testcase_12 AC 176 ms
17,152 KB
testcase_13 AC 89 ms
12,160 KB
testcase_14 AC 88 ms
12,288 KB
testcase_15 AC 87 ms
12,160 KB
testcase_16 AC 89 ms
12,032 KB
testcase_17 AC 86 ms
12,288 KB
testcase_18 AC 86 ms
12,032 KB
testcase_19 AC 87 ms
12,032 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

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
0