結果
| 問題 |
No.140 みんなで旅行
|
| コンテスト | |
| ユーザー |
tktk_snsn
|
| 提出日時 | 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 = MOD
self.__make_factorial_list(N)
def __call__(self, n, k):
if k < 0 or k > n:
return 0
res = self.fact[n] * self.inv[k] % self.MOD
res = res * self.inv[n - k] % self.MOD
return res
def nPk(self, n, k):
if k < 0 or k > n:
return 0
return self.fact[n] * self.inv[n - k] % self.MOD
def nHk(self, n, k):
if k == 0:
return 1
return self.__call__(n + k - 1, k)
def __make_factorial_list(self, N):
self.fact = [1] * (N + 1)
self.inv = [1] * (N + 1)
MOD = self.MOD
for i in range(1, N + 1):
self.fact[i] = (self.fact[i - 1] * i) % MOD
self.inv[N] = pow(self.fact[N], MOD - 2, MOD)
for i in range(N, 0, -1):
self.inv[i - 1] = (self.inv[i] * i) % MOD
return
mod = 10 ** 9 + 7
N = 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] = 1
stir[i][i] = 1
for 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] %= mod
ans = 0
for g in range(1, N + 1):
ans += stir[N][g]
ans %= mod
for G in range(1, N):
ng = N - G # ばらばらにする男女
tmp = 0
for g in range(2, G + 1):
C = stir[G][g] * pow(g * (g - 1), ng, mod)
tmp += C
tmp %= mod
ans += tmp * comb(N, G)
ans %= mod
print(ans)
tktk_snsn