結果

問題 No.1127 変形パスカルの三角形
ユーザー karinohitokarinohito
提出日時 2021-07-26 01:11:56
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 61 ms / 1,500 ms
コード長 1,334 bytes
コンパイル時間 1,507 ms
コンパイル使用メモリ 174,464 KB
実行使用メモリ 16,128 KB
最終ジャッジ日時 2024-07-21 09:45:35
合計ジャッジ時間 3,508 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

//#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
//using namespace atcoder;
using ll = long long;
#define all(A) A.begin(),A.end()
using vll = vector<ll>;
#define rep(i, n) for (long long i = 0; i < (long long)(n); i++)
using Graph = vector<vector<ll>>;
Graph G;
vll dist;
vector<bool> seen;
set<pair<ll, ll>> S;

ll gcd(ll(a), ll(b)) {
	ll c = a;
	while (a % b != 0) {
		c = a % b;
		a = b;
		b = c;
	}
	return b;
}
vector<ll> fact, factinv, inv;
ll mod = 1e9 + 7;
void prenCkModp(ll n) {
	fact.resize(n + 5);
	factinv.resize(n + 5);
	inv.resize(n + 5);
	fact.at(0) = fact.at(1) = 1;
	factinv.at(0) = factinv.at(1) = 1;
	inv.at(1) = 1;
	for (ll i = 2; i < n + 5; i++) {
		fact.at(i) = (fact.at(i - 1) * i) % mod;
		inv.at(i) = mod - (inv.at(mod % i) * (mod / i)) % mod;
		factinv.at(i) = (factinv.at(i - 1) * inv.at(i)) % mod;
	}

}
ll nCk(ll n, ll k) {
	if (n < k) return 0;
	return fact.at(n) * (factinv.at(k) * factinv.at(n - k) % mod) % mod;
}

int main() {
	ll A,B,N,K;
	cin>>A>>B>>N>>K;
	A%=mod;
	B%=mod;
	prenCkModp(2*N+K);
	ll AA=(A%mod)*nCk(N-1,K-1);
	ll BB=(B%mod)*nCk(N-1,N-K+1);
	cout<<(AA%mod+BB%mod)%mod<<endl;
	ll an=A*A+B*B;
	an%=mod;
	for(ll k=2;k<=N;k++){
		AA=(A%mod)*nCk(N-1,k-1);
		BB=(B%mod)*nCk(N-1,N-k+1);
		AA%=mod;
		BB%=mod;
		an+=((AA+BB)*(AA+BB))%mod;
		
	}
	cout<<an%mod<<endl;

}
0