結果

問題 No.802 だいたい等差数列
ユーザー tempura_pptempura_pp
提出日時 2019-03-09 19:45:32
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 914 ms / 2,000 ms
コード長 825 bytes
コンパイル時間 85 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 112,640 KB
最終ジャッジ日時 2024-07-01 19:32:25
合計ジャッジ時間 12,310 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 1000000007

def mod_build(n) :
    fact = [1]*(n+1)
    inv_fact = [1]*(n+1)
    for i in range(1,n+1):
        fact[i] = fact[i-1]*i %mod
    inv_fact[n] = pow(fact[n],mod-2,mod)
    for i in range(n,0,-1):
        inv_fact[i-1] = inv_fact[i]*i %mod
    return fact, inv_fact
    
def comb(n,k,fact,inv_fact) : 
    if n<0 or k<0 or k>n :
        return 0
    return fact[n]*inv_fact[k]*inv_fact[n-k]%mod

def solve(n,m,d1,d2) :
    m = m-(n-1)*(d1-1)
    if m<0 :
        return 0
    d2=d2-d1+1
    fact, inv_fact = mod_build(max(m+1,n))
    ans = 0
    for i in range(0,n):
        ret=comb(m-d2*i,n,fact,inv_fact)*comb(n-1,i,fact,inv_fact)%mod
        if i%2 ==1 : 
            ans+=mod-ret
        else :
            ans+=ret
    return ans%mod

n, m, d1, d2 = map(int, input().split())
print(solve(n, m, d1, d2))
0