結果

問題 No.1637 Easy Tree Query
ユーザー hiragn
提出日時 2022-11-29 23:25:17
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 746 ms / 2,000 ms
コード長 822 bytes
コンパイル時間 216 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 54,292 KB
最終ジャッジ日時 2024-10-07 08:58:00
合計ジャッジ時間 16,838 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
from sys import setrecursionlimit


def main():
    setrecursionlimit(10 ** 6)

    n, q = map(int, input().split())
    adj = defaultdict(list)
    for _ in range(n - 1):
        a, b = map(int, input().split())
        a -= 1
        b -= 1
        adj[a].append(b)
        adj[b].append(a)

    visited = [0] * n  # 0 = False
    subnodes = [0] * n

    def dfs(i):
        visited[i] = True
        res = 1
        for j in adj[i]:
            if visited[j]:
                continue
            res += dfs(j)
        subnodes[i] = res
        return res

    dfs(0)

    ans = []
    tmp = 0
    for _ in range(q):
        p, x = map(int, input().split())
        tmp += subnodes[p - 1] * x
        ans.append(tmp)
    print(*ans, sep='\n')


if __name__ == "__main__":
    main()
0