結果
問題 |
No.827 総神童数
|
ユーザー |
|
提出日時 | 2019-04-11 06:45:57 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 921 bytes |
コンパイル時間 | 677 ms |
コンパイル使用メモリ | 82,980 KB |
実行使用メモリ | 109,048 KB |
最終ジャッジ日時 | 2024-07-18 00:29:10 |
合計ジャッジ時間 | 11,011 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 WA * 2 |
other | AC * 25 WA * 1 RE * 10 |
ソースコード
import sys MAX_N = 200005 MOD = 1000000007 def make(): fac[0] = fac[1] = 1 finv[0] = finv[1] = 1 inv[1] = 1 for i in range(2, MAX_N): inv[i] = MOD - inv[MOD % i] * (MOD // i) % MOD fac[i] = fac[i-1] * i % MOD finv[i] = finv[i-1] * inv[i] % MOD def comb(a, b): if a < b: return 0 return fac[a] * finv[b] * finv[a-b] % MOD def dfs(u, p, depth): d[depth] += 1 for v in G[u]: if v != p: dfs(v, u, depth + 1) if __name__ == '__main__': n = int(sys.stdin.readline()) G = [[] for _ in range(n)] fac, finv, inv, d= [0]*MAX_N, [0]*MAX_N, [0]*MAX_N, [0]*MAX_N for i in range(n-1): u, v = map(int, sys.stdin.readline().split()) G[u-1].append(v-1), G[v-1].append(u-1) make() dfs(0, -1, 1) ans = 0 for i in range(n): ans += fac[n-i]*fac[i-1]*d[i]*comb(n, i) % MOD print(ans % MOD)