結果
問題 | 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 1000000007using 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;}