結果
問題 | No.1637 Easy Tree Query |
ユーザー |
![]() |
提出日時 | 2021-08-06 21:25:43 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 497 ms / 2,000 ms |
コード長 | 648 bytes |
コンパイル時間 | 204 ms |
コンパイル使用メモリ | 82,408 KB |
実行使用メモリ | 97,280 KB |
最終ジャッジ日時 | 2024-09-17 03:32:22 |
合計ジャッジ時間 | 12,686 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 33 |
ソースコード
def tree_dfs(g, root=0): s = [root] d = [1] * len(g) order = [] while s: p = s.pop() d[p] = 0 order.append(p) for node in g[p]: if d[node]: s.append(node) return order n, q = map(int, input().split()) g = [[] for _ in range(n)] for _ in range(n - 1): a, b = map(int, input().split()) g[a - 1].append(b - 1) g[b - 1].append(a - 1) order = tree_dfs(g) d = [0] * n d[0] = 0 for v in order[::-1]: d[v] = 1 for node in g[v]: d[v] += d[node] ans = 0 for i in range(q): p, x = map(int, input().split()) ans += d[p - 1] * x print(ans)