結果
問題 | No.1637 Easy Tree Query |
ユーザー |
|
提出日時 | 2021-12-04 12:40:34 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,000 ms / 2,000 ms |
コード長 | 1,060 bytes |
コンパイル時間 | 606 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 37,504 KB |
最終ジャッジ日時 | 2024-07-06 17:56:54 |
合計ジャッジ時間 | 22,035 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 33 |
ソースコード
N,Q = list(map(int,input().split()))a = [0] * (N-1)b = [0] * (N-1)for i in range(N - 1):a[i],b[i] = list(map(int,input().split()))p = [0] * Qx = [0] * Qfor i in range(Q):p[i],x[i] = list(map(int,input().split()))tree = [[] for _ in range(N+1)]for i in range(N-1):tree[a[i]].append(b[i])tree[b[i]].append(a[i])sub = [1] * (N+1)"""def tansaku(now = 1,parent = 0):if len(tree[now]) == 1:return 1else:for i in tree[now]:if i != parent:sub[now] += tansaku(i,now)return sub[now]tansaku()"""#再帰1000回しかしてくれないq = []flags = [False] * (N+1)q.append((1,0))while len(q):now,parent = q.pop()if flags[now]:for i in tree[now]:if i != parent:sub[now] += sub[i]else:flags[now] = Trueq.append((now,parent))for i in tree[now]:if i != parent:q.append((i,now))_sum = 0for i in range(Q):_sum += sub[p[i]] * x[i]print(_sum)