結果

問題 No.1507 Road Blocked
コンテスト
ユーザー ああいい
提出日時 2021-12-31 19:18:47
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 658 ms / 2,000 ms
コード長 623 bytes
コンパイル時間 490 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 47,488 KB
最終ジャッジ日時 2024-10-08 20:57:25
合計ジャッジ時間 18,501 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10 ** 6)

N = int(input())
P = 998244353

G = [[] for _ in range(N + 1)]
for _ in range(N-1):
    u,v = map(int,input().split())
    G[u].append(v)
    G[v].append(u)

memo = [0] * (N + 1)
ans = 0
def dfs(now = 1,parent = 0):
    global ans
    if now != 1 and len(G[now]) == 1:
        ans = (ans + N - 1) % P
        return 1
    count = 1
    for v in G[now]:
        if v != parent:
            count += dfs(v,now)
    ans = (ans + count * (N - count)) % P
    return count
dfs()
inv = pow(N * (N - 1) * (N - 1) // 2,P - 2,P)
ans = N * (N - 1) * (N - 1) // 2 - ans
print(ans * inv % P)
0