結果

問題 No.1127 変形パスカルの三角形
ユーザー k
提出日時 2020-08-20 17:10:57
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 39 ms / 1,500 ms
コード長 1,168 bytes
コンパイル時間 2,233 ms
コンパイル使用メモリ 192,780 KB
最終ジャッジ日時 2025-01-13 04:26:42
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

#define REP(i,n) for(int i=0; i<(int)(n); i++)
#define FOR(i,b,e) for (int i=(int)(b); i<(int)(e); i++)
#define ALL(x) (x).begin(), (x).end()

const double PI = acos(-1);

const long long MOD = 1e9 + 7;
long long f[200000], r[200000];

long long modpow(long long x, long long p, long long mod) {
  long long ret = 1;
  while (p) {
    if (p & 1)
      ret = ret * x % mod;
    x = x * x % mod;
    p >>= 1;
  }
  return ret;
}

long long comb(int n, int k) {
  if (0 <= k && k <= n)
    return f[n] * r[k] % MOD * r[n-k] % MOD;
  return 0;
}

int main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);

  r[0] = f[0] = 1;
  for (int i = 1; i < 200000; i++) {
    f[i] = i * f[i-1] % MOD;
    r[i] = modpow(f[i], MOD - 2, MOD);
  }

  int n, k;
  long long a, b;
  cin >> a >> b >> n >> k;
  a %= MOD;
  b %= MOD;
  
  long long val = (a * comb(n-1, k-1) + b * comb(n-1, k-2)) % MOD;
  long long sum = 0;
  for (int k = 1; k <= n+1; k++) {
    long long tmp = (a * comb(n-1, k-1) + b * comb(n-1, k-2)) % MOD;
    sum += tmp * tmp % MOD;
    sum %= MOD;
  }

  cout << val << endl;
  cout << sum << endl;
  
  return 0;
}
0