結果
問題 | No.391 CODING WAR |
ユーザー | tktk_snsn |
提出日時 | 2020-12-15 12:11:04 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 680 ms / 2,000 ms |
コード長 | 1,191 bytes |
コンパイル時間 | 80 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 18,688 KB |
最終ジャッジ日時 | 2024-09-20 01:36:16 |
合計ジャッジ時間 | 6,201 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 33 ms
10,880 KB |
testcase_01 | AC | 31 ms
10,624 KB |
testcase_02 | AC | 31 ms
10,624 KB |
testcase_03 | AC | 32 ms
10,880 KB |
testcase_04 | AC | 31 ms
10,880 KB |
testcase_05 | AC | 32 ms
10,624 KB |
testcase_06 | AC | 31 ms
10,624 KB |
testcase_07 | AC | 31 ms
10,624 KB |
testcase_08 | AC | 31 ms
10,624 KB |
testcase_09 | AC | 680 ms
18,688 KB |
testcase_10 | AC | 558 ms
18,560 KB |
testcase_11 | AC | 451 ms
18,560 KB |
testcase_12 | AC | 32 ms
10,624 KB |
testcase_13 | AC | 629 ms
18,560 KB |
testcase_14 | AC | 541 ms
16,640 KB |
testcase_15 | AC | 616 ms
17,408 KB |
testcase_16 | AC | 378 ms
14,464 KB |
testcase_17 | AC | 429 ms
15,488 KB |
testcase_18 | AC | 307 ms
13,568 KB |
testcase_19 | AC | 311 ms
13,696 KB |
ソースコード
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 __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, M = map(int, input().split()) comb = Combination(M + 10, mod) ans = 0 for k in range(M + 1): #M個中k個の問題を解かないの確定、他は任意 cnt = comb(M, k) * pow(M - k, N, mod) % mod if k & 1: cnt = mod - cnt ans += cnt ans %= mod print(ans)