結果
| 問題 |
No.827 総神童数
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2022-06-24 19:55:52 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 526 ms / 2,000 ms |
| コード長 | 762 bytes |
| コンパイル時間 | 548 ms |
| コンパイル使用メモリ | 82,432 KB |
| 実行使用メモリ | 103,296 KB |
| 最終ジャッジ日時 | 2024-11-08 16:11:57 |
| 合計ジャッジ時間 | 13,446 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 36 |
ソースコード
n = int(input())
e = [[] for i in range(n)]
for _ in range(n-1):
u,v = [int(x)-1 for x in input().split()]
e[u].append(v)
e[v].append(u)
mod = 10**9+7
### for bigger prime
N = n+5
fact = [1]*N
finv = [1]*N
for i in range(2,N):
fact[i] = (fact[i-1]*i)%mod
finv[-1] = pow(fact[-1],mod-2,mod)
for i in range(1,N)[::-1]:
finv[i-1] = (finv[i]*i)%mod
def nCr(n,r):
if r > n:
return 0
else:
return fact[n]*finv[r]%mod*finv[n-r]%mod
dis = [n]*n
q = [0]
dis[0] = 0
while q:
now = q.pop()
for nex in e[now]:
if dis[nex] > dis[now]+1:
dis[nex] = dis[now]+1
q.append(nex)
ans = 0
for i in range(n):
ans += nCr(n,dis[i]+1)*fact[dis[i]]*fact[n-dis[i]-1]
ans %= mod
print(ans)