結果

問題 No.1127 変形パスカルの三角形
ユーザー Kome_soudou
提出日時 2022-11-10 02:04:37
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 134 ms / 1,500 ms
コード長 1,566 bytes
コンパイル時間 1,744 ms
コンパイル使用メモリ 170,032 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2024-07-23 06:48:09
合計ジャッジ時間 5,276 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <bits/stdc++.h>
using namespace std;
// a^b mod p O(logP)
int64_t modpow(int64_t a, int64_t b, int64_t p)
{
    int64_t ans = 1;
    while(b > 0)
    {
        if(b & 1) ans = ans * a % p; //bの最下位bitが1ならa^(2^i)をかける
        a = a * a % p;
        b >>= 1; //bを1bit右にシフト
    }
    return ans;
}
vector<int64_t> fact; // 要素数(N + 1)
vector<int64_t> fact_inv; // 要素数(N + 1)
// a! mod pと(a!)^-1 mod pを配列に記録 O(NlogP) (やや遅い まだ高速化可能)
// ModComb実行前にこれを実行すること
void combinit(int64_t n, int64_t p)
{
    fact.resize(n + 1);
    fact_inv.resize(n + 1);
    fact.at(0) = 1; fact.at(1) = 1;
    for(int64_t i = 2; i <= n; i++)
    {
        fact.at(i) = fact.at(i - 1) * i;
        fact.at(i) %= p;
    }
    fact_inv.at(0) = 1; fact_inv.at(1) = 1;
    for(int64_t i = 2; i <= n; i++) fact_inv.at(i) = modpow(fact.at(i), p - 2, p);
}
// nCk mod p O(1)
int64_t modcomb(int64_t n, int64_t k, int64_t p)
{
  return ((fact.at(n) * fact_inv.at(k)) % p * fact_inv.at(n - k)) % p;
}
int main()
{
    int64_t a, b, n, k;
    cin >> a >> b >> n >> k;
    const int64_t mod = 1000000007;
    a %= mod;
    b %= mod;
    combinit(n - 1, mod);
    int64_t x = ((modcomb(n - 1, k - 1, mod) * a) % mod + modcomb(n - 1, k - 2, mod) * b) % mod;
    cout << x << endl;
    
    int64_t y = 0;
    int64_t cnt;
    for(int64_t i = 0; i <= n; i++)
    {
        cnt = 0;
        if(i != n)
        {
            cnt += modcomb(n - 1, i, mod) * a;
            cnt %= mod;
        }
        if(i != 0)
        {
            cnt += modcomb(n - 1, i - 1, mod) * b;
            cnt %= mod;
        }
        y += (cnt * cnt) % mod;
        y %= mod;
    }
    cout << y << endl;
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0