結果
問題 | No.1637 Easy Tree Query |
ユーザー |
![]() |
提出日時 | 2021-10-03 18:19:22 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 374 ms / 2,000 ms |
コード長 | 1,078 bytes |
コンパイル時間 | 314 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 178,560 KB |
最終ジャッジ日時 | 2024-07-21 13:19:41 |
合計ジャッジ時間 | 14,357 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 33 |
ソースコード
import itertools as iterimport collections as collimport heapq as hqimport bisect as bisfrom decimal import Decimal as decfrom copy import deepcopy as dcopyimport mathimport syssys.setrecursionlimit(10**6)def input():return sys.stdin.readline().rstrip()def getN():return int(sys.stdin.readline().rstrip())def getNs():return map(int,sys.stdin.readline().rstrip().split())def getList():return list(map(int,sys.stdin.readline().rstrip().split()))def strinps(n):return [sys.stdin.readline().rstrip() for _ in range(n)]pi = 3.141592653589793mod = 10**9+7MOD = 998244353INF = math.infdx = [1,0,-1,0]; dy = [0,1,0,-1]"""Main Code"""n,q = getNs()route = [[] for _ in [0]*n]for _ in [0]*(n-1):a,b = getNs()a -= 1; b -= 1route[a].append(b)route[b].append(a)visited = [False]*ncnt = [0]*ndef dfs(v):res = 1for nv in route[v]:if not(visited[nv]):visited[nv] = Trueres += dfs(nv)cnt[v] = resreturn resvisited[0] = Truedfs(0)ans = 0query = [getList() for _ in [0]*q]for p,x in query:ans += x*cnt[p-1]print(ans)