結果

問題 No.2980 Planar Tree 2
ユーザー lif4635lif4635
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 1,115 ms
175,488 KB
testcase_02 WA -
testcase_03 AC 37 ms
52,864 KB
testcase_04 AC 41 ms
52,736 KB
testcase_05 WA -
testcase_06 AC 1,226 ms
175,112 KB
testcase_07 WA -
testcase_08 AC 1,608 ms
175,460 KB
testcase_09 AC 1,168 ms
168,512 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 AC 1,159 ms
174,080 KB
testcase_13 AC 1,190 ms
174,976 KB
testcase_14 WA -
testcase_15 AC 1,263 ms
175,428 KB
testcase_16 WA -
testcase_17 AC 54 ms
65,152 KB
testcase_18 AC 70 ms
65,664 KB
testcase_19 WA -
testcase_20 WA -
testcase_21 AC 55 ms
65,152 KB
testcase_22 AC 39 ms
54,740 KB
testcase_23 AC 41 ms
55,168 KB
testcase_24 WA -
testcase_25 WA -
testcase_26 AC 539 ms
190,840 KB
testcase_27 AC 528 ms
191,020 KB
testcase_28 AC 541 ms
190,604 KB
testcase_29 WA -
権限があれば一括ダウンロードができます

ソースコード

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