結果
問題 | No.1127 変形パスカルの三角形 |
ユーザー |
|
提出日時 | 2020-08-18 20:15:03 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 13 ms / 1,500 ms |
コード長 | 1,400 bytes |
コンパイル時間 | 726 ms |
コンパイル使用メモリ | 73,452 KB |
最終ジャッジ日時 | 2025-01-13 03:26:20 |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
#include<iostream>#include<vector>using namespace std;typedef long long ll;const ll mod = 1e9 + 7;ll modpow(ll a, ll b, ll p = 1e9+7){if(b == 0) return 1;if(b % 2 == 0){ll d = modpow(a, b/2, p);return (d*d) % p;}else{return (a%p * modpow(a, b-1, p)) % p;}}struct ModComb{vector<ll> po, inv;ll N;ModComb(ll n) : N(n), po(n), inv(n) {po[0] = 1;for(int i = 1; i < N; i++) po[i] = (po[i-1] * i) % mod;inv[N-1] = modpow(po[N-1], mod-2, mod);for(int i = N-2; i >= 0; i--) inv[i] = (inv[i+1] * (i+1)) % mod;}ll nCk(ll n, ll k){if(k == 0) return 1;if(n < k) return 0;return (((po[n]*inv[n-k])%mod)*inv[k])%mod;}ll nPk(ll n, ll k){if(k == 0) return 1;if(n < k) return 0;return (po[n]*inv[n-k])%mod;}ll nHk(ll n, ll k){if(n == 0 && k == 0) return 1;return nCk(n+k-1, k-1);}};int main(){ll a, b, n, k;cin >> a >> b >> n >> k;a %= mod, b %= mod;ModComb mc(n+1);ll x = (a*mc.nCk(n-1, k-1)%mod + b*mc.nCk(n-1, n+1-k)%mod)%mod;ll y = (a*a%mod+b*b%mod)%mod;for(int i = 1; i < n; i++){ll t = (a*mc.nCk(n-1, i)%mod + b*mc.nCk(n-1, n-i)%mod)%mod;(y += t*t%mod) %= mod;}cout << x << endl << y << endl;return 0;}