結果

問題 No.3373 Partial Complement Tree
コンテスト
ユーザー i_taku
提出日時 2026-01-21 08:33:02
言語 PyPy3
(7.3.17)
結果
WA  
実行時間 -
コード長 683 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 362 ms
コンパイル使用メモリ 82,708 KB
実行使用メモリ 353,044 KB
最終ジャッジ日時 2026-01-21 08:33:17
合計ジャッジ時間 12,341 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 4 WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
sys.setrecursionlimit(10**7)


def dfs(u, p):
    global ans
    for v in g[u]:
        if v == p:
            continue
        dfs(v, u)
        for i in range(4):
            ans += dp[i][u] * dp[2 - i][v]
            ans %= mod
        for i in range(1, 4):
            dp[i][u] += dp[i - 1][v]
            dp[i][u] %= mod

T = int(input())
mod = 998244353
for _ in range(T):
    N = int(input())
    g = [[] for _ in range(N)]
    for _ in range(N - 1):
        u, v = map(int, input().split())
        u, v = u - 1, v - 1
        g[u].append(v)
        g[v].append(u)
    dp = [[0] * N for _ in range(4)]
    dp[0] = [1] * N
    ans = 0
    dfs(0, -1)
    print(ans)
0