結果

問題 No.2980 Planar Tree 2
コンテスト
ユーザー lif4635
提出日時 2024-12-04 00:14:54
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 2,555 bytes
コンパイル時間 463 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 200,008 KB
最終ジャッジ日時 2024-12-04 00:17:56
合計ジャッジ時間 22,551 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

def II() -> int : return int(input())
def MI() -> int : return map(int, input().split())
def TI() -> tuple[int] : return tuple(MI())
def LI() -> list[int] : return list(MI())
#str-input
def SI() -> str : return input()
def MSI() -> str : return input().split()
def SI_L() -> list[str] : return list(SI())
def SI_LI() -> list[int] : return list(map(int, SI()))
#multiple-input
def LLI(n) -> list[list[int]]: return [LI() for _ in range(n)]
def LSI(n) -> list[str]: return [SI() for _ in range(n)]
#1-indexを0-indexでinput
def MI_1() -> int : return map(lambda x:int(x)-1, input().split())
def TI_1() -> tuple[int] : return tuple(MI_1())
def LI_1() -> list[int] : return list(MI_1())

n = II()

edge = [set() for i in range(n)]
for i in range(n-1):
    u,v = MI_1()
    edge[u].add(v)
    edge[v].add(u)

mod = 998244353
class Comb: #combination列挙
    def __init__(self,lim,mod = mod):
        """
        mod : prime指定
        lim以下のmodでcomdination計算
        """
        self.fac = [1,1]
        self.inv = [1,1]
        self.finv = [1,1]
        self.mod = mod
        for i in range(2,lim+1):
            self.fac.append(self.fac[i-1]*i%self.mod)
            self.inv.append(-self.inv[mod%i]*(mod//i)%self.mod)
            self.finv.append(self.finv[i-1]*self.inv[i]%self.mod)

    def F(self,a):
        return self.fac[a]
    
    def C(self,a,b):
        #自然な拡張
        assert b >= 0, "第2引数の値が負です"
        if a < b: return 0
        if a < 0: return 0
        # 0 <= a ∧ b <= a (b < 0でバグる)
        return self.fac[a]*self.finv[b]*self.finv[a-b]%self.mod
    
    def P(self,a,b):
        assert b >= 0, "第2引数の値が負です"
        if a < b: return 0
        if a < 0: return 0
        return self.fac[a]*self.finv[a-b]%self.mod
    
    def H(self,a,b):
        return self.C(a+b-1,b)
    
    def Fi(self,a):
        return self.finv[a]
    
comb = Comb(n)

def DFS_par(edge): #親の情報をうまく持つDFS
    n = len(edge)
    st = [(0,None,0)]
    size = [1]*n
    ans = 1
    while st:
        now,par,t = st.pop()
        if t == 0:
            st.append((now,par,1))
            #行きがけ
            for chi in edge[now]:
                if chi != par:
                    st.append((chi,now,0))
        else:
            if now == 0:
                return ans
            l = len(edge[now]) #子どもの個数
            ans *= comb.F(l)
            size[par] += size[now]
            pass
            #帰りがけ
    

print(DFS_par(edge)*comb.Fi(n-1)%mod)
0