結果
問題 | No.140 みんなで旅行 |
ユーザー |
![]() |
提出日時 | 2020-12-27 16:06:15 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 76 ms / 5,000 ms |
コード長 | 1,794 bytes |
コンパイル時間 | 429 ms |
コンパイル使用メモリ | 82,264 KB |
実行使用メモリ | 66,624 KB |
最終ジャッジ日時 | 2024-10-01 14:33:54 |
合計ジャッジ時間 | 2,226 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
class Combination:"""SIZEが10**6程度以下の二項係数を何回も呼び出したいときに使う使い方:comb = Combination(SIZE, MOD)comb(10, 3) => 120"""def __init__(self, N, MOD=10 ** 9 + 7):self.MOD = MODself.__make_factorial_list(N)def __call__(self, n, k):if k < 0 or k > n:return 0res = self.fact[n] * self.inv[k] % self.MODres = res * self.inv[n - k] % self.MODreturn resdef nPk(self, n, k):if k < 0 or k > n:return 0return self.fact[n] * self.inv[n - k] % self.MODdef nHk(self, n, k):if k == 0:return 1return self.__call__(n + k - 1, k)def __make_factorial_list(self, N):self.fact = [1] * (N + 1)self.inv = [1] * (N + 1)MOD = self.MODfor i in range(1, N + 1):self.fact[i] = (self.fact[i - 1] * i) % MODself.inv[N] = pow(self.fact[N], MOD - 2, MOD)for i in range(N, 0, -1):self.inv[i - 1] = (self.inv[i] * i) % MODreturnmod = 10 ** 9 + 7N = int(input())comb = Combination(10000, mod)# スターリング数のtableつくるstir = [[0] * (N + 1) for _ in range(N + 1)]for i in range(N + 1):stir[i][1] = 1stir[i][i] = 1for n in range(2, N + 1):for k in range(2, n + 1):stir[n][k] = stir[n-1][k-1] + k*stir[n-1][k]stir[n][k] %= modans = 0for g in range(1, N + 1):ans += stir[N][g]ans %= modfor G in range(1, N):ng = N - G # ばらばらにする男女tmp = 0for g in range(2, G + 1):C = stir[G][g] * pow(g * (g - 1), ng, mod)tmp += Ctmp %= modans += tmp * comb(N, G)ans %= modprint(ans)