#1029824 (PyPy3) No.2980 Planar Tree 2

• 問題: No.2980 Planar Tree 2
• ユーザー: lif4635
• 提出日時: 2024-12-04 00:17:41
• 言語: PyPy3 (7.3.15)
• 結果: AC
• 実行時間: 492 ms / 2,000 ms
• コード長: 2,029 bytes
• コンパイル時間: 397 ms
• コンパイル使用メモリ: 82,504 KB
• 実行使用メモリ: 188,788 KB
• 最終ジャッジ日時: 2024-12-04 00:21:10
• 合計ジャッジ時間: 10,880 ms
• ジャッジサーバーID （参考情報）: judge3 / judge4

ファイルパターン	結果
other	AC * 30

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
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)
ans = comb.Fi(n-1)
for i in range(n):
    ans *= comb.F(len(edge[i]))
    ans %= mod
print(ans)
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX