結果
問題 | No.802 だいたい等差数列 |
ユーザー |
|
提出日時 | 2024-10-26 17:31:34 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 744 bytes |
コンパイル時間 | 453 ms |
コンパイル使用メモリ | 82,468 KB |
実行使用メモリ | 94,612 KB |
最終ジャッジ日時 | 2024-10-26 17:31:40 |
合計ジャッジ時間 | 5,098 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 23 RE * 7 |
ソースコード
N, M, D1, D2 = map(int, input().split())MOD = 10**9 + 7num = 10**6frac = [1] * numinv_frac = [1] * numfor i in range(2, num):frac[i] = (frac[i-1] * i) % MODinv_frac[-1] = pow(frac[-1], MOD-2, MOD)for i in range(num-1, 0, -1):inv_frac[i-1] = (inv_frac[i] * i) % MODdef comb(n, r):if r < 0 or r > n:return 0return (frac[n] * (inv_frac[n-r] * inv_frac[r])%MOD) % MODdef calc(n, m):# 合計 m個を n人で分ける# n人それぞれの最小個数は0個# {m+1}_H_{n-1} = {m+n-1}_C_{n-1}return comb(n+m-1, n-1)M = M - 1 - D1*(N-1)ans = calc(N+1, M)for k in range(1, N):m = M - (D2-D1+1)*kans += (-1)**(k%2) * comb(N-1, k) * calc(N+1, m)ans %= MODprint(ans)