結果
| 問題 |
No.1637 Easy Tree Query
|
| コンテスト | |
| ユーザー |
nephrologist
|
| 提出日時 | 2021-08-06 21:45:08 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 414 ms / 2,000 ms |
| コード長 | 1,017 bytes |
| コンパイル時間 | 270 ms |
| コンパイル使用メモリ | 82,464 KB |
| 実行使用メモリ | 96,592 KB |
| 最終ジャッジ日時 | 2024-09-17 03:49:27 |
| 合計ジャッジ時間 | 11,054 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 33 |
ソースコード
n, q = map(int, input().split())
graph = [[] for _ in range(n)]
for _ in range(n - 1):
a, b = map(int, input().split())
a, b = a - 1, b - 1
graph[a].append(b)
graph[b].append(a)
# graph and n is necessary
def dfs(start):
par = [-1] * n
depth = [-1] * n
size = [0] * n
stack = []
stack.append(~start)
stack.append(start)
depth[start] = 0
while stack:
v = stack.pop()
if v >= 0:
d = depth[v]
for u in graph[v]:
if par[v] == u:
continue
par[u] = v
depth[u] = d + 1
stack.append(~u)
stack.append(u)
else:
a = ~v
size[a] = 1
for u in graph[a]:
if u == par[a]:
continue
size[a] += size[u]
return size
size = dfs(0)
ans = 0
for i in range(q):
p, x = map(int, input().split())
p -= 1
ans += size[p] * x
print(ans)
nephrologist