結果
問題 | No.758 VVVVV |
ユーザー |
![]() |
提出日時 | 2018-12-01 20:57:09 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 559 ms / 2,000 ms |
コード長 | 1,185 bytes |
コンパイル時間 | 121 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 31,236 KB |
最終ジャッジ日時 | 2024-10-14 01:45:20 |
合計ジャッジ時間 | 7,328 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
mod = 10**9 + 7 def combination(n, r): assert 0 <= r <= n if n - r < r: r = n - r if r == 0: return 1 if r == 1: return n numerator = [n - r + i + 1 for i in range(r)] denominator = [i + 1 for i in range(r)] for p in range(2, r + 1): pivot = denominator[p - 1] if pivot > 1: offset = (n - r) % p for i in range(p - 1, r, p): numerator[i - offset] //= pivot denominator[i] //= pivot result = 1 for i in range(r): if numerator[i] > 1: result *= numerator[i] return result def main(): n = int(input()) assert 1 <= n <= 10**5 graph = [[] for _ in range(n)] for _ in range(n - 1): a, b = map(int, input().split()) assert 1 <= a <= n assert 1 <= b <= n graph[a - 1].append(b - 1) graph[b - 1].append(a - 1) if n == 1: print(1) exit() leaf_cnt = len(list(filter(lambda x: len(x) == 1, graph[1:]))) print(combination(n - 1, leaf_cnt - 1) * combination(n - 2, leaf_cnt - 1) // leaf_cnt % mod) if __name__ == "__main__": main()