結果

問題 No.2504 NOT Path Painting
ユーザー suisensuisen
提出日時 2023-07-22 17:02:56
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,471 ms / 2,000 ms
コード長 1,217 bytes
コンパイル時間 203 ms
コンパイル使用メモリ 81,944 KB
実行使用メモリ 236,552 KB
最終ジャッジ日時 2024-09-22 16:41:02
合計ジャッジ時間 20,153 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 62 ms
67,888 KB
testcase_01 AC 860 ms
86,580 KB
testcase_02 AC 781 ms
85,940 KB
testcase_03 AC 910 ms
90,512 KB
testcase_04 AC 968 ms
88,288 KB
testcase_05 AC 939 ms
91,336 KB
testcase_06 AC 875 ms
88,552 KB
testcase_07 AC 925 ms
91,844 KB
testcase_08 AC 896 ms
91,048 KB
testcase_09 AC 881 ms
89,588 KB
testcase_10 AC 895 ms
86,324 KB
testcase_11 AC 907 ms
91,972 KB
testcase_12 AC 935 ms
89,736 KB
testcase_13 AC 761 ms
83,108 KB
testcase_14 AC 731 ms
84,852 KB
testcase_15 AC 957 ms
105,016 KB
testcase_16 AC 982 ms
107,892 KB
testcase_17 AC 1,037 ms
107,280 KB
testcase_18 AC 757 ms
104,304 KB
testcase_19 AC 1,471 ms
236,552 KB
testcase_20 AC 1,055 ms
206,976 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from sys import setrecursionlimit
from typing import List

setrecursionlimit(1000000)

def solve(n: int, g: List[List[int]]):
    m = n * (n + 1) // 2

    ans = 0

    def dfs(u: int, p: int) -> int:
        nonlocal ans

        p_u = m

        sub_u = 1
        for v in g[u]:
            if v == p:
                continue
            sub_v = dfs(v, u)

            p_u -= sub_v * (sub_v + 1) // 2

            p_uv = sub_v * (n - sub_v)
            ans -= m * modinv(m - p_uv) % P

            sub_u += sub_v
        
        if p != -1:
            sub_p = n - sub_u
            p_u -= sub_p * (sub_p + 1) // 2
        
        ans += m * modinv(m - p_u) % P

        return sub_u
    
    dfs(0, -1)

    return ans % P

if __name__ == '__main__':
    P = 998244353

    def modinv(v: int):
        return pow(v, P - 2, P)
    
    answers = []

    T = int(input())
    for _ in range(T):
        n = int(input())
        g = [[] for _ in range(n)]
        for _ in range(n - 1):
            u, v = map(int, input().split())
            u -= 1
            v -= 1
            g[u].append(v)
            g[v].append(u)
        
        answers.append(solve(n, g))
    
    print('\n'.join(map(str, answers)))
0