結果
問題 | No.623 fudan no modulus to tigau |
ユーザー | rlangevin |
提出日時 | 2023-01-29 19:40:54 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 23 ms / 2,000 ms |
コード長 | 728 bytes |
コンパイル時間 | 1,654 ms |
コンパイル使用メモリ | 11,008 KB |
実行使用メモリ | 9,124 KB |
最終ジャッジ日時 | 2023-09-12 02:17:41 |
合計ジャッジ時間 | 1,723 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 21 ms
8,932 KB |
testcase_01 | AC | 21 ms
8,772 KB |
testcase_02 | AC | 22 ms
8,908 KB |
testcase_03 | AC | 21 ms
8,772 KB |
testcase_04 | AC | 21 ms
8,780 KB |
testcase_05 | AC | 23 ms
9,124 KB |
testcase_06 | AC | 23 ms
9,032 KB |
testcase_07 | AC | 20 ms
8,676 KB |
testcase_08 | AC | 21 ms
8,876 KB |
testcase_09 | AC | 20 ms
8,720 KB |
testcase_10 | AC | 21 ms
8,856 KB |
testcase_11 | AC | 21 ms
8,708 KB |
testcase_12 | AC | 20 ms
8,660 KB |
testcase_13 | AC | 19 ms
8,776 KB |
ソースコード
import sys sys.setrecursionlimit(10**7) from functools import lru_cache mod = 998244353 @lru_cache(maxsize=None) def f(t, i, a, b, x): if i == 0: return 1 if i == 1: return x if t == 1: return (f(T[a], a, A[a], B[a], x) + f(T[b], b, A[b], B[b], x)) % mod elif t == 2: return a * f(T[b], b, A[b], B[b], x) % mod else: return (f(T[a], a, A[a], B[a], x) * f(T[b], b, A[b], B[b], x)) % mod N = int(input()) T, A, B = [0] * 2, [0] * 2, [0] * 2 for i in range(N - 1): t, a, b = map(int, input().split()) T.append(t) A.append(a) B.append(b) Q = int(input()) X = list(map(int, input().split())) for x in X: print(f(T[N], N, A[N], B[N], x))