結果
| 問題 |
No.872 All Tree Path
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2022-01-19 00:21:45 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 836 ms / 3,000 ms |
| コード長 | 515 bytes |
| コンパイル時間 | 166 ms |
| コンパイル使用メモリ | 81,920 KB |
| 実行使用メモリ | 294,400 KB |
| 最終ジャッジ日時 | 2024-11-23 13:38:32 |
| 合計ジャッジ時間 | 8,425 ms |
|
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 18 |
ソースコード
import sys
sys.setrecursionlimit(10 ** 9)
n = int(input())
edges = [[] for _ in range(n)]
for _ in range(n - 1):
u, v, w = map(int, input().split())
u -= 1
v -= 1
edges[u].append((v, w))
edges[v].append((u, w))
child = [1] * n
ans = 0
def dfs(pos, bpos):
global ans
for npos, w in edges[pos]:
if npos == bpos:
continue
dfs(npos, pos)
ans += child[npos] * (n - child[npos]) * w
child[pos] += child[npos]
dfs(0, -1)
print(ans * 2)