結果

問題 No.2325 Skill Tree
ユーザー cozy_sauna
提出日時 2023-05-28 13:52:07
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 602 ms / 3,000 ms
コード長 2,729 bytes
コンパイル時間 1,699 ms
コンパイル使用メモリ 81,536 KB
実行使用メモリ 126,336 KB
最終ジャッジ日時 2024-12-26 22:07:23
合計ジャッジ時間 21,496 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

from sys import setrecursionlimit, stdin
from collections import defaultdict, deque, Counter
from itertools import permutations, combinations, product
from functools import lru_cache
from bisect import bisect_left, bisect_right
from heapq import heappush, heappop
from copy import copy, deepcopy
from decimal import Decimal
from random import random, randrange

# from pypyjit import set_param
# set_param('max_unroll_recursion=-1')

setrecursionlimit(1 << 20)
readline = stdin.readline
INF = 10 ** 18
MOD = 998244353
# MOD = 1000000007
ALP = 26
''' Input '''
def I(): return int(readline())
def ST(): return readline()[:-1]
def LI(): return list(map(int, readline().split()))
def LII(): return list(map(lambda x: int(x) - 1, readline().split()))
def LF(x, func): return [func() for _ in [0] * x]
def SPI(): return map(int, readline().split())
def SPII(): return map(lambda x: int(x) - 1, readline().split())
def FIE(x): return [readline()[:-1] for _ in [0] * x]

''' Array '''
def cmin(dp, i, x):
    if x < dp[i]: dp[i] = x
def cmax(dp, i, x):
    if x > dp[i]: dp[i] = x

''' Alphabet '''
def alp_a_to_i(s): return ord(s) - ord('a')
def alp_A_to_i(s): return ord(s) - ord('A')
def alp_i_to_a(i): return chr(ord('a') + i)
def alp_i_to_A(i): return chr(ord('A') + i)

''' Other'''
def nynx(y, x, H, W): return [(y + dy, x + dx) for dy, dx in [(-1, 0), (1, 0), (0, -1), (0, 1)] if 0 <= y + dy < H and 0 <= x + dx < W]
def gen(x, *args):
    if len(args) == 1: return [x] * args[0]
    if len(args) == 2: return [[x] * args[1] for _ in [0] * args[0]]
    if len(args) == 3: return [[[x] * args[2] for _ in [0] * args[1]] for _ in [0] * args[0]]
    if len(args) == 4: return [[[[x] * args[3] for _ in [0] * args[2]] for _ in [0] * args[1]] for _ in [0] * args[0]]

''' Output '''
def pprint(E): 
    print()
    for e in E: print(e)
def Yes(): print("Yes")
def No(): print("No")
def YES(): print("YES")
def NO(): print("NO")
def yn(x): print("Yes" if x else "No")
def YN(x): print("YES" if x else "NO")
###############################################################################################

N = I()
node = [[] for _ in range(N)]
LA = []
for i in range(N - 1):
    i += 1
    l, a = SPI()
    a -= 1
    node[a].append((i, l))

COST = []
dist = gen(INF, N)
que = deque([(0, 0)])
while que:
    v, c = que.popleft()
    if dist[v] != INF: continue
    COST.append(c)
    dist[v] = c
    for nx, nx_c in node[v]:
        que.appendleft((nx, max(c, nx_c)))


COST.sort()
# print(COST)
Q = I()
for _ in range(Q):
    c, *A = SPI()
    if c == 1:
        x = A[0]
        ans = bisect_right(COST, x)
        print(ans)
    if c == 2:
        y = A[0] - 1
        ans = dist[y]
        if ans == INF: ans = -1
        print(ans)
0