結果
問題 | No.802 だいたい等差数列 |
ユーザー |
![]() |
提出日時 | 2019-03-21 16:44:39 |
言語 | C++11(廃止可能性あり) (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 30 ms / 2,000 ms |
コード長 | 1,227 bytes |
コンパイル時間 | 460 ms |
コンパイル使用メモリ | 55,708 KB |
実行使用メモリ | 26,808 KB |
最終ジャッジ日時 | 2024-09-19 01:45:08 |
合計ジャッジ時間 | 2,544 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
#include <iostream> #define llint long long #define mod 1000000007 using namespace std; llint n, m, d1, d2, d; const int FACT_MAX = 1500005; llint fact[FACT_MAX], fact_inv[FACT_MAX]; llint modpow(llint a, llint n) { if(n == 0) return 1; if(n % 2){ return ((a%mod) * (modpow(a, n-1)%mod)) % mod; } else{ return modpow((a*a)%mod, n/2) % mod; } } void make_fact() { llint val = 1; fact[0] = 1; for(int i = 1; i < FACT_MAX; i++){ val *= i; val %= mod; fact[i] = val; } fact_inv[FACT_MAX-1] = modpow(fact[FACT_MAX-1], mod-2); for(int i = FACT_MAX-2; i >= 0; i--){ fact_inv[i] = fact_inv[i+1] * (i+1) % mod; } } llint comb(llint n, llint k) { llint ret = 1; ret *= fact[n]; ret *= fact_inv[k], ret %= mod; ret *= fact_inv[n-k], ret %= mod; return ret; } int main(void) { cin >> n >> m >> d1 >> d2; make_fact(); m -= (n-1)*d1; if(m < 1){ cout << 0 << endl; return 0; } d = d2-d1; d++, n--, m--; llint ans = 0; for(int k = 1; m - d*k >= 0 && k <= n; k++){ llint tmp = comb(n, k); tmp *= comb(m-d*k+n+1, n+1), tmp %= mod; if(k % 2 == 0) ans += mod - tmp; else ans += tmp; ans %= mod; } ans = comb(m+n+1, n+1) - ans + mod; ans %= mod; cout << ans << endl; return 0; }