結果

問題 No.827 総神童数
ユーザー aaaaaaaaaa2230
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

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)
0