結果

問題 No.2214 Products on Tree
ユーザー ShirotsumeShirotsume
提出日時 2022-11-30 00:00:53
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 1,363 bytes
コンパイル時間 448 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 166,380 KB
最終ジャッジ日時 2024-10-07 09:00:30
合計ジャッジ時間 23,952 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
54,144 KB
testcase_01 AC 43 ms
53,888 KB
testcase_02 AC 42 ms
53,760 KB
testcase_03 AC 1,012 ms
153,756 KB
testcase_04 AC 1,062 ms
157,904 KB
testcase_05 AC 990 ms
148,324 KB
testcase_06 AC 688 ms
126,216 KB
testcase_07 AC 269 ms
92,016 KB
testcase_08 AC 214 ms
87,292 KB
testcase_09 AC 319 ms
98,236 KB
testcase_10 AC 665 ms
124,216 KB
testcase_11 AC 387 ms
104,208 KB
testcase_12 AC 455 ms
106,716 KB
testcase_13 AC 1,110 ms
158,408 KB
testcase_14 AC 1,070 ms
157,752 KB
testcase_15 AC 1,132 ms
158,504 KB
testcase_16 AC 1,110 ms
157,716 KB
testcase_17 AC 1,081 ms
156,636 KB
testcase_18 AC 756 ms
132,228 KB
testcase_19 AC 843 ms
138,496 KB
testcase_20 AC 753 ms
133,332 KB
testcase_21 AC 600 ms
120,644 KB
testcase_22 AC 1,041 ms
154,992 KB
testcase_23 AC 42 ms
53,888 KB
testcase_24 AC 43 ms
54,272 KB
testcase_25 AC 42 ms
54,400 KB
testcase_26 AC 42 ms
53,888 KB
testcase_27 AC 43 ms
53,888 KB
testcase_28 RE -
testcase_29 AC 290 ms
97,024 KB
testcase_30 AC 769 ms
166,380 KB
testcase_31 AC 1,077 ms
164,868 KB
testcase_32 AC 1,067 ms
162,272 KB
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 AC 805 ms
155,796 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import deque, Counter
#↓これ忘れたやつ
#sys.setrecursionlimit(5 * 10 ** 5)
from pypyjit import set_param
set_param('max_unroll_recursion=-1')
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
mi = lambda: map(int, input().split())
li = lambda: list(mi())
inf = 2 ** 63 - 1
mod = 998244353

def solve(n, AB):
    dp = [[0] * 2 for _ in range(n)]
    s = set()
    graph = [[] for _ in range(n)]
    for a, b in AB:
        graph[a - 1].append(b - 1)
        graph[b - 1].append(a - 1)
    s.add(0)
    def dfs(now):
        dp[now][0] = dp[now][1] = 1
        des = []
        for to in graph[now]:
            if to not in s:
                des.append(to)
                s.add(to)
                dfs(to)

        for to in des:
            dp[now][0] *= (dp[to][0] + dp[to][1])
            dp[now][0] %= mod

        P = dp[now][0]
        dp[now][1] = P
        for to in des:
            P *= pow(dp[to][0] + dp[to][1], mod - 2, mod)
            P %= mod
            P *= dp[to][1]
            P %= mod
            dp[now][1] += P
            dp[now][1] %= mod
            P *= pow(dp[to][1], mod - 2, mod)
            P %= mod
            P *= (dp[to][0] + dp[to][1])
            P %= mod
    dfs(0)
    return dp[0][1]
        


n = ii()

AB = [li() for _ in range(n - 1)]

print(solve(n, AB))
0