結果

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

ソースコード

diff # 

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define ALL(v) v.begin(), v.end()
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

const int MAX = 200010;
const int MOD = 1000000007;

ll fac[MAX], finv[MAX], inv[MAX];

void init(){
  fac[0]=fac[1]=1;
  finv[0]=finv[1]=1;
  inv[1]=1;
  for(int i=2;i<MAX;i++){
    fac[i]=fac[i-1]*i%MOD;
    inv[i]=MOD-inv[MOD%i]*(MOD/i)%MOD;
    finv[i]=finv[i-1]*inv[i]%MOD;
  }
}

ll nCr(int n,int k){
  if(n<k) return 0;
  if(n<0 || k<0) return 0;
  return fac[n]*(finv[k]*finv[n-k]%MOD)%MOD;
}

int main(){
  ll a,b,n,k;
  cin>>a>>b>>n>>k;
  a=a%MOD,b=b%MOD;
  
  init();
  
  vector<ll> A(n+2);
  for(int i=1;i<=n+1;i++){
    if(i==1) A[i]=a;
    else if(i==n+1) A[i]=b;
    else A[i]=(nCr(n-1,i-1)*a%MOD+nCr(n-1,i-2)*b%MOD)%MOD;
  }
  
  ll ans=0;
  for(int i=1;i<=n+1;i++) ans=(ans+A[i]*A[i]%MOD)%MOD;
  
  cout<<A[k]<<endl;
  cout<<ans<<endl;

  return 0;
}
0