結果

問題 No.802 だいたい等差数列
ユーザー ptotq
提出日時 2020-04-18 14:09:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 162 ms / 2,000 ms
コード長 604 bytes
コンパイル時間 292 ms
コンパイル使用メモリ 82,832 KB
実行使用メモリ 111,640 KB
最終ジャッジ日時 2024-10-03 23:35:25
合計ジャッジ時間 4,881 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
readline = sys.stdin.readline

N, M, d1, d2 = map(int, input().split())
mod = 10**9 + 7

max_n = 2 * 10**6
fac, inv = [1]*(max_n+1), [0]*(max_n+1)
for i in range(2, max_n+1):
    fac[i] = fac[i-1] * i % mod
inv[-1] = pow(fac[-1], mod-2, mod)
for i in range(max_n, 0, -1):
    inv[i-1] = inv[i] * i % mod

d = d2 - d1
x = M - 1 - d1 * (N-1)
if x < 0:
    print(0)
    exit()

ans = 0

for i in range(N):
    sign = -1 if i & 1 else 1
    if x+N-(d+1)*i >= N:
        ans += fac[x+N-(d+1)*i] * inv[N] * inv[x-(d+1)*i]\
            * fac[N-1] * inv[i] * inv[N-1-i] * sign % mod

print(ans % mod)
0