結果
問題 | No.802 だいたい等差数列 |
ユーザー | tktk_snsn |
提出日時 | 2021-04-28 00:54:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 151 ms / 2,000 ms |
コード長 | 1,469 bytes |
コンパイル時間 | 302 ms |
コンパイル使用メモリ | 86,860 KB |
実行使用メモリ | 98,612 KB |
最終ジャッジ日時 | 2023-09-21 01:28:25 |
合計ジャッジ時間 | 5,803 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 77 ms
76,324 KB |
testcase_01 | AC | 74 ms
75,904 KB |
testcase_02 | AC | 75 ms
76,268 KB |
testcase_03 | AC | 76 ms
76,572 KB |
testcase_04 | AC | 80 ms
76,312 KB |
testcase_05 | AC | 75 ms
75,656 KB |
testcase_06 | AC | 77 ms
76,304 KB |
testcase_07 | AC | 77 ms
76,064 KB |
testcase_08 | AC | 75 ms
75,888 KB |
testcase_09 | AC | 75 ms
75,656 KB |
testcase_10 | AC | 125 ms
92,264 KB |
testcase_11 | AC | 137 ms
96,476 KB |
testcase_12 | AC | 132 ms
94,404 KB |
testcase_13 | AC | 136 ms
95,576 KB |
testcase_14 | AC | 143 ms
95,284 KB |
testcase_15 | AC | 141 ms
98,612 KB |
testcase_16 | AC | 144 ms
95,976 KB |
testcase_17 | AC | 133 ms
95,332 KB |
testcase_18 | AC | 134 ms
95,460 KB |
testcase_19 | AC | 150 ms
96,632 KB |
testcase_20 | AC | 149 ms
97,108 KB |
testcase_21 | AC | 151 ms
97,196 KB |
testcase_22 | AC | 140 ms
96,540 KB |
testcase_23 | AC | 149 ms
96,912 KB |
testcase_24 | AC | 137 ms
96,380 KB |
testcase_25 | AC | 124 ms
91,472 KB |
testcase_26 | AC | 93 ms
81,228 KB |
testcase_27 | AC | 135 ms
95,548 KB |
testcase_28 | AC | 117 ms
85,784 KB |
testcase_29 | AC | 149 ms
96,900 KB |
testcase_30 | AC | 72 ms
71,560 KB |
testcase_31 | AC | 70 ms
71,360 KB |
testcase_32 | AC | 73 ms
71,348 KB |
testcase_33 | AC | 113 ms
85,872 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 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 N, M, D1, D2 = map(int, input().split()) mod = 10**9+7 comb = Combination(N+M, mod) ans = 0 for i in range(N): power = D1*(N-1) + i*(D2-D1+1) need = M-1-power if need < 0: continue cnt = comb(N-1, i) * comb(need + N, N) % mod if i % 2 == 0: ans += cnt else: ans += mod - cnt ans %= mod print(ans)