結果
問題 | No.2120 場合の数の下8桁 |
ユーザー | とりゐ |
提出日時 | 2022-11-04 22:32:19 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 119 ms / 2,000 ms |
コード長 | 2,632 bytes |
コンパイル時間 | 593 ms |
コンパイル使用メモリ | 86,252 KB |
実行使用メモリ | 81,644 KB |
最終ジャッジ日時 | 2023-09-26 01:02:17 |
合計ジャッジ時間 | 4,169 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 115 ms
81,300 KB |
testcase_01 | AC | 119 ms
81,452 KB |
testcase_02 | AC | 114 ms
81,428 KB |
testcase_03 | AC | 112 ms
81,220 KB |
testcase_04 | AC | 118 ms
81,220 KB |
testcase_05 | AC | 113 ms
81,464 KB |
testcase_06 | AC | 112 ms
81,404 KB |
testcase_07 | AC | 115 ms
81,128 KB |
testcase_08 | AC | 112 ms
81,164 KB |
testcase_09 | AC | 112 ms
81,376 KB |
testcase_10 | AC | 115 ms
81,200 KB |
testcase_11 | AC | 113 ms
81,444 KB |
testcase_12 | AC | 112 ms
81,380 KB |
testcase_13 | AC | 112 ms
81,624 KB |
testcase_14 | AC | 113 ms
81,328 KB |
testcase_15 | AC | 112 ms
81,440 KB |
testcase_16 | AC | 114 ms
81,644 KB |
testcase_17 | AC | 112 ms
81,220 KB |
testcase_18 | AC | 113 ms
81,632 KB |
testcase_19 | AC | 116 ms
81,456 KB |
ソースコード
class BinomialCoefficient: def __init__(self, m): self.MOD = m self.factorization = self._factorize(m) self.facs = [] self.invs = [] self.coeffs = [] self.pows = [] for p, pe in self.factorization: fac = [1]*pe for i in range(1, pe): fac[i] = fac[i-1]*(i if i % p else 1) % pe inv = [1]*pe inv[-1] = fac[-1] for i in range(1, pe)[::-1]: inv[i-1] = inv[i]*(i if i % p else 1) % pe self.facs.append(fac) self.invs.append(inv) # coeffs c = self._modinv(m // pe, pe) self.coeffs.append(m//pe*c % m) # pows powp = [1] while powp[-1]*p != pe: powp.append(powp[-1]*p) self.pows.append(powp) def __call__(self, n, k): if k < 0 or k > n: return 0 if k == 0 or k == n: return 1 % self.MOD res = 0 for i, (p, pe) in enumerate(self.factorization): res += self._choose_pe(n, k, p, pe, self.facs[i], self.invs[i], self.pows[i]) * self.coeffs[i] res %= self.MOD return res def _E(self, n, k, r, p): res = 0 while n: n //= p k //= p r //= p res += n - k - r return res def _choose_pe(self, n, k, p, pe, fac, inv, powp): r = n-k e0 = self._E(n, k, r, p) if e0 >= len(powp): return 0 res = powp[e0] if (p != 2 or pe == 4) and self._E(n//(pe//p), k//(pe//p), r//(pe//p), p) % 2: res = pe-res while n: res = res * fac[n % pe] % pe * inv[k % pe] % pe * inv[r % pe] % pe n //= p k //= p r //= p return res def _factorize(self, N): factorization = [] for i in range(2, N+1): if i*i > N: break if N % i: continue c = 0 while N % i == 0: N //= i c += 1 factorization.append((i, i**c)) if N != 1: factorization.append((N, N)) return factorization def _modinv(self, a, MOD): r0, r1, s0, s1 = a, MOD, 1, 0 while r1: r0, r1, s0, s1 = r1, r0 % r1, s1, s0-r0//r1*s1 return s0 % MOD m=int(input()) n=int(input()) P=10**8 c=BinomialCoefficient(P) ans=c(m,n) if ans==0: print('0'*8) else: ans=str(ans) print('0'*(8-len(ans))+ans)